def test_ssl_enabled(self): cfg = mock.Mock() cfg.ssl_enabled = True cfg.ssl_ca_certs = '/absolute/path/to/certificate.crt' cfg.migrations_consistency_level = 'LOCAL_QUORUM' cfg.load_balance_strategy = 'RoundRobinPolicy' driver._connection(cfg, None) kwargs = self.mock_cluster.call_args[1] self.assertTrue('ssl_options' in kwargs) ssl_options = kwargs['ssl_options'] self.assertEqual(cfg.ssl_ca_certs, ssl_options['ca_certs']) self.assertEqual(ssl.PROTOCOL_TLSv1, ssl_options['ssl_version'])
def test_ssl_disabled(self): cfg = mock.Mock() cfg.ssl_enabled = False cfg.migrations_consistency_level = 'LOCAL_QUORUM' cfg.load_balance_strategy = 'RoundRobinPolicy' driver._connection(cfg, None) kwargs = self.mock_cluster.call_args[1] # ssl_options may or may not be provided to the Cluster constructor # depending on the implementation, but if it is provided, ensure it # has been set to None if 'ssl_options' in kwargs: self.assertIsNone(kwargs['ssl_options'])
def test_ssl_enabled(self): cfg = mock.Mock() cfg.ssl_enabled = True cfg.ssl_ca_certs = '/absolute/path/to/certificate.crt' cfg.load_balance_strategy = 'RoundRobinPolicy' with mock.patch('cassandra.cluster.Cluster') as mock_cluster: driver._connection(cfg, None) kwargs = mock_cluster.call_args[1] self.assertTrue('ssl_options' in kwargs) ssl_options = kwargs['ssl_options'] self.assertEqual(cfg.ssl_ca_certs, ssl_options['ca_certs']) self.assertEqual(ssl.PROTOCOL_TLSv1, ssl_options['ssl_version'])
def test_ssl_disabled(self): cfg = mock.Mock() cfg.ssl_enabled = False cfg.load_balance_strategy = 'RoundRobinPolicy' with mock.patch('cassandra.cluster.Cluster') as mock_cluster: driver._connection(cfg, None) kwargs = mock_cluster.call_args[1] # ssl_options may or may not be provided to the Cluster constructor # depending on the implementation, but if it is provided, ensure it # has been set to None if 'ssl_options' in kwargs: self.assertIsNone(kwargs['ssl_options'])
def test_create_dc_aware_policy(self): cfg = mock.Mock() cfg.migrations_consistency_level = 'LOCAL_QUORUM' cfg.load_balance_strategy = "DCAwareRoundRobinPolicy" cfg.cluster = ['localhost'] cluster_instance = mock.Mock() self.mock_cluster.return_value = cluster_instance mock_session = mock.Mock() cluster_instance.connect.return_value = mock_session session = driver._connection(cfg, "ORD") self.assertEqual(mock_session, session)
def test_create_dc_aware_policy(self): cfg = mock.Mock() cfg.load_balance_strategy = "DCAwareRoundRobinPolicy" cfg.cluster = ['localhost'] with mock.patch("cassandra.cluster.Cluster") as mock_cluster: cluster_instance = mock.Mock() mock_cluster.return_value = cluster_instance mock_session = mock.Mock() cluster_instance.connect.return_value = mock_session session = driver._connection(cfg, "ORD") self.assertEqual(mock_session, session)
def test_auth_enabled(self): cfg = mock.Mock() cfg.migrations_consistency_level = 'LOCAL_QUORUM' cfg.load_balance_strategy = "RoundRobinPolicy" cfg.auth_enabled = True cfg.cluster = ['localhost'] with mock.patch("cassandra.cluster.Cluster") as mock_cluster: cluster_instance = mock.Mock() mock_cluster.return_value = cluster_instance mock_session = mock.Mock() cluster_instance.connect.return_value = mock_session session = driver._connection(cfg, "ORD") self.assertEqual(mock_session, session)