Tests Stand Up Cassandra

This allows the tests to run and confirm that the guide
still works.

Author: jzheaux

complete/build.gradle

@@ -19,6 +19,7 @@ dependencies {
 	implementation 'org.springframework.boot:spring-boot-starter-data-cassandra'
 	//implementation 'com.datastax.astra:astra-spring-boot-starter:0.1.13'
 	testImplementation 'org.springframework.boot:spring-boot-starter-test'
+	runtimeOnly "org.springframework.boot:spring-boot-docker-compose"
 }
 
 tasks.named('test') {

complete/pom.xml

@@ -35,6 +35,12 @@
 			<artifactId>spring-boot-starter-test</artifactId>
 			<scope>test</scope>
 		</dependency>
+
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-docker-compose</artifactId>
+			<optional>true</optional>
+		</dependency>
 	</dependencies>
 
 	<build>

complete/src/main/java/com/example/accessingdatacassandra/AccessingDataCassandraApplication.java

@@ -6,11 +6,11 @@
 
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+
 import org.springframework.boot.CommandLineRunner;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.context.annotation.Bean;
-import org.springframework.data.cassandra.core.CassandraTemplate;
 
 @SpringBootApplication
 public class AccessingDataCassandraApplication {

complete/src/main/resources/application.properties

@@ -6,6 +6,11 @@ spring.cassandra.connection.init-query-timeout=10s
 spring.cassandra.local-datacenter=datacenter1
 spring.cassandra.keyspace-name=spring_cassandra
 
+### Properties for Docker Compose
+spring.docker.compose.file=classpath:docker/compose.yml
+spring.docker.compose.skip.in-tests=false
+
+### Properties for Astra DB
 # Credentials to Astra DB
 #astra.client-id=<CLIENT_ID>
 #astra.client-secret=<CLIENT_SECRET>

complete/src/main/resources/docker/compose.yml

@@ -0,0 +1,10 @@
+services:
+  cassandra:
+    image: cassandra:4.0.7
+    ports:
+      - "9042:9042"
+    healthcheck:
+      test: ["CMD", "cqlsh", "-u", "cassandra", "-p", "cassandra"]
+      interval: 10s
+      timeout: 10s
+      retries: 10

complete/src/test/java/com/example/accessingdatacassandra/AccessingDataCassandraApplicationTests.java

@@ -1,7 +1,17 @@
 package com.example.accessingdatacassandra;
 
+import java.util.List;
+
 import org.junit.jupiter.api.Test;
+
+import org.springframework.boot.autoconfigure.cassandra.CassandraProperties;
 import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.context.TestConfiguration;
+import org.springframework.data.cassandra.config.AbstractCassandraConfiguration;
+import org.springframework.data.cassandra.config.SchemaAction;
+import org.springframework.data.cassandra.core.cql.keyspace.CreateKeyspaceSpecification;
+import org.springframework.data.cassandra.core.cql.keyspace.SpecificationBuilder;
+import org.springframework.lang.NonNull;
 
 @SpringBootTest
 class AccessingDataCassandraApplicationTests {
@@ -10,4 +20,34 @@ class AccessingDataCassandraApplicationTests {
 	void contextLoads() {
 	}
 
+	@TestConfiguration
+	static class CreateKeyspaceConfiguration extends AbstractCassandraConfiguration {
+
+		private final CassandraProperties properties;
+
+		CreateKeyspaceConfiguration(CassandraProperties properties) {
+			this.properties = properties;
+		}
+
+		@NonNull
+		@Override
+		protected List<CreateKeyspaceSpecification> getKeyspaceCreations() {
+			CreateKeyspaceSpecification specification = SpecificationBuilder
+				.createKeyspace(getKeyspaceName()).ifNotExists()
+				.withSimpleReplication(1);
+
+			return List.of(specification);
+		}
+
+		@Override
+		public SchemaAction getSchemaAction() {
+			return SchemaAction.RECREATE_DROP_UNUSED;
+		}
+
+		@NonNull
+		@Override
+		protected String getKeyspaceName() {
+			return this.properties.getKeyspaceName();
+		}
+	}
 }

initial/build.gradle

@@ -16,6 +16,7 @@ dependencies {
 	implementation 'org.springframework.boot:spring-boot-starter-data-cassandra'
 	//implementation 'com.datastax.astra:astra-spring-boot-starter:0.1.13'
 	testImplementation 'org.springframework.boot:spring-boot-starter-test'
+	runtimeOnly "org.springframework.boot:spring-boot-docker-compose"
 }
 
 test {

initial/pom.xml

@@ -38,6 +38,13 @@
 			<artifactId>spring-boot-starter-test</artifactId>
 			<scope>test</scope>
 		</dependency>
+
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-docker-compose</artifactId>
+			<optional>true</optional>
+		</dependency>
+
 	</dependencies>
 
 	<build>

initial/src/main/resources/application.properties

@@ -6,6 +6,11 @@ spring.cassandra.connection.init-query-timeout=10s
 spring.cassandra.local-datacenter=datacenter1
 spring.cassandra.keyspace-name=spring_cassandra
 
+### Properties for Docker Compose
+spring.docker.compose.file=classpath:docker/compose.yml
+spring.docker.compose.skip.in-tests=false
+
+### Properties for Astra DB
 # Credentials to Astra DB
 #astra.client-id=<CLIENT_ID>
 #astra.client-secret=<CLIENT_SECRET>

initial/src/main/resources/docker/compose.yml

@@ -0,0 +1,10 @@
+services:
+  cassandra:
+    image: cassandra:4.0.7
+    ports:
+      - "9042:9042"
+    healthcheck:
+      test: ["CMD", "cqlsh", "-u", "cassandra", "-p", "cassandra"]
+      interval: 10s
+      timeout: 10s
+      retries: 10