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_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_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"))
