Exemple #1
0
    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'])
Exemple #2
0
    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'])
Exemple #3
0
    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'])
Exemple #4
0
    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'])
Exemple #5
0
    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'])
Exemple #6
0
    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'])
Exemple #7
0
    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)
Exemple #8
0
    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)
Exemple #9
0
    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)
Exemple #10
0
    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)
Exemple #11
0
    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)