Fixed unit tests after PYTHON-734

bjmb · bjmb · commit cf75d547f132 · 2017-05-18T17:30:56.000-04:00
diff --git a/tests/unit/test_connection.py b/tests/unit/test_connection.py
@@ -358,7 +358,8 @@ def test_no_req_ids(self, *args):
         self.assertEqual(max_connection.send_msg.call_count, 0)
         self.assertEqual(max_connection.send_msg.call_count, 0)
         max_connection.defunct.assert_has_calls([call(ANY)] * get_holders.call_count)
-        holder.return_connection.assert_has_calls([call(max_connection)] * get_holders.call_count)
+        holder.return_connection.assert_has_calls(
+            [call(max_connection, mark_host_down=True)] * get_holders.call_count)
 
     def test_unexpected_response(self, *args):
         request_id = 999
@@ -386,7 +387,8 @@ def send_msg(msg, req_id, msg_callback):
         exc = connection.defunct.call_args_list[0][0][0]
         self.assertIsInstance(exc, ConnectionException)
         self.assertRegexpMatches(exc.args[0], r'^Received unexpected response to OptionsMessage.*')
-        holder.return_connection.assert_has_calls([call(connection)] * get_holders.call_count)
+        holder.return_connection.assert_has_calls(
+            [call(connection, mark_host_down=True)] * get_holders.call_count)
 
     def test_timeout(self, *args):
         request_id = 999
@@ -415,7 +417,8 @@ def send_msg(msg, req_id, msg_callback):
         self.assertIsInstance(exc, OperationTimedOut)
         self.assertEqual(exc.errors, 'Connection heartbeat timeout after 0.05 seconds')
         self.assertEqual(exc.last_host, 'localhost')
-        holder.return_connection.assert_has_calls([call(connection)] * get_holders.call_count)
+        holder.return_connection.assert_has_calls(
+            [call(connection, mark_host_down=True)] * get_holders.call_count)
 
 
 class TimerTest(unittest.TestCase):
