generator-jhipster 7.6.0 → 7.8.1

package/generators/server/templates/src/test/java/package/config/CassandraTestContainer.java.ejs

@@ -0,0 +1,122 @@
+package <%= packageName %>.config;
+
+import static java.nio.file.Files.newDirectoryStream;
+import static java.nio.file.Paths.get;
+import static java.util.Spliterator.SORTED;
+import static java.util.Spliterators.spliteratorUnknownSize;
+import static java.util.stream.StreamSupport.stream;
+
+import com.datastax.driver.core.Cluster;
+import com.datastax.driver.core.Session;
+import com.datastax.oss.driver.api.core.CqlSession;
+import java.io.IOException;
+import java.net.InetSocketAddress;
+import java.net.URISyntaxException;
+import java.net.URL;
+import java.nio.file.Path;
+import java.time.Duration;
+import java.time.temporal.ChronoUnit;
+import java.util.Iterator;
+import java.util.Spliterator;
+
+import com.datastax.oss.driver.api.core.CqlSessionBuilder;
+import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
+import com.datastax.oss.driver.api.core.config.DriverConfigLoader;
+import org.cassandraunit.CQLDataLoader;
+import org.cassandraunit.dataset.cql.ClassPathCQLDataSet;
+import org.springframework.beans.factory.DisposableBean;
+import org.springframework.beans.factory.InitializingBean;
+import org.springframework.test.context.junit.jupiter.SpringExtension;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.testcontainers.containers.CassandraContainer;
+import org.testcontainers.containers.output.Slf4jLogConsumer;
+
+/**
+ * Base class for starting/stopping Cassandra during tests.
+ */
+public class CassandraTestContainer implements InitializingBean, DisposableBean {
+
+    private static final Logger log = LoggerFactory.getLogger(CassandraTestContainer.class);
+    private static final Integer DATABASE_REQUEST_TIMEOUT = 20;
+    private static final Integer CONTAINER_STARTUP_TIMEOUT_MINUTES = 10;
+    private CassandraContainer cassandraContainer;
+
+    @Override
+    public void destroy() {
+        if (null != cassandraContainer && cassandraContainer.isRunning()) {
+            cassandraContainer.stop();
+        }
+    }
+
+    @Override
+    public void afterPropertiesSet() {
+        if (null == cassandraContainer) {
+            cassandraContainer = (CassandraContainer) new CassandraContainer("<%= DOCKER_CASSANDRA %>")
+                .withStartupTimeout(Duration.of(CONTAINER_STARTUP_TIMEOUT_MINUTES, ChronoUnit.MINUTES))
+                .withLogConsumer(new Slf4jLogConsumer(log))
+                .withReuse(true);
+        }
+        if (!cassandraContainer.isRunning()) {
+            cassandraContainer.start();
+            Cluster cluster = cassandraContainer.getCluster();
+
+            try(Session session = cluster.connect()) {
+                createTestKeyspace(session);
+            }
+            CqlSession cqlSession = new CqlSessionBuilder()
+                .addContactPoint(
+                    new InetSocketAddress(
+                        cassandraContainer.getHost(),
+                        cassandraContainer.getMappedPort(CassandraContainer.CQL_PORT))
+                ).withLocalDatacenter(cluster.getMetadata().getAllHosts().iterator().next().getDatacenter())
+                .withKeyspace(CQLDataLoader.DEFAULT_KEYSPACE_NAME)
+                .withConfigLoader(getConfigLoader())
+                .build();
+            CQLDataLoader dataLoader = new CQLDataLoader(cqlSession);
+            applyScripts(dataLoader, "config/cql/changelog/", "*.cql");
+            cqlSession.close();
+        }
+    }
+
+    public CassandraContainer getCassandraContainer() {
+        return cassandraContainer;
+    }
+
+    private DriverConfigLoader getConfigLoader() {
+        return  DriverConfigLoader.programmaticBuilder()
+            .withDuration(DefaultDriverOption.REQUEST_TIMEOUT, Duration.ofSeconds(DATABASE_REQUEST_TIMEOUT))
+            .build();
+    }
+
+    private void createTestKeyspace(Session session) {
+        String createQuery =
+            "CREATE KEYSPACE " + CQLDataLoader.DEFAULT_KEYSPACE_NAME + " WITH replication={'class' : 'SimpleStrategy', 'replication_factor':1}";
+        session.execute(createQuery);
+    }
+
+    private void applyScripts(CQLDataLoader dataLoader, String cqlDir, String pattern) {
+        URL dirUrl = ClassLoader.getSystemResource(cqlDir);
+        if (dirUrl == null) { // protect for empty directory
+            return;
+        }
+
+        Iterator<Path> pathIterator = null;
+        try {
+            pathIterator = newDirectoryStream(get(dirUrl.toURI()), pattern).iterator();
+        } catch (IOException e) {
+            log.error("error trying to reading CQL chagelog", e);
+        } catch (URISyntaxException e) {
+            log.error("error trying to get CQL chagelog uri", e);
+        }
+
+        Spliterator<Path> pathSpliterator = spliteratorUnknownSize(pathIterator, SORTED);
+        stream(pathSpliterator, false)
+            .map(Path::getFileName)
+            .map(Path::toString)
+            .sorted()
+            .map(file -> cqlDir + file)
+            .map(dataSetLocation -> new ClassPathCQLDataSet(dataSetLocation, false, false, dataLoader.getSession().getKeyspace().get().toString()))
+            .forEach(dataLoader::load);
+    }
+}

package/generators/server/templates/src/test/java/package/config/EmbeddedCassandra.java.ejs

@@ -0,0 +1,11 @@
+package <%= packageName %>.config;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.RUNTIME)
+public @interface EmbeddedCassandra {
+}

package/generators/server/templates/src/test/java/package/config/EmbeddedKafka.java.ejs

@@ -0,0 +1,11 @@
+package <%= packageName %>.config;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.RUNTIME)
+public @interface EmbeddedKafka {
+}

package/generators/server/templates/src/test/java/package/config/EmbeddedMongo.java.ejs

@@ -0,0 +1,11 @@
+package <%= packageName %>.config;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.RUNTIME)
+public @interface EmbeddedMongo {
+}

package/generators/server/templates/src/test/java/package/config/JHipsterBlockHoundIntegration.java.ejs

@@ -30,6 +30,8 @@ public class JHipsterBlockHoundIntegration implements BlockHoundIntegration {
         builder.allowBlockingCallsInside("org.springframework.validation.beanvalidation.SpringValidatorAdapter", "validate");
     <%_ if (databaseTypeCouchbase) { _%>
         builder.allowBlockingCallsInside("org.springframework.data.couchbase.core.convert.MappingCouchbaseConverter", "read");
+        builder.allowBlockingCallsInside("com.github.couchmove.repository.CouchbaseRepositoryImpl", "lambda$query$2");
+        builder.allowBlockingCallsInside("com.github.couchmove.repository.CouchbaseRepositoryImpl", "lambda$importFtsIndex$1");
     <%_} _%>
     }
 

package/generators/server/templates/src/test/java/package/config/KafkaTestContainer.java.ejs

@@ -0,0 +1,38 @@
+package <%= packageName %>.config;
+
+import org.springframework.beans.factory.DisposableBean;
+import org.springframework.beans.factory.InitializingBean;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.testcontainers.containers.KafkaContainer;
+import org.testcontainers.containers.output.Slf4jLogConsumer;
+import org.testcontainers.utility.DockerImageName;
+
+public class KafkaTestContainer implements InitializingBean, DisposableBean {
+
+    private KafkaContainer kafkaContainer;
+    private static final Logger log = LoggerFactory.getLogger(KafkaTestContainer.class);
+
+    @Override
+    public void destroy() {
+        if (null != kafkaContainer && kafkaContainer.isRunning()) {
+            kafkaContainer.stop();
+        }
+    }
+
+    @Override
+    public void afterPropertiesSet() {
+        if (null == kafkaContainer) {
+            kafkaContainer = new KafkaContainer(DockerImageName.parse("<%= DOCKER_KAFKA %>"))
+                .withLogConsumer(new Slf4jLogConsumer(log))
+                .withReuse(true);
+        }
+        if (!kafkaContainer.isRunning()) {
+            kafkaContainer.start();
+        }
+    }
+
+    public KafkaContainer getKafkaContainer() {
+        return kafkaContainer;
+    }
+}

package/generators/server/templates/src/test/java/package/config/MongoDbTestContainer.java.ejs

@@ -0,0 +1,67 @@
+<%#
+ Copyright 2013-2022 the original author or authors from the JHipster project.
+
+ This file is part of the JHipster project, see https://www.jhipster.tech/
+ for more information.
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+      https://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+-%>
+package <%= packageName %>.config;
+
+import java.util.Collections;
+
+import org.springframework.beans.factory.DisposableBean;
+import org.springframework.beans.factory.InitializingBean;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.testcontainers.containers.MongoDBContainer;
+import org.testcontainers.containers.output.Slf4jLogConsumer;
+
+public class MongoDbTestContainer implements InitializingBean, DisposableBean {
+    private final long memoryInBytes = Math.round(1024 * 1024 * 1024 * 0.6);
+    private final long memorySwapInBytes = Math.round(1024 * 1024 * 1024 * 0.8);
+    private final long nanoCpu = Math.round(1_000_000_000L * 0.1);
+    private static final Logger log = LoggerFactory.getLogger(MongoDbTestContainer.class);
+
+    private MongoDBContainer mongodbContainer;
+
+    @Override
+    public void destroy() {
+        if (null != mongodbContainer && mongodbContainer.isRunning()) {
+            mongodbContainer.stop();
+        }
+    }
+
+    @Override
+    public void afterPropertiesSet() {
+        if (null == mongodbContainer) {
+            mongodbContainer = new MongoDBContainer("<%= DOCKER_MONGODB %>")
+                .withTmpFs(Collections.singletonMap("/testtmpfs", "rw"))
+                .withCommand(
+                    "--nojournal --wiredTigerCacheSizeGB 0.25 --wiredTigerCollectionBlockCompressor none --slowOpSampleRate 0 --setParameter ttlMonitorEnabled=false --setParameter diagnosticDataCollectionEnabled=false --setParameter logicalSessionRefreshMillis=6000000 --setParameter enableFlowControl=false --setParameter oplogFetcherUsesExhaust=false --setParameter disableResumableRangeDeleter=true --setParameter enableShardedIndexConsistencyCheck=false --setParameter enableFinerGrainedCatalogCacheRefresh=false --setParameter readHedgingMode=off --setParameter loadRoutingTableOnStartup=false --setParameter rangeDeleterBatchDelayMS=2000000 --setParameter skipShardingConfigurationChecks=true --setParameter syncdelay=3600"
+                )
+                .withCreateContainerCmdModifier(cmd ->
+                    cmd.getHostConfig().withMemory(memoryInBytes).withMemorySwap(memorySwapInBytes).withNanoCPUs(nanoCpu)
+                )
+                .withLogConsumer(new Slf4jLogConsumer(log))
+                .withReuse(true);
+        }
+        if (!mongodbContainer.isRunning()) {
+            mongodbContainer.start();
+        }
+    }
+
+    public MongoDBContainer getMongoDBContainer() {
+        return mongodbContainer;
+    }
+}

package/generators/server/templates/src/test/java/package/config/TestContainersSpringContextCustomizerFactory.java.ejs

@@ -0,0 +1,114 @@
+<%#
+ Copyright 2013-2022 the original author or authors from the JHipster project.
+
+ This file is part of the JHipster project, see https://www.jhipster.tech/
+ for more information.
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+      https://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+-%>
+package <%= packageName %>.config;
+
+import java.util.List;
+
+<%_ if (databaseTypeCassandra) { _%>
+import org.cassandraunit.CQLDataLoader;
+<%_ } _%>
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
+import org.springframework.beans.factory.support.DefaultSingletonBeanRegistry;
+import org.springframework.core.annotation.AnnotatedElementUtils;
+
+import org.springframework.test.context.ContextConfigurationAttributes;
+import org.springframework.test.context.ContextCustomizer;
+import org.springframework.test.context.ContextCustomizerFactory;
+import org.springframework.boot.test.util.TestPropertyValues;
+<%_ if (databaseTypeCassandra) { _%>
+import org.testcontainers.containers.CassandraContainer;
+<%_ } _%>
+<%_ if (messageBrokerKafka) { _%>
+import org.testcontainers.containers.KafkaContainer;
+<%_ } _%>
+
+
+public class TestContainersSpringContextCustomizerFactory implements ContextCustomizerFactory {
+
+    private Logger log = LoggerFactory.getLogger(TestContainersSpringContextCustomizerFactory.class);
+<%_ if (messageBrokerKafka) { _%>
+    private static KafkaTestContainer kafkaBean;
+<%_ } _%>
+<%_ if (databaseTypeMongodb) { _%>
+    private static MongoDbTestContainer mongoDbBean;
+<%_ } _%>
+<%_ if (databaseTypeCassandra) { _%>
+    private static CassandraTestContainer cassandraBean;
+<%_ } _%>
+
+    @Override
+    public ContextCustomizer createContextCustomizer(Class<?> testClass, List<ContextConfigurationAttributes> configAttributes) {
+
+        return (context, mergedConfig) -> {
+            ConfigurableListableBeanFactory beanFactory = context.getBeanFactory();
+            TestPropertyValues testValues = TestPropertyValues.empty();
+<%_ if (databaseTypeMongodb) { _%>
+        EmbeddedMongo mongoAnnotation = AnnotatedElementUtils.findMergedAnnotation(testClass, EmbeddedMongo.class);
+        if (null != mongoAnnotation) {
+            log.debug("detected the EmbeddedMongo annotation on class {}", testClass.getName());
+            if (mongoDbBean == null) {
+                log.info("Warming up the mongo database");
+                mongoDbBean =  new MongoDbTestContainer();
+                beanFactory.initializeBean(mongoDbBean, MongoDbTestContainer.class.getName().toLowerCase());
+                beanFactory.registerSingleton(MongoDbTestContainer.class.getName().toLowerCase(), mongoDbBean);
+                ((DefaultSingletonBeanRegistry)beanFactory).registerDisposableBean(MongoDbTestContainer.class.getName().toLowerCase(), mongoDbBean);
+            }
+            testValues = testValues.and("spring.data.mongodb.uri=" + mongoDbBean.getMongoDBContainer().getReplicaSetUrl());
+        }
+<%_ } _%>
+<%_ if (databaseTypeCassandra) { _%>
+        EmbeddedCassandra cassandraAnnotation = AnnotatedElementUtils.findMergedAnnotation(testClass, EmbeddedCassandra.class);
+        if (null != cassandraAnnotation) {
+            log.debug("detected the EmbeddedCassandra annotation on class {}", testClass.getName());
+            if (cassandraBean == null) {
+                log.info("Warming up the cassandra database");
+                cassandraBean =  new CassandraTestContainer();
+                beanFactory.initializeBean(cassandraBean, CassandraTestContainer.class.getName().toLowerCase());
+                beanFactory.registerSingleton(CassandraTestContainer.class.getName().toLowerCase(), cassandraBean);
+                ((DefaultSingletonBeanRegistry)beanFactory).registerDisposableBean(CassandraTestContainer.class.getName().toLowerCase(), cassandraBean);
+            }
+            testValues = testValues.and("spring.data.cassandra.port=" +  cassandraBean.getCassandraContainer().getMappedPort(CassandraContainer.CQL_PORT))
+                    .and("spring.data.cassandra.contact-points=" +  cassandraBean.getCassandraContainer().getHost())
+                    .and("spring.data.cassandra.keyspace-name=" + CQLDataLoader.DEFAULT_KEYSPACE_NAME)
+                    .and("spring.data.cassandra.local-datacenter=" + cassandraBean.getCassandraContainer().getCluster().getMetadata().getAllHosts().iterator().next().getDatacenter())
+                    .and("spring.data.cassandra.cluster-name=" + cassandraBean.getCassandraContainer().getCluster().getMetadata().getClusterName());
+        }
+<%_ } _%>
+
+<%_ if (messageBrokerKafka) { _%>
+            EmbeddedKafka kafkaAnnotation = AnnotatedElementUtils.findMergedAnnotation(testClass, EmbeddedKafka.class);
+            if (null != kafkaAnnotation) {
+                log.debug("detected the EmbeddedKafka annotation on class {}", testClass.getName());
+                if (kafkaBean == null) {
+                    log.info("Warming up the kafka broker");
+                    kafkaBean = new KafkaTestContainer();
+                    beanFactory.initializeBean(kafkaBean, KafkaTestContainer.class.getName().toLowerCase());
+                    beanFactory.registerSingleton(KafkaTestContainer.class.getName().toLowerCase(), kafkaBean);
+                    ((DefaultSingletonBeanRegistry)beanFactory).registerDisposableBean(KafkaTestContainer.class.getName().toLowerCase(), kafkaBean);
+                }
+                testValues = testValues.and("spring.cloud.stream.kafka.binder.brokers=" + kafkaBean.getKafkaContainer().getHost() + ':' + kafkaBean.getKafkaContainer().getMappedPort(KafkaContainer.KAFKA_PORT));
+            }
+<%_ } _%>
+            testValues.applyTo(context);
+        };
+    }
+}

package/generators/server/templates/src/test/java/package/cucumber/CucumberIT.java.ejs

@@ -18,18 +18,11 @@
 -%>
 package <%= packageName %>.cucumber;
 
-<%_ if (databaseTypeCassandra) { _%>
-import <%= packageName %>.AbstractCassandraTest;
-<%_ } _%>
-<%_ if (databaseTypeNeo4j) { _%>
-import <%= packageName %>.AbstractNeo4jIT;
-<%_ } _%>
+import <%= packageName %>.IntegrationTest;
 import io.cucumber.junit.platform.engine.Cucumber;
 
 @Cucumber
-<%_ if (databaseTypeNeo4j) { _%>
-@ExtendWith(AbstractNeo4jIT)
-<%_ } _%>
-class CucumberIT <% if (databaseTypeCassandra) { %>extends AbstractCassandraTest<% } %> {
+@IntegrationTest
+class CucumberIT {
 
 }

package/generators/server/templates/src/test/java/package/cucumber/stepdefs/StepDefs.java.ejs

@@ -18,10 +18,18 @@
 -%>
 package <%= packageName %>.cucumber.stepdefs;
 
+<%_ if (reactive) { _%>
+import org.springframework.test.web.reactive.server.WebTestClient;
+<%_ } else { _%>
 import org.springframework.test.web.servlet.ResultActions;
+<%_ } _%>
 
 public abstract class StepDefs {
 
+<%_ if (reactive) { _%>
+    protected WebTestClient.ResponseSpec actions;
+<%_ } else { _%>
     protected ResultActions actions;
+<%_ } _%>
 
 }

package/generators/server/templates/src/test/java/package/cucumber/stepdefs/UserStepDefs.java.ejs

@@ -40,6 +40,7 @@ import <%= packageName %>.security.AuthoritiesConstants;
 
 <%_ if (reactive) { _%>
 import org.springframework.test.web.reactive.server.WebTestClient;
+import static org.hamcrest.Matchers.is;
 <%_ } else { _%>
 import org.springframework.test.web.servlet.setup.MockMvcBuilders;
 import org.springframework.test.web.servlet.MockMvc;
@@ -72,29 +73,42 @@ public class UserStepDefs extends StepDefs {
         SecurityContext context = SecurityContextHolder.createEmptyContext();
         context.setAuthentication(authentication);
         SecurityContextHolder.setContext(context);
+<%_ if (reactive) { _%>
+        this.userResourceMock = WebTestClient.bindToController(userResource).build();
+<%_ } else { _%>
         this.userResourceMock = MockMvcBuilders.standaloneSetup(userResource).build();
+<%_ } _%>
     }
 
     @When("I search user {string}")
     public void i_search_user(String userId) throws Throwable {
 <%_ if (reactive) { _%>
-        actions = userResourceMock.get().uri("/api/admin/users/" + userId)
+        actions = userResourceMock.get().uri("/api/admin/users/" + userId).accept(MediaType.APPLICATION_JSON).exchange();
 <%_ } else { _%>
-        actions = userResourceMock.perform(get("/api/admin/users/" + userId)
+        actions = userResourceMock.perform(get("/api/admin/users/" + userId).accept(MediaType.APPLICATION_JSON));
 <%_ } _%>
-                    .accept(MediaType.APPLICATION_JSON)<% if (!reactive) { %>)<%_ } _%>;
     }
 
     @Then("the user is found")
     public void the_user_is_found() throws Throwable {
+<%_ if (reactive) { _%>
+        actions
+            .expectStatus().isOk()
+            .expectHeader().contentTypeCompatibleWith(MediaType.APPLICATION_JSON_VALUE);
+<%_ } else { _%>
         actions
             .andExpect(status().isOk())
             .andExpect(content().contentType(MediaType.APPLICATION_JSON_VALUE));
+<%_ } _%>
     }
 
     @Then("his last name is {string}")
     public void his_last_name_is(String lastName) throws Throwable {
+<%_ if (reactive) { _%>
+        actions.expectBody().jsonPath("$.lastName").value(is(lastName));
+<%_ } else { _%>
         actions.andExpect(jsonPath("$.lastName").value(lastName));
+<%_ } _%>
     }
 
 }

package/generators/server/templates/src/test/java/package/security/DomainUserDetailsServiceIT.java.ejs

@@ -18,9 +18,6 @@
 -%>
 package <%= packageName %>.security;
 
-<%_ if (databaseTypeCassandra) { _%>
-import <%= packageName %>.AbstractCassandraTest;
-<%_ } _%>
 import <%= packageName %>.IntegrationTest;
 <%_ if (databaseTypeSql && reactive) { _%>
 import <%= packageName %>.config.Constants;
@@ -64,7 +61,7 @@ import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
 @WithMockUser("test-user-one")
 <%_ } _%>
 @IntegrationTest
-class DomainUserDetailsServiceIT <% if (databaseTypeCassandra) { %>extends AbstractCassandraTest <% } %>{
+class DomainUserDetailsServiceIT {
 
     private static final String USER_ONE_LOGIN = "test-user-one";
     private static final String USER_ONE_EMAIL = "test-user-one@localhost";

package/generators/server/templates/src/test/java/package/security/SecurityUtilsUnitTest_reactive.java.ejs

@@ -1,5 +1,5 @@
 <%#
- Copyright 2013-2021 the original author or authors from the JHipster project.
+ Copyright 2013-2022 the original author or authors from the JHipster project.
 
  This file is part of the JHipster project, see https://www.jhipster.tech/
  for more information.

package/generators/server/templates/src/test/java/package/security/jwt/JWTFilterTest.java.ejs

@@ -93,7 +93,7 @@ class JWTFilterTest {
         MockServerWebExchange exchange = MockServerWebExchange.from(request);
         jwtFilter.filter(
             exchange,
-            it -> Mono.subscriberContext()
+            it -> Mono.deferContextual(Mono::just)
                 .flatMap(c -> ReactiveSecurityContextHolder.getContext())
                 .map(SecurityContext::getAuthentication)
                 .doOnSuccess(auth -> assertThat(auth.getName()).isEqualTo("test-user"))
@@ -123,7 +123,7 @@ class JWTFilterTest {
         MockServerWebExchange exchange = MockServerWebExchange.from(request);
         jwtFilter.filter(
             exchange,
-            it -> Mono.subscriberContext()
+            it -> Mono.deferContextual(Mono::just)
                 .flatMap(c -> ReactiveSecurityContextHolder.getContext())
                 .map(SecurityContext::getAuthentication)
                 .doOnSuccess(auth -> assertThat(auth).isNull())
@@ -149,7 +149,7 @@ class JWTFilterTest {
         MockServerWebExchange exchange = MockServerWebExchange.from(request);
         jwtFilter.filter(
             exchange,
-            it -> Mono.subscriberContext()
+            it -> Mono.deferContextual(Mono::just)
                 .flatMap(c -> ReactiveSecurityContextHolder.getContext())
                 .map(SecurityContext::getAuthentication)
                 .doOnSuccess(auth -> assertThat(auth).isNull())
@@ -175,7 +175,7 @@ class JWTFilterTest {
         MockServerWebExchange exchange = MockServerWebExchange.from(request);
         jwtFilter.filter(
             exchange,
-            it -> Mono.subscriberContext()
+            it -> Mono.deferContextual(Mono::just)
                 .flatMap(c -> ReactiveSecurityContextHolder.getContext())
                 .map(SecurityContext::getAuthentication)
                 .doOnSuccess(auth -> assertThat(auth).isNull())
@@ -208,7 +208,7 @@ class JWTFilterTest {
         MockServerWebExchange exchange = MockServerWebExchange.from(request);
         jwtFilter.filter(
             exchange,
-            it -> Mono.subscriberContext()
+            it -> Mono.deferContextual(Mono::just)
                 .flatMap(c -> ReactiveSecurityContextHolder.getContext())
                 .map(SecurityContext::getAuthentication)
                 .doOnSuccess(auth -> assertThat(auth).isNull())

package/generators/server/templates/src/test/java/package/security/oauth2/CustomClaimConverterIT.java.ejs

@@ -1,5 +1,5 @@
 <%#
- Copyright 2013-2020 the original author or authors from the JHipster project.
+ Copyright 2013-2022 the original author or authors from the JHipster project.
 
  This file is part of the JHipster project, see https://jhipster.github.io/
  for more information.

package/generators/server/templates/src/test/java/package/service/MailServiceIT.java.ejs

@@ -19,8 +19,6 @@
 package <%= packageName %>.service;
 
 import <%= packageName %>.config.Constants;
-<% if (databaseTypeCassandra) { %>
-import <%= packageName %>.AbstractCassandraTest;<% } %>
 import <%= packageName %>.IntegrationTest;
 import <%= packageName %>.domain.<%= asEntity('User') %>;
 import tech.jhipster.config.JHipsterProperties;
@@ -59,7 +57,7 @@ import static org.mockito.Mockito.*;
  * Integration tests for {@link MailService}.
  */
 @IntegrationTest
-class MailServiceIT <% if (databaseTypeCassandra) { %>extends AbstractCassandraTest <% } %>{
+class MailServiceIT {
 
     private static final String[] languages = {
         // jhipster-needle-i18n-language-constant - JHipster will add/remove languages in this array
@@ -154,9 +152,9 @@ class MailServiceIT <% if (databaseTypeCassandra) { %>extends AbstractCassandraT
     @Test
     void testSendEmailFromTemplate() throws Exception {
         <%= asEntity('User') %> user = new <%= asEntity('User') %>();
+        user.setLangKey(Constants.DEFAULT_LANGUAGE);
         user.setLogin("john");
         user.setEmail("john.doe@example.com");
-        user.setLangKey("en");
         mailService.sendEmailFromTemplate(user, "mail/testEmail", "email.test.title");
         verify(javaMailSender).send(messageCaptor.capture());
         MimeMessage message = messageCaptor.getValue();

package/generators/server/templates/src/test/java/package/service/UserServiceIT.java.ejs

@@ -18,9 +18,6 @@
 -%>
 package <%= packageName %>.service;
 
-<%_ if (databaseTypeCassandra) { _%>
-import <%= packageName %>.AbstractCassandraTest;
-<%_ } _%>
 import <%= packageName %>.IntegrationTest;
 import <%= packageName %>.config.Constants;
 <%_ if ((databaseTypeSql || databaseTypeMongodb || databaseTypeNeo4j || databaseTypeCouchbase) && authenticationTypeSession && !reactive) { _%>
@@ -135,7 +132,7 @@ import static org.mockito.Mockito.when;
 <%_ if (databaseTypeSql && !reactive) { _%>
 @Transactional
 <%_ } _%>
-class UserServiceIT <% if (databaseTypeCassandra) { %>extends AbstractCassandraTest <% } %>{
+class UserServiceIT {
 
     private static final String DEFAULT_LOGIN = "johndoe";
 

package/generators/server/templates/src/test/java/package/service/mapper/UserMapperTest.java.ejs

@@ -44,7 +44,7 @@ class UserMapperTest {
 
     private static final String DEFAULT_LOGIN = "johndoe";
 <%_ if (databaseTypeCouchbase) { _%>
-    private static final String DEFAULT_ID = <%= asEntity('User') %>.PREFIX + DEFAULT_LOGIN;
+    private static final String DEFAULT_ID = DEFAULT_LOGIN;
 <%_ } else { _%>
     private static final <%= user.primaryKey.type %> DEFAULT_ID = <%- getPrimaryKeyValue(user.primaryKey.type) %>;
 <%_ } _%>