test fix

qubole · Apr 21, 2020 · 20c530b · 20c530b
1 parent 8328c31
commit 20c530b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_connection.py b/tests/test_connection.py
@@ -33,7 +33,7 @@ def test_connection_override(self):
         Connection.__init__ = Mock(return_value=None)
         Connection._api_call = Mock(return_value={})
         qds.main()
-        Connection.__init__.assert_called_with(ANY, ANY, ANY, ANY, 5, 10)
+        Connection.__init__.assert_called_with(ANY, ANY, ANY, ANY, 7, 10)
 
     #Test with no values given should set default
     def test_connection_default(self):
@@ -42,7 +42,7 @@ def test_connection_default(self):
         Connection.__init__ = Mock(return_value=None)
         Connection._api_call = Mock(return_value={})
         qds.main()
-        Connection.__init__.assert_called_with(ANY, ANY, ANY, ANY, 5, 10)
+        Connection.__init__.assert_called_with(ANY, ANY, ANY, ANY, 7, 10)
 
 if __name__ == '__main__':
     unittest.main()