From af3c346b335989530bbd7382f8064a31d24b32fc Mon Sep 17 00:00:00 2001 From: Adam Holmberg Date: Wed, 11 May 2016 15:46:17 -0500 Subject: [PATCH] update unit tests for new conditional refresh PYTHON-557 --- tests/unit/test_control_connection.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_control_connection.py b/tests/unit/test_control_connection.py index d15801a3..e5a6dcc1 100644 --- a/tests/unit/test_control_connection.py +++ b/tests/unit/test_control_connection.py @@ -347,7 +347,7 @@ class ControlConnectionTest(unittest.TestCase): } self.cluster.scheduler.reset_mock() self.control_connection._handle_topology_change(event) - self.cluster.scheduler.schedule_unique.assert_called_once_with(ANY, self.control_connection.refresh_node_list_and_token_map) + self.cluster.scheduler.schedule_unique.assert_called_once_with(ANY, self.control_connection._refresh_nodes_if_not_up, '1.2.3.4') event = { 'change_type': 'REMOVED_NODE', @@ -363,7 +363,7 @@ class ControlConnectionTest(unittest.TestCase): } self.cluster.scheduler.reset_mock() self.control_connection._handle_topology_change(event) - self.cluster.scheduler.schedule_unique.assert_called_once_with(ANY, self.control_connection.refresh_node_list_and_token_map) + self.cluster.scheduler.schedule_unique.assert_called_once_with(ANY, self.control_connection._refresh_nodes_if_not_up, '1.2.3.4') def test_handle_status_change(self): event = { @@ -453,7 +453,7 @@ class ControlConnectionTest(unittest.TestCase): cc_no_schema_refresh._handle_status_change(status_event) cc_no_schema_refresh._handle_topology_change(topo_event) cluster.scheduler.schedule_unique.assert_has_calls([call(ANY, cc_no_schema_refresh.refresh_node_list_and_token_map), - call(ANY, cc_no_schema_refresh.refresh_node_list_and_token_map)]) + call(ANY, cc_no_schema_refresh._refresh_nodes_if_not_up, '1.2.3.4')]) cc_no_topo_refresh = ControlConnection(cluster, 1, 0, -1, 0) cluster.scheduler.reset_mock()