Add test for Cluster keyword args

This commit is contained in:
Tyler Hobbs
2013-06-03 15:13:22 -05:00
parent 1ee1a536c5
commit 146b74a189

View File

@@ -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()