Use Testcontainers for Cassandra examples.

See #606.

Author: mp911de

cassandra/util/pom.xml

@@ -21,40 +21,18 @@
 		</dependency>
 
 		<dependency>
-			<groupId>org.apache.cassandra</groupId>
-			<artifactId>cassandra-all</artifactId>
-			<version>3.11.5</version>
-			<exclusions>
-				<exclusion>
-					<artifactId>guava</artifactId>
-					<groupId>com.google.guava</groupId>
-				</exclusion>
-				<exclusion>
-					<groupId>io.netty</groupId>
-					<artifactId>netty-all</artifactId>
-				</exclusion>
-				<exclusion>
-					<groupId>com.addthis.metrics</groupId>
-					<artifactId>reporter-config3</artifactId>
-				</exclusion>
-			</exclusions>
+			<groupId>junit</groupId>
+			<artifactId>junit</artifactId>
 		</dependency>
 
 		<dependency>
-			<groupId>com.datastax.oss</groupId>
-			<artifactId>java-driver-core</artifactId>
+			<groupId>org.testcontainers</groupId>
+			<artifactId>cassandra</artifactId>
 		</dependency>
 
 		<dependency>
-			<groupId>org.cassandraunit</groupId>
-			<artifactId>cassandra-unit</artifactId>
-			<version>4.3.1.0</version>
-			<exclusions>
-				<exclusion>
-					<groupId>io.netty</groupId>
-					<artifactId>netty-handler</artifactId>
-				</exclusion>
-			</exclusions>
+			<groupId>com.datastax.oss</groupId>
+			<artifactId>java-driver-core</artifactId>
 		</dependency>
 
 		<dependency>

cassandra/util/src/main/java/example/springdata/cassandra/util/CassandraExtension.java

@@ -22,6 +22,10 @@
 import org.junit.jupiter.api.extension.ExtensionContext;
 import org.junit.platform.commons.util.AnnotationUtils;
 
+import org.springframework.util.StringUtils;
+
+import org.testcontainers.containers.CassandraContainer;
+
 import com.datastax.oss.driver.api.core.CqlSession;
 
 /**
@@ -36,20 +40,26 @@ class CassandraExtension implements BeforeAllCallback {
 			.create(CassandraExtension.class);
 
 	@Override
-	public void beforeAll(ExtensionContext context) throws Exception {
+	public void beforeAll(ExtensionContext context) {
+
+		var store = context.getStore(NAMESPACE);
+		var cassandra = findAnnotation(context);
 
-		ExtensionContext.Store store = context.getStore(NAMESPACE);
-		CassandraKeyspace cassandra = findAnnotation(context);
+		var keyspace = store.getOrComputeIfAbsent(CassandraServer.class, it -> {
 
-		CassandraServer keyspace = store.getOrComputeIfAbsent(CassandraServer.class, it -> {
-			return CassandraServer.embeddedIfNotRunning("localhost", 9042);
+			CassandraContainer container = runTestcontainer();
+			System.setProperty("spring.data.cassandra.port", "" + container.getMappedPort(9042));
+			System.setProperty("spring.data.cassandra.contact-points", "" + container.getHost());
+
+			return new CassandraServer(container.getHost(), container.getMappedPort(9042),
+					CassandraServer.RuntimeMode.EMBEDDED_IF_NOT_RUNNING);
 		}, CassandraServer.class);
 
 		keyspace.before();
 
-		CqlSession session = store.getOrComputeIfAbsent(CqlSession.class, it -> {
+		var session = store.getOrComputeIfAbsent(CqlSession.class, it -> {
 
-			return CqlSession.builder().addContactPoint(new InetSocketAddress("localhost", 9042))
+			return CqlSession.builder().addContactPoint(new InetSocketAddress(keyspace.host(), keyspace.port()))
 					.withLocalDatacenter("datacenter1").build();
 		}, CqlSession.class);
 
@@ -59,10 +69,26 @@ public void beforeAll(ExtensionContext context) throws Exception {
 
 	private static CassandraKeyspace findAnnotation(ExtensionContext context) {
 
-		Class<?> testClass = context.getRequiredTestClass();
+		var testClass = context.getRequiredTestClass();
 
-		Optional<CassandraKeyspace> annotation = AnnotationUtils.findAnnotation(testClass, CassandraKeyspace.class);
+		var annotation = AnnotationUtils.findAnnotation(testClass, CassandraKeyspace.class);
 
 		return annotation.orElseThrow(() -> new IllegalStateException("Test class not annotated with @Cassandra"));
 	}
+
+	private CassandraContainer<?> runTestcontainer() {
+
+		var container = new CassandraContainer<>(getCassandraDockerImageName());
+		container.withReuse(true);
+
+		container.start();
+
+		return container;
+	}
+
+	private String getCassandraDockerImageName() {
+
+		return String.format("cassandra:%s",
+				Optional.ofNullable(System.getenv("CASSANDRA_VERSION")).filter(StringUtils::hasText).orElse("3.11.10"));
+	}
 }

cassandra/util/src/main/java/example/springdata/cassandra/util/CassandraServer.java

@@ -19,7 +19,6 @@
 import java.net.Socket;
 import java.util.concurrent.TimeUnit;
 
-import org.cassandraunit.utils.EmbeddedCassandraServerHelper;
 import org.junit.jupiter.api.Assumptions;
 
 import org.springframework.util.Assert;
@@ -30,18 +29,8 @@
  *
  * @author Mark Paluch
  */
-class CassandraServer {
-
-	private final String host;
-	private final int port;
-	private final RuntimeMode runtimeMode;
-
-	private CassandraServer(String host, int port, RuntimeMode runtimeMode) {
-
-		this.host = host;
-		this.port = port;
-		this.runtimeMode = runtimeMode;
-	}
+record CassandraServer(String host, int port,
+		example.springdata.cassandra.util.CassandraServer.RuntimeMode runtimeMode) {
 
 	/**
 	 * Require a running instance on {@code host:port}. Fails with {@link AssumptionViolatedException} if Cassandra is not
@@ -75,7 +64,7 @@ public static boolean isConnectable(String host, int port) {
 
 		Assert.hasText(host, "Host must not be null or empty!");
 
-		try (Socket socket = new Socket()) {
+		try (var socket = new Socket()) {
 
 			socket.setSoLinger(true, 0);
 			socket.connect(new InetSocketAddress(host, port), (int) TimeUnit.MILLISECONDS.convert(10, TimeUnit.SECONDS));
@@ -95,7 +84,7 @@ public int getPort() {
 		return port;
 	}
 
-	protected void before() throws Exception {
+	protected void before() {
 
 		if (runtimeMode == RuntimeMode.REQUIRE_RUNNING_INSTANCE) {
 			Assumptions.assumeTrue(isConnectable(getHost(), getPort()),
@@ -107,12 +96,9 @@ protected void before() throws Exception {
 				return;
 			}
 		}
-
-		EmbeddedCassandraServerHelper.startEmbeddedCassandra("embedded-cassandra.yaml", "target/embeddedCassandra",
-				TimeUnit.SECONDS.toMillis(60));
 	}
 
-	private enum RuntimeMode {
+	enum RuntimeMode {
 		REQUIRE_RUNNING_INSTANCE, EMBEDDED_IF_NOT_RUNNING;
 	}
 }

cassandra/util/src/main/java/example/springdata/cassandra/util/CassandraVersion.java

@@ -42,8 +42,8 @@ public static Version getReleaseVersion(CqlSession session) {
 
 		Assert.notNull(session, "Session must not be null");
 
-		ResultSet resultSet = session.execute("SELECT release_version FROM system.local;");
-		Row row = resultSet.one();
+		var resultSet = session.execute("SELECT release_version FROM system.local;");
+		var row = resultSet.one();
 
 		return Version.parse(row.getString(0));
 	}