def test_credentials_but_no_secrets(self):
with self.assertRaises(TypeError):
cluster_from_config(
{
"cassandra.contact_points": "127.0.0.1",
"cassandra.credentials_secret": "secret/foo/bar",
}
)
def test_port(self):
cluster = cluster_from_config({
"cassandra.contact_points": "127.0.1.1",
"cassandra.port": "9999",
})
self.assertEqual(cluster.port, 9999)
def test_kwarg_passthrough(self):
cluster = cluster_from_config(
{
"cassandra.contact_points": "127.0.0.1",
}, protocol_version=3)
self.assertEqual(cluster.protocol_version, 3)
def test_contact_points(self):
cluster = cluster_from_config({
"cassandra.contact_points":
"127.0.1.1, 127.0.1.2",
})
self.assertEqual(cluster.contact_points, ["127.0.1.1", "127.0.1.2"])
def test_alternate_prefix(self):
cluster = cluster_from_config(
{
"noodle.contact_points": "127.0.1.1, 127.0.1.2",
},
prefix="noodle.")
self.assertEqual(cluster.contact_points, ["127.0.1.1", "127.0.1.2"])
def test_credentials(self):
secrets = mock.Mock(autospec=SecretsStore)
cluster = cluster_from_config(
{
"cassandra.contact_points": "127.0.0.1",
"cassandra.credentials_secret": "secret/foo/bar",
},
secrets=secrets,
)
self.assertIsNotNone(cluster.auth_provider)
def create_schema(app_config):
"""Create schema in the database.
Run this with:
baseplate-script example.ini {{cookiecutter.module_name}}.models.cql:create_schema
"""
keyspace = '{{ cookiecutter.project_slug }}'
cluster = cluster_from_config(app_config)
session = cluster.connect()
conn = connection.Connection(session)
create_keyspace_simple(conn, keyspace, 1)
session = cluster.connect(keyspace)
conn = connection.Connection(session)
models = [
MyModel,
]
for model in models:
sync_table(conn, model)
def test_alternate_prefix(self):
cluster = cluster_from_config({
"noodle.contact_points": "127.0.1.1, 127.0.1.2",
}, prefix="noodle.")
self.assertEqual(cluster.contact_points, ["127.0.1.1", "127.0.1.2"])
def test_kwarg_passthrough(self):
cluster = cluster_from_config({
"cassandra.contact_points": "127.0.0.1",
}, protocol_version=3)
self.assertEqual(cluster.protocol_version, 3)
def test_contact_points(self):
cluster = cluster_from_config({
"cassandra.contact_points": "127.0.1.1, 127.0.1.2",
})
self.assertEqual(cluster.contact_points, ["127.0.1.1", "127.0.1.2"])
def test_empty_config(self):
with self.assertRaises(ConfigurationError):
cluster_from_config({})
