fix unit test following cluster/policy update

2016-04-07 14:50:36 -05:00
parent cdfb03e7b7
commit e8fda07c2f
1 changed files with 1 additions and 1 deletions
--- a/tests/unit/test_policies.py
+++ b/tests/unit/test_policies.py
@@ -483,7 +483,7 @@ class DCAwareRoundRobinPolicyTest(unittest.TestCase):
        host_none = Host(1, SimpleConvictionPolicy)

        # contact point is '1'
-        cluster = Mock(contact_points=[1])
+        cluster = Mock(contact_points_resolved=[1])

        # contact DC first
        policy = DCAwareRoundRobinPolicy()