def get_new_connection(self, connection_settings): contact_points = connection_settings.pop( 'contact_points', self.default_settings['CONTACT_POINTS'] ) keyspace = connection_settings.pop( 'keyspace', self.settings_dict['DEFAULT_KEYSPACE'] ) self.keyspace = keyspace try: connection.get_connection() except CQLEngineException: connection.setup( contact_points, keyspace, **connection_settings ) self.session = connection.get_session() self.cluster = connection.get_cluster() self.session.default_timeout = None # Should be in config. return CassandraCursor(self.session)
def test_uncommitted_session_uses_legacy(self): cluster = Cluster() session = cluster.connect() session.row_factory = dict_factory connection.set_session(session) conn = connection.get_connection() self.assertEqual(conn.cluster._config_mode, _ConfigMode.LEGACY)
def test_connection_from_session_with_legacy_settings(self): cluster = Cluster(load_balancing_policy=RoundRobinPolicy()) session = cluster.connect() session.row_factory = dict_factory connection.set_session(session) conn = connection.get_connection() self.assertEqual(conn.cluster._config_mode, _ConfigMode.LEGACY)
def test_legacy_insert_query(self): connection.setup(hosts=[CASSANDRA_IP], default_keyspace=DEFAULT_KEYSPACE, consistency=ConsistencyLevel.LOCAL_ONE) self.assertEqual(connection.get_connection().cluster._config_mode, _ConfigMode.LEGACY) sync_table(ConnectionModel) ConnectionModel.objects.create(key=0, some_data='text0') ConnectionModel.objects.create(key=1, some_data='text1') self.assertEqual(ConnectionModel.objects(key=0)[0].some_data, 'text0')
def test_connection_from_session_with_execution_profile(self): cluster = Cluster( execution_profiles={ EXEC_PROFILE_DEFAULT: ExecutionProfile( row_factory=dict_factory) }) session = cluster.connect() connection.default() connection.set_session(session) conn = connection.get_connection() self.assertEqual(conn.cluster._config_mode, _ConfigMode.PROFILES)
def test_connection_from_hosts(self): """ Test to ensure that you can register a connection from a list of hosts @since 3.8 @jira_ticket PYTHON-692 @expected_result queries should execute appropriately @test_category object_mapper """ connection_name = 'from_hosts' conn.register_connection(connection_name, hosts=[CASSANDRA_IP]) self.assertIsNotNone(conn.get_connection(connection_name).cluster.metadata.get_host(CASSANDRA_IP)) self.addCleanup(conn.unregister_connection, connection_name)
def test_execution_profile_insert_query(self): cluster = Cluster( execution_profiles={ EXEC_PROFILE_DEFAULT: ExecutionProfile( row_factory=dict_factory) }) session = cluster.connect() connection.default() connection.set_session(session) self.assertEqual(connection.get_connection().cluster._config_mode, _ConfigMode.PROFILES) sync_table(ConnectionModel) ConnectionModel.objects.create(key=0, some_data='text0') ConnectionModel.objects.create(key=1, some_data='text1') self.assertEqual(ConnectionModel.objects(key=0)[0].some_data, 'text0')
def test_connection_creation_from_session(self): """ Test to ensure that you can register a connection from a session @since 3.8 @jira_ticket PYTHON-649 @expected_result queries should execute appropriately @test_category object_mapper """ cluster = Cluster([CASSANDRA_IP]) session = cluster.connect() connection_name = 'from_session' conn.register_connection(connection_name, session=session) self.assertIsNotNone(conn.get_connection(connection_name).cluster.metadata.get_host(CASSANDRA_IP)) self.addCleanup(conn.unregister_connection, connection_name) cluster.shutdown()
def test_connection_setup_with_default(self): connection.default() self.assertIsNotNone( connection.get_connection("default").cluster.metadata.get_host( "127.0.0.1"))
def test_connection_setup_with_setup(self): connection.setup(hosts=None, default_keyspace=None) self.assertIsNotNone( connection.get_connection("default").cluster.metadata.get_host( "127.0.0.1"))
def test_connection_with_legacy_settings(self): connection.setup(hosts=[CASSANDRA_IP], default_keyspace=DEFAULT_KEYSPACE, consistency=ConsistencyLevel.LOCAL_ONE) conn = connection.get_connection() self.assertEqual(conn.cluster._config_mode, _ConfigMode.LEGACY)
def test_default_connection_uses_legacy(self): connection.default() conn = connection.get_connection() self.assertEqual(conn.cluster._config_mode, _ConfigMode.LEGACY)
def test_connection_setup_with_setup(self): connection.setup(hosts=None, default_keyspace=None) self.assertIsNotNone(connection.get_connection("default").cluster.metadata.get_host("127.0.0.1"))
def test_connection_setup_with_default(self): connection.default() self.assertIsNotNone(connection.get_connection("default").cluster.metadata.get_host("127.0.0.1"))