diff --git a/tests/integration/test_cluster.py b/tests/integration/test_cluster.py index a33642f6..e3477c83 100644 --- a/tests/integration/test_cluster.py +++ b/tests/integration/test_cluster.py @@ -1,4 +1,5 @@ import unittest +from mock import Mock from cassandra.cluster import Cluster, NoHostAvailable @@ -40,6 +41,25 @@ class ClusterTests(unittest.TestCase): cluster = Cluster(['127.1.2.9', '127.1.2.10']) self.assertRaises(NoHostAvailable, cluster.connect) + def test_init_kwargs(self): + def foo(*args, **kwargs): + return Mock() + + for kw in ('auth_provider', 'load_balancing_policy_factory', + 'retry_policy_factory', 'conviction_policy_factory'): + kwargs = {kw: 123} + self.assertRaises(ValueError, Cluster, **kwargs) + + kwargs = {kw: foo} + c = Cluster(**kwargs) + self.assertEquals(getattr(c, kw), foo) + + for kw in ('contact_points', 'port', 'compression', 'metrics_enabled', + 'reconnection_policy', 'sockopts', 'max_schema_agreement_wait'): + kwargs = {kw: (1, 2, 3)} + c = Cluster(**kwargs) + self.assertEquals(getattr(c, kw), (1, 2, 3)) + def test_submit_schema_refresh(self): cluster = Cluster() cluster.connect()