@ -1,5 +1,5 @@
/ *
* Copyright 2012 - 20 19 the original author or authors .
* Copyright 2012 - 20 20 the original author or authors .
*
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
@ -16,23 +16,25 @@
package org.springframework.boot.test.autoconfigure.data.cassandra ;
import java.time.Duration ;
import java.util.UUID ;
import com.datastax.driver.core.Cluster ;
import com.datastax.oss.driver.api.core.CqlSession ;
import com.datastax.oss.driver.api.core.CqlSessionBuilder ;
import org.junit.jupiter.api.Test ;
import org.testcontainers.containers.CassandraContainer ;
import org.testcontainers.junit.jupiter.Container ;
import org.testcontainers.junit.jupiter.Testcontainers ;
import org.springframework.beans.factory.NoSuchBeanDefinitionException ;
import org.springframework.beans.factory.annotation.Autowired ;
import org.springframework.boot.test.autoconfigure.data.redis.ExampleService ;
import org.springframework.boot.test.util.TestPropertyValues ;
import org.springframework.boot.testsupport.testcontainers.CassandraContainer ;
import org.springframework.boot.testsupport.testcontainers.DisabledWithoutDockerTestcontainers ;
import org.springframework.boot.test.context.TestConfiguration ;
import org.springframework.context.ApplicationContext ;
import org.springframework.context.ApplicationContextInitializer ;
import org.springframework.context.ConfigurableApplicationContext ;
import org.springframework.context.annotation.Bean ;
import org.springframework.data.cassandra.core.CassandraTemplate ;
import org.springframework.test.context.ContextConfiguration ;
import org.springframework.test.context.DynamicPropertyRegistry ;
import org.springframework.test.context.DynamicPropertySource ;
import static org.assertj.core.api.Assertions.assertThat ;
import static org.assertj.core.api.Assertions.assertThatExceptionOfType ;
@ -42,13 +44,21 @@ import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
*
* @author Artsiom Yudovin
* /
@DisabledWithoutDockerTestcontainers
@ContextConfiguration ( initializers = DataCassandraTestIntegrationTests . Initializer . class )
@DataCassandraTest
@DataCassandraTest ( properties = { "spring.data.cassandra.local-datacenter=datacenter1" ,
"spring.data.cassandra.schema-action=create-if-not-exists" ,
"spring.data.cassandra.connection.connect-timeout=10s" , "spring.data.cassandra.request.timeout=5s" } )
@Testcontainers ( disabledWithoutDocker = true )
class DataCassandraTestIntegrationTests {
@Container
public static CassandraContainer cassandra = new CassandraContainer ( ) ;
static final CassandraContainer < ? > cassandra = new CassandraContainer < > ( ) . withStartupAttempts ( 5 )
. withStartupTimeout ( Duration . ofMinutes ( 10 ) ) ;
@DynamicPropertySource
static void cassandraProperties ( DynamicPropertyRegistry registry ) {
registry . add ( "spring.data.cassandra.contact-points" ,
( ) - > cassandra . getHost ( ) + ":" + cassandra . getFirstMappedPort ( ) ) ;
}
@Autowired
private CassandraTemplate cassandraTemplate ;
@ -67,36 +77,28 @@ class DataCassandraTestIntegrationTests {
@Test
void testRepository ( ) {
Person person = new Person ( ) ;
person . setDescription ( "Look, new @DataCassandraTest!" ) ;
ExampleEntity entity = new ExampleEntity ( ) ;
entity . setDescription ( "Look, new @DataCassandraTest!" ) ;
String id = UUID . randomUUID ( ) . toString ( ) ;
person . setId ( id ) ;
Person savedEntity = this . exampleRepository . save ( person ) ;
Person getEntity = this . cassandraTemplate . selectOneById ( id , Person . class ) ;
entity . setId ( id ) ;
ExampleEntity savedEntity = this . exampleRepository . save ( entity ) ;
ExampleEntity getEntity = this . cassandraTemplate . selectOneById ( id , ExampleEntity . class ) ;
assertThat ( getEntity ) . isNotNull ( ) ;
assertThat ( getEntity . getId ( ) ) . isNotNull ( ) ;
assertThat ( getEntity . getId ( ) ) . isEqualTo ( savedEntity . getId ( ) ) ;
this . exampleRepository . deleteAll ( ) ;
}
static class Initializer implements ApplicationContextInitializer < ConfigurableApplicationContext > {
@Override
public void initialize ( ConfigurableApplicationContext configurableApplicationContext ) {
TestPropertyValues . of ( "spring.data.cassandra.port=" + cassandra . getFirstMappedPort ( ) )
. and ( "spring.data.cassandra.keyspaceName=test" )
. and ( "spring.data.cassandra.schema-action=CREATE_IF_NOT_EXISTS" )
. applyTo ( configurableApplicationContext . getEnvironment ( ) ) ;
@TestConfiguration ( proxyBeanMethods = false )
static class KeyspaceTestConfiguration {
Cluster cluster = Cluster . builder ( ) . withoutJMXReporting ( )
. addContactPoints ( cassandra . getContainerIpAddress ( ) ) . withPort ( cassandra . getFirstMappedPort ( ) )
. build ( ) ;
cluster . connect ( ) . execute ( "CREATE KEYSPACE test "
+ "WITH replication = {'class':'SimpleStrategy', 'replication_factor' : 1};" ) ;
@Bean
CqlSession cqlSession ( CqlSessionBuilder cqlSessionBuilder ) {
try ( CqlSession session = cqlSessionBuilder . build ( ) ) {
session . execute ( "CREATE KEYSPACE IF NOT EXISTS boot_test"
+ " WITH REPLICATION = { 'class' : 'SimpleStrategy', 'replication_factor' : 1 };" ) ;
}
return cqlSessionBuilder . withKeyspace ( "boot_test" ) . build ( ) ;
}
}
Stephane Nicoll
4 years ago