Adding support for Cassandra and Scylla

This add the support for those cassandra based dbs
and their drivers, cassandra-driver, scylla-driver

Ref: https://cassandra.apache.org/
Ref: https://www.scylladb.com/
Ref: https://pypi.org/project/cassandra-driver/
Ref: https://pypi.org/project/scylla-driver/

Author: fruch

.github/workflows/main.yml

@@ -31,6 +31,8 @@ jobs:
           - webdriver.py
           - keycloak.py
           - arangodb.py
+          - scylladb.py
+          - cassandra.py
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2

docs/database.rst

@@ -13,3 +13,5 @@ Allows to spin up database images such as MySQL, PostgreSQL, MariaDB, Oracle XE,
 .. autoclass:: testcontainers.clickhouse.ClickHouseContainer
 .. autoclass:: testcontainers.neo4j.Neo4jContainer
 .. autoclass:: testcontainers.arangodb.ArangoDbContainer
+.. autoclass:: testcontainers.cassandra.CassandraContainer
+.. autoclass:: testcontainers.scylla.ScyllaContainer

requirements.in

@@ -1,4 +1,4 @@
--e file:.[docker-compose,mysql,oracle,postgresql,selenium,google-cloud-pubsub,mongo,redis,mssqlserver,neo4j,kafka,rabbitmq,clickhouse,keycloak,arangodb]
+-e file:.[docker-compose,mysql,oracle,postgresql,selenium,google-cloud-pubsub,mongo,redis,mssqlserver,neo4j,kafka,rabbitmq,clickhouse,keycloak,arangodb,cassandra,scylla]
 codecov>=2.1.0
 cryptography<37
 flake8<3.8.0  # 3.8.0 adds a dependency on importlib-metadata which conflicts with other packages.

setup.py

@@ -68,6 +68,8 @@
         'clickhouse': ['clickhouse-driver'],
         'keycloak': ['python-keycloak'],
         'arangodb': ['python-arango'],
+        'cassandra': ['cassandra-driver'],
+        'scylla': ['scylla-driver']
     },
     long_description_content_type="text/x-rst",
     long_description=long_description,

testcontainers/cassandra.py

@@ -0,0 +1,40 @@
+from testcontainers.core.generic import DockerContainer
+from testcontainers.core.waiting_utils import wait_container_is_ready
+
+
+class CassandraContainer(DockerContainer):
+    """
+    Cassandra database container.
+
+    Example
+    -------
+    ::
+
+        with CassandraContainer() as cassandra:
+            cluster = cassandra.get_cluster()
+            with cluster.connect() as session:
+                session.execute(
+                    "CREATE KEYSPACE keyspace1 WITH replication = "
+                    "{'class': 'SimpleStrategy', 'replication_factor': '1'};")
+
+    """
+    def __init__(self, image="rinscy/cassandra:latest", ports_to_expose=[9042]):
+        super(CassandraContainer, self).__init__(image)
+        self.ports_to_expose = ports_to_expose
+        self.with_exposed_ports(*self.ports_to_expose)
+
+    @wait_container_is_ready()
+    def _connect(self):
+        cluster = self.get_cluster()
+        cluster.connect()
+
+    def start(self):
+        super(CassandraContainer, self).start()
+        self._connect()
+        return self
+
+    def get_cluster(self, **kwargs):
+        from cassandra.cluster import Cluster
+        hostname = self.get_container_host_ip()
+        port = self.get_exposed_port(9042)
+        return Cluster(contact_points=[hostname], port=port, **kwargs)

testcontainers/scylla.py

@@ -0,0 +1,41 @@
+from testcontainers.core.generic import DockerContainer
+from testcontainers.core.waiting_utils import wait_container_is_ready
+
+
+class ScyllaContainer(DockerContainer):
+    """
+    Scylla database container.
+
+    Example
+    -------
+    ::
+
+        with ScyllaContainer() as scylla:
+            cluster = scylla.get_cluster()
+            with cluster.connect() as session:
+                session.execute(
+                    "CREATE KEYSPACE keyspace1 WITH replication "
+                    "= {'class': 'SimpleStrategy', 'replication_factor': '1'};")
+
+    """
+    def __init__(self, image="scylladb/scylla:4.6.rc1", ports_to_expose=[9042]):
+        super(ScyllaContainer, self).__init__(image)
+        self.ports_to_expose = ports_to_expose
+        self.with_exposed_ports(*self.ports_to_expose)
+        self.with_command("--skip-wait-for-gossip-to-settle=0")
+
+    @wait_container_is_ready()
+    def _connect(self):
+        cluster = self.get_cluster()
+        cluster.connect()
+
+    def start(self):
+        super(ScyllaContainer, self).start()
+        self._connect()
+        return self
+
+    def get_cluster(self, **kwargs):
+        from cassandra.cluster import Cluster
+        hostname = self.get_container_host_ip()
+        port = self.get_exposed_port(9042)
+        return Cluster(contact_points=[hostname], port=port, **kwargs)

tests/test_cassandra.py

@@ -0,0 +1,18 @@
+from testcontainers.cassandra import CassandraContainer
+
+
+def test_docker_run_cassandra():
+    with CassandraContainer() as cassandra:
+        cluster = cassandra.get_cluster()
+        with cluster.connect() as session:
+            session.execute(
+                "CREATE KEYSPACE keyspace1 WITH replication = "
+                "{'class': 'SimpleStrategy', 'replication_factor': '1'};")
+            session.execute(
+                "CREATE TABLE keyspace1.table1 (key1 int, key2 int, PRIMARY KEY (key1));")
+            session.execute("INSERT INTO keyspace1.table1 (key1,key2) values (1,2);")
+
+            response = session.execute("SELECT * FROM keyspace1.table1")
+
+            assert response.one().key1 == 1
+            assert response.one().key2 == 2

tests/test_scylla.py

@@ -0,0 +1,18 @@
+from testcontainers.scylla import ScyllaContainer
+
+
+def test_docker_run_scylla():
+    with ScyllaContainer() as scylla:
+        cluster = scylla.get_cluster()
+        with cluster.connect() as session:
+            session.execute(
+                "CREATE KEYSPACE keyspace1 WITH replication = "
+                "{'class': 'SimpleStrategy', 'replication_factor': '1'};")
+            session.execute(
+                "CREATE TABLE keyspace1.table1 (key1 int, key2 int, PRIMARY KEY (key1));")
+            session.execute("INSERT INTO keyspace1.table1 (key1,key2) values (1,2);")
+
+            response = session.execute("SELECT * FROM keyspace1.table1")
+
+            assert response.one().key1 == 1
+            assert response.one().key2 == 2