fix unit test following addition of address_translator

aholmberg · aholmberg · commit 613c0052825c · 2016-04-08T16:31:19.000-05:00
diff --git a/tests/unit/test_control_connection.py b/tests/unit/test_control_connection.py
@@ -25,7 +25,7 @@
 from cassandra.cluster import ControlConnection, _Scheduler
 from cassandra.pool import Host
 from cassandra.policies import (SimpleConvictionPolicy, RoundRobinPolicy,
-                                ConstantReconnectionPolicy)
+                                ConstantReconnectionPolicy, IdentityTranslator)
 
 PEER_IP = "foobar"
 
@@ -61,6 +61,7 @@ class MockCluster(object):
     max_schema_agreement_wait = 5
     load_balancing_policy = RoundRobinPolicy()
     reconnection_policy = ConstantReconnectionPolicy(2)
+    address_translator = IdentityTranslator()
     down_host = None
     contact_points = []
     is_shutdown = False
