From 087293050b322b736a7468ae70c2044d1510c47a Mon Sep 17 00:00:00 2001 From: Adam Holmberg Date: Wed, 6 May 2015 16:29:23 -0500 Subject: [PATCH] Update unit tests for new refresh_schema signature. --- tests/unit/test_control_connection.py | 6 +++--- tests/unit/test_response_future.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unit/test_control_connection.py b/tests/unit/test_control_connection.py index 1f1ccc1b..9c93f5af 100644 --- a/tests/unit/test_control_connection.py +++ b/tests/unit/test_control_connection.py @@ -410,12 +410,12 @@ class ControlConnectionTest(unittest.TestCase): } self.cluster.scheduler.reset_mock() self.control_connection._handle_schema_change(event) - self.cluster.scheduler.schedule_unique.assert_called_once_with(0.0, self.control_connection.refresh_schema, 'ks1', 'table1', None) + self.cluster.scheduler.schedule_unique.assert_called_once_with(0.0, self.control_connection.refresh_schema, 'ks1', 'table1', None, None, None) self.cluster.scheduler.reset_mock() event['table'] = None self.control_connection._handle_schema_change(event) - self.cluster.scheduler.schedule_unique.assert_called_once_with(0.0, self.control_connection.refresh_schema, 'ks1', None, None) + self.cluster.scheduler.schedule_unique.assert_called_once_with(0.0, self.control_connection.refresh_schema, 'ks1', None, None, None, None) def test_refresh_disabled(self): cluster = MockCluster() @@ -463,4 +463,4 @@ class ControlConnectionTest(unittest.TestCase): cc_no_topo_refresh._handle_schema_change(schema_event) cluster.scheduler.schedule_unique.assert_has_calls([call(ANY, cc_no_topo_refresh.refresh_node_list_and_token_map), call(0.0, cc_no_topo_refresh.refresh_schema, - schema_event['keyspace'], schema_event['table'], None)]) + schema_event['keyspace'], schema_event['table'], None, None, None)]) diff --git a/tests/unit/test_response_future.py b/tests/unit/test_response_future.py index 027fe732..92351a9d 100644 --- a/tests/unit/test_response_future.py +++ b/tests/unit/test_response_future.py @@ -105,7 +105,7 @@ class ResponseFutureTests(unittest.TestCase): kind=RESULT_KIND_SCHEMA_CHANGE, results={'keyspace': "keyspace1", "table": "table1"}) rf._set_result(result) - session.submit.assert_called_once_with(ANY, 'keyspace1', 'table1', None, ANY, rf) + session.submit.assert_called_once_with(ANY, 'keyspace1', 'table1', None, None, None, ANY, rf) def test_other_result_message_kind(self): session = self.make_session()