generator-jhipster 7.5.0 → 7.8.0

package/generators/server/templates/src/test/java/package/web/rest/AccountResourceIT_oauth2.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.
@@ -20,21 +20,18 @@ package <%= packageName %>.web.rest;
 
 import static <%= packageName %>.test.util.OAuth2TestUtil.TEST_USER_LOGIN;
 import static <%= packageName %>.test.util.OAuth2TestUtil.registerAuthenticationToken;
-<%_ if (!reactive) { _%>
+<%_ if (reactive) { _%>
+import static <%= packageName %>.test.util.OAuth2TestUtil.authenticationToken;
+import static org.springframework.security.test.web.reactive.server.SecurityMockServerConfigurers.*;
+<%_ } else { _%>
 import static <%= packageName %>.test.util.OAuth2TestUtil.testAuthenticationToken;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
-<%_ } else { _%>
-import static <%= packageName %>.test.util.OAuth2TestUtil.authenticationToken;
-import static org.springframework.security.test.web.reactive.server.SecurityMockServerConfigurers.*;
   <%_ if (searchEngineElasticsearch) { _%>
 import static org.mockito.Mockito.*;
   <%_ } _%>
 <%_ } _%>
 
-<%_ if (databaseTypeCassandra) { _%>
-import <%= packageName %>.AbstractCassandraTest;
-<%_ } _%>
 import <%= packageName %>.IntegrationTest;
 import <%= packageName %>.security.AuthoritiesConstants;
 
@@ -42,17 +39,7 @@ import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.MediaType;
 import org.springframework.security.test.context.support.WithMockUser;
-<%_ if (!reactive) { _%>
-import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
-import org.springframework.security.oauth2.client.OAuth2AuthorizedClientService;
-import org.springframework.security.oauth2.client.registration.ClientRegistration;
-import org.springframework.security.test.context.TestSecurityContextHolder;
-import org.springframework.test.web.servlet.MockMvc;
-  <%_ if (databaseTypeSql && !reactive) { _%>
-import org.springframework.transaction.annotation.Transactional;
-  <%_ } _%>
-
-<%_ } else { _%>
+<%_ if (reactive) { _%>
 import org.junit.jupiter.api.BeforeEach;
 import org.springframework.boot.test.autoconfigure.web.reactive.AutoConfigureWebTestClient;
 import org.springframework.security.oauth2.client.ReactiveOAuth2AuthorizedClientService;
@@ -65,30 +52,31 @@ import reactor.core.publisher.Mono;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
+<%_ } else { _%>
+import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
+import org.springframework.security.oauth2.client.OAuth2AuthorizedClientService;
+import org.springframework.security.oauth2.client.registration.ClientRegistration;
+import org.springframework.security.test.context.TestSecurityContextHolder;
+import org.springframework.test.web.servlet.MockMvc;
+  <%_ if (databaseTypeSql && !reactive) { _%>
+import org.springframework.transaction.annotation.Transactional;
+  <%_ } _%>
+
 <%_ } _%>
 
 /**
  * Integration tests for the {@link AccountResource} REST controller.
  */
-<%_ if (!reactive) { _%>
-@AutoConfigureMockMvc
+<%_ if (reactive) { _%>
+@AutoConfigureWebTestClient(timeout = IntegrationTest.DEFAULT_TIMEOUT)
 <%_ } else { _%>
-@AutoConfigureWebTestClient
+@AutoConfigureMockMvc
 <%_ } _%>
 @WithMockUser(value = TEST_USER_LOGIN)
 @IntegrationTest
-class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassandraTest <% } %>{
-<%_ if (!reactive) { _%>
-    @Autowired
-    private MockMvc restAccountMockMvc;
-
-    @Autowired
-    OAuth2AuthorizedClientService authorizedClientService;
-
-    @Autowired
-    ClientRegistration clientRegistration;
-<%_ } else { _%>
+class AccountResourceIT {
 
+<%_ if (reactive) { _%>
     private Map<String, Object> claims;
 
     @Autowired
@@ -99,6 +87,15 @@ class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassan
 
     @Autowired
     private ClientRegistration clientRegistration;
+<%_ } else { _%>
+    @Autowired
+    private MockMvc restAccountMockMvc;
+
+    @Autowired
+    OAuth2AuthorizedClientService authorizedClientService;
+
+    @Autowired
+    ClientRegistration clientRegistration;
 <%_ } _%>
 
 <%_ if (reactive) { _%>
@@ -116,19 +113,7 @@ class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassan
     @Transactional
 <%_ } _%>
     void testGetExistingAccount() <% if (!reactive) { %>throws Exception <% } %>{
-<%_ if (!reactive) { _%>
-        TestSecurityContextHolder
-            .getContext()
-            .setAuthentication(registerAuthenticationToken(authorizedClientService, clientRegistration, testAuthenticationToken()));
-
-        restAccountMockMvc.perform(get("/api/account")
-            .accept(MediaType.APPLICATION_JSON))
-            .andExpect(status().isOk())
-            .andExpect(content().contentType(MediaType.APPLICATION_JSON_VALUE))
-            .andExpect(jsonPath("$.login").value(TEST_USER_LOGIN))
-            .andExpect(jsonPath("$.email").value("john.doe@jhipster.com"))
-            .andExpect(jsonPath("$.authorities").value(AuthoritiesConstants.ADMIN));
-<%_ } else { _%>
+<%_ if (reactive) { _%>
         webTestClient
             .mutateWith(mockAuthentication(registerAuthenticationToken(authorizedClientService, clientRegistration, authenticationToken(claims))))
             .mutateWith(csrf())
@@ -141,44 +126,62 @@ class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassan
             .jsonPath("$.login").isEqualTo("jane")
             .jsonPath("$.email").isEqualTo("jane.doe@jhipster.com")
             .jsonPath("$.authorities").isEqualTo(AuthoritiesConstants.ADMIN);
+<%_ } else { _%>
+        TestSecurityContextHolder
+            .getContext()
+            .setAuthentication(registerAuthenticationToken(authorizedClientService, clientRegistration, testAuthenticationToken()));
+
+        restAccountMockMvc.perform(get("/api/account")
+            .accept(MediaType.APPLICATION_JSON))
+            .andExpect(status().isOk())
+            .andExpect(content().contentType(MediaType.APPLICATION_JSON_VALUE))
+            .andExpect(jsonPath("$.login").value(TEST_USER_LOGIN))
+            .andExpect(jsonPath("$.email").value("john.doe@jhipster.com"))
+            .andExpect(jsonPath("$.authorities").value(AuthoritiesConstants.ADMIN));
 <%_ } _%>
     }
 
     @Test
     void testGetUnknownAccount() <% if (!reactive) { %>throws Exception <% } %>{
-<%_ if (!reactive) { _%>
-        restAccountMockMvc.perform(get("/api/account")
-            .accept(MediaType.APPLICATION_JSON))
-            .andExpect(status().isInternalServerError());
-<%_ } else { _%>
+<%_ if (reactive) { _%>
         webTestClient.get().uri("/api/account")
             .accept(MediaType.APPLICATION_JSON)
             .exchange()
             .expectStatus().is5xxServerError();
+<%_ } else { _%>
+        restAccountMockMvc.perform(get("/api/account")
+            .accept(MediaType.APPLICATION_JSON))
+            .andExpect(status().isInternalServerError());
 <%_ } _%>
     }
 
     @Test
     @WithUnauthenticatedMockUser
-<%_ if (!reactive) { _%>
     void testNonAuthenticatedUser() <% if (!reactive) { %>throws Exception <% } %>{
-        restAccountMockMvc.perform(get("/api/authenticate")
-            .accept(MediaType.APPLICATION_JSON))
-            .andExpect(status().isOk())
-            .andExpect(content().string(""));
-<%_ } else { _%>
-    void testNonAuthenticatedUser() {
+<%_ if (reactive) { _%>
         webTestClient.get().uri("/api/authenticate")
             .accept(MediaType.APPLICATION_JSON)
             .exchange()
             .expectStatus().isOk()
             .expectBody().isEmpty();
+<%_ } else { _%>
+        restAccountMockMvc.perform(get("/api/authenticate")
+            .accept(MediaType.APPLICATION_JSON))
+            .andExpect(status().isOk())
+            .andExpect(content().string(""));
 <%_ } _%>
     }
 
     @Test
-<%_ if (!reactive) { _%>
     void testAuthenticatedUser() <% if (!reactive) { %>throws Exception <% } %>{
+<%_ if (reactive) { _%>
+        webTestClient
+            .get().uri("/api/authenticate")
+            .accept(MediaType.APPLICATION_JSON)
+            .exchange()
+            .expectStatus().isOk()
+            .expectBody(String.class).isEqualTo(TEST_USER_LOGIN);
+<%_ } else { _%>
         restAccountMockMvc.perform(get("/api/authenticate")
             .with(request -> {
                 request.setRemoteUser(TEST_USER_LOGIN);
@@ -187,14 +190,6 @@ class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassan
             .accept(MediaType.APPLICATION_JSON))
             .andExpect(status().isOk())
             .andExpect(content().string(TEST_USER_LOGIN));
-<%_ } else { _%>
-    void testAuthenticatedUser() {
-        webTestClient
-            .get().uri("/api/authenticate")
-            .accept(MediaType.APPLICATION_JSON)
-            .exchange()
-            .expectStatus().isOk()
-            .expectBody(String.class).isEqualTo(TEST_USER_LOGIN);
 <%_ } _%>
     }
 }

package/generators/server/templates/src/test/java/package/web/rest/AccountResourceIT_skipUserManagement.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.
@@ -18,9 +18,6 @@
 -%>
 package <%= packageName %>.web.rest;
 
-<%_ if (databaseTypeCassandra) { _%>
-import <%= packageName %>.AbstractCassandraTest;
-<%_ } _%>
 import <%= packageName %>.IntegrationTest;
 <%_ if (reactive && searchEngineElasticsearch) { _%>
 import <%= packageName %>.repository.search.UserSearchRepository;
@@ -31,10 +28,10 @@ import org.junit.jupiter.api.Test;
 import <%= packageName %>.ReactiveSqlTestContainerExtension;
 <%_ } _%>
 import org.springframework.beans.factory.annotation.Autowired;
-<%_ if (!reactive) { _%>
-import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
-<%_ } else { _%>
+<%_ if (reactive) { _%>
 import org.springframework.boot.test.autoconfigure.web.reactive.AutoConfigureWebTestClient;
+<%_ } else { _%>
+import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
 <%_ } _%>
 import org.springframework.http.MediaType;
 import org.springframework.security.test.context.support.WithMockUser;
@@ -56,21 +53,21 @@ import static <%= packageName %>.web.rest.AccountResourceIT.TEST_USER_LOGIN;
 /**
  * Integration tests for the {@link AccountResource} REST controller.
  */
-<%_ if (!reactive) { _%>
-@AutoConfigureMockMvc
+<%_ if (reactive) { _%>
+@AutoConfigureWebTestClient(timeout = IntegrationTest.DEFAULT_TIMEOUT)
 <%_ } else { _%>
-@AutoConfigureWebTestClient
+@AutoConfigureMockMvc
 <%_ } _%>
 @WithMockUser(value = TEST_USER_LOGIN)
 @IntegrationTest
-class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassandraTest <% } %>{
+class AccountResourceIT {
     static final String TEST_USER_LOGIN = "test";
 
     @Autowired
-<%_ if (!reactive) { _%>
-    private MockMvc restAccountMockMvc;
-<%_ } else { _%>
+<%_ if (reactive) { _%>
     private WebTestClient accountWebTestClient;
+<%_ } else { _%>
+    private MockMvc restAccountMockMvc;
 <%_ } _%>
 <%_ if (searchEngineElasticsearch && reactive) { _%>
 
@@ -81,14 +78,7 @@ class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassan
     @Test
     @WithMockUser(username = TEST_USER_LOGIN, authorities = AuthoritiesConstants.ADMIN)
     void testGetExistingAccount() <% if (!reactive) { %>throws Exception <% } %>{
-<%_ if (!reactive) { _%>
-        restAccountMockMvc.perform(get("/api/account")
-            .accept(MediaType.APPLICATION_JSON))
-            .andExpect(status().isOk())
-            .andExpect(content().contentType(MediaType.APPLICATION_JSON_VALUE))
-            .andExpect(jsonPath("$.login").value(TEST_USER_LOGIN))
-            .andExpect(jsonPath("$.authorities").value(AuthoritiesConstants.ADMIN));
-<%_ } else { _%>
+<%_ if (reactive) { _%>
         accountWebTestClient.get().uri("/api/account")
             .accept(MediaType.APPLICATION_JSON)
             .exchange()
@@ -97,30 +87,43 @@ class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassan
             .expectBody()
             .jsonPath("$.login").isEqualTo(TEST_USER_LOGIN)
             .jsonPath("$.authorities").isEqualTo(AuthoritiesConstants.ADMIN);
+<%_ } else { _%>
+        restAccountMockMvc.perform(get("/api/account")
+            .accept(MediaType.APPLICATION_JSON))
+            .andExpect(status().isOk())
+            .andExpect(content().contentType(MediaType.APPLICATION_JSON_VALUE))
+            .andExpect(jsonPath("$.login").value(TEST_USER_LOGIN))
+            .andExpect(jsonPath("$.authorities").value(AuthoritiesConstants.ADMIN));
 <%_ } _%>
     }
 
     @Test
     @WithUnauthenticatedMockUser
-<%_ if (!reactive) { _%>
     void testNonAuthenticatedUser() <% if (!reactive) { %>throws Exception <% } %>{
-        restAccountMockMvc.perform(get("/api/authenticate")
-            .accept(MediaType.APPLICATION_JSON))
-            .andExpect(status().isOk())
-            .andExpect(content().string(""));
-<%_ } else { _%>
-    void testNonAuthenticatedUser() {
+<%_ if (reactive) { _%>
         accountWebTestClient.get().uri("/api/authenticate")
             .accept(MediaType.APPLICATION_JSON)
             .exchange()
             .expectStatus().isOk()
             .expectBody().isEmpty();
+<%_ } else { _%>
+        restAccountMockMvc.perform(get("/api/authenticate")
+            .accept(MediaType.APPLICATION_JSON))
+            .andExpect(status().isOk())
+            .andExpect(content().string(""));
 <%_ } _%>
     }
 
     @Test
-<%_ if (!reactive) { _%>
     void testAuthenticatedUser() <% if (!reactive) { %>throws Exception <% } %>{
+<%_ if (reactive) { _%>
+        accountWebTestClient
+            .get().uri("/api/authenticate")
+            .accept(MediaType.APPLICATION_JSON)
+            .exchange()
+            .expectStatus().isOk()
+            .expectBody(String.class).isEqualTo(TEST_USER_LOGIN);
+<%_ } else { _%>
         restAccountMockMvc.perform(get("/api/authenticate")
             .with(request -> {
                 request.setRemoteUser(TEST_USER_LOGIN);
@@ -129,14 +132,6 @@ class AccountResourceIT <% if (databaseTypeCassandra) { %>extends AbstractCassan
             .accept(MediaType.APPLICATION_JSON))
             .andExpect(status().isOk())
             .andExpect(content().string(TEST_USER_LOGIN));
-<%_ } else { _%>
-    void testAuthenticatedUser() {
-        accountWebTestClient
-            .get().uri("/api/authenticate")
-            .accept(MediaType.APPLICATION_JSON)
-            .exchange()
-            .expectStatus().isOk()
-            .expectBody(String.class).isEqualTo(TEST_USER_LOGIN);
 <%_ } _%>
     }
 }

package/generators/server/templates/src/test/java/package/web/rest/ClientForwardControllerTest.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/web/rest/KafkaResourceIT.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.
@@ -18,156 +18,85 @@
 -%>
 package <%= packageName %>.web.rest;
 
-import <%= packageName %>.config.KafkaProperties;
-import org.apache.kafka.clients.consumer.ConsumerRecord;
-import org.apache.kafka.clients.consumer.ConsumerRecords;
-import org.apache.kafka.clients.consumer.KafkaConsumer;
-import org.apache.kafka.clients.producer.KafkaProducer;
-import org.apache.kafka.clients.producer.ProducerRecord;
-import org.junit.jupiter.api.BeforeAll;
-import org.junit.jupiter.api.BeforeEach;
-import org.junit.jupiter.api.Test;
-import org.springframework.http.MediaType;
-<%_ if (!reactive) { _%>
-import org.springframework.test.web.servlet.MockMvc;
-import org.springframework.test.web.servlet.MvcResult;
-import org.springframework.test.web.servlet.setup.MockMvcBuilders;
-<%_ } else { _%>
-import org.springframework.test.web.reactive.server.WebTestClient;
-<%_ } _%>
-import org.testcontainers.containers.KafkaContainer;
-import org.testcontainers.utility.DockerImageName;
-
-import java.time.Duration;
-import java.util.Collections;
-import java.util.HashMap;
-import java.util.Map;
-
 import static org.assertj.core.api.Assertions.assertThat;
-<%_ if (!reactive) { _%>
 import static org.junit.jupiter.api.Assertions.fail;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
-<%_ } _%>
 
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.BlockingQueue;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
+import org.springframework.cloud.stream.test.binder.MessageCollector;
+import org.springframework.messaging.Message;
+import org.springframework.messaging.MessageChannel;
+import org.springframework.messaging.MessageHeaders;
+import org.springframework.messaging.support.GenericMessage;
+import org.springframework.security.test.context.support.WithMockUser;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.MvcResult;
+import org.springframework.util.MimeTypeUtils;
+import <%= packageName %>.IntegrationTest;
+import <%= packageName %>.config.EmbeddedKafka;
+import <%= packageName %>.config.KafkaSseConsumer;
+import <%= packageName %>.config.KafkaSseProducer;
+
+@IntegrationTest
+@AutoConfigureMockMvc
+@WithMockUser
+@EmbeddedKafka
 class <%= upperFirstCamelCase(baseName) %>KafkaResourceIT {
 
-    private static boolean started = false;
-    private static KafkaContainer kafkaContainer;
+    @Autowired
+    private MessageCollector collector;
 
-<%_ if (!reactive) { _%>
+    @Autowired
     private MockMvc restMockMvc;
-<%_ } else { _%>
-    private WebTestClient client;
-<%_ } _%>
-
-    @BeforeAll
-    static void startServer() {
-        if (!started) {
-            startTestcontainer();
-            started = true;
-        }
-    }
 
-    private static void startTestcontainer() {
-        // TODO: withNetwork will need to be removed soon
-        // See discussion at https://github.com/jhipster/generator-jhipster/issues/11544#issuecomment-609065206
-        kafkaContainer = new KafkaContainer(DockerImageName.parse("<%= DOCKER_KAFKA %>")).withNetwork(null);
-        kafkaContainer.start();
-    }
-
-    @BeforeEach
-    void setup() {
-        KafkaProperties kafkaProperties = new KafkaProperties();
-        Map<String, String> producerProps = getProducerProps();
-        kafkaProperties.setProducer(new HashMap<>(producerProps));
-
-        Map<String, String> consumerProps = getConsumerProps("default-group");
-        consumerProps.put("client.id", "default-client");
-        kafkaProperties.setConsumer(consumerProps);
-
-        <%= upperFirstCamelCase(baseName) %>KafkaResource kafkaResource = new <%= upperFirstCamelCase(baseName) %>KafkaResource(kafkaProperties);
+    @Autowired
+    @Qualifier(KafkaSseProducer.CHANNELNAME)
+    private MessageChannel output;
 
-<%_ if (!reactive) { _%>
-        restMockMvc = MockMvcBuilders.standaloneSetup(kafkaResource).build();
-<%_ } else { _%>
-        client = WebTestClient.bindToController(kafkaResource).build();
-<%_ } _%>
-    }
+    @Autowired
+    @Qualifier(KafkaSseConsumer.CHANNELNAME)
+    private MessageChannel input;
 
     @Test
-    void producesMessages()<% if (!reactive) { %> throws Exception<% } %> {
-<%_ if (!reactive) { _%>
-        restMockMvc.perform(post("/api/<%= dasherizedBaseName %>-kafka/publish/topic-produce?message=value-produce"))
-            .andExpect(status().isOk())
-            .andExpect(content().contentType(MediaType.APPLICATION_JSON));
-<%_ } else { _%>
-        client.post().uri("/api/<%= dasherizedBaseName %>-kafka/publish/topic-produce?message=value-produce")
-            .exchange()
-            .expectStatus().isOk()
-            .expectHeader().contentType(MediaType.APPLICATION_JSON);
-<%_ } _%>
-
-        Map<String, Object> consumerProps = new HashMap<>(getConsumerProps("group-produce"));
-        KafkaConsumer<String, String> consumer = new KafkaConsumer<>(consumerProps);
-        consumer.subscribe(Collections.singletonList("topic-produce"));
-        ConsumerRecords<String, String> records = consumer.poll(Duration.ofSeconds(1));
-
-        assertThat(records.count()).isEqualTo(1);
-        ConsumerRecord<String, String> record = records.iterator().next();
-        assertThat(record.value()).isEqualTo("value-produce");
+    void producesMessages() throws Exception {
+        restMockMvc
+            .perform(post("/api/<%= dasherizedBaseName %>-kafka/publish?message=value-produce"))
+            .andExpect(status().isOk());
+        BlockingQueue<Message<?>> messages = collector.forChannel(output);
+        GenericMessage<String> payload = (GenericMessage<String>) messages.take();
+        assertThat(payload.getPayload()).isEqualTo("value-produce");
     }
 
     @Test
-    void consumesMessages()<% if (!reactive) { %> throws Exception<% } %> {
-        Map<String, Object> producerProps = new HashMap<>(getProducerProps());
-        KafkaProducer<String, String> producer = new KafkaProducer<>(producerProps);
-
-        producer.send(new ProducerRecord<>("topic-consume", "value-consume"));
-
-<%_ if (!reactive) { _%>
-        MvcResult mvcResult = restMockMvc.perform(get("/api/<%= dasherizedBaseName %>-kafka/consume?topic=topic-consume"))
+    void consumesMessages() throws Exception {
+        Map<String, Object> map = new HashMap<>();
+        map.put(MessageHeaders.CONTENT_TYPE, MimeTypeUtils.TEXT_PLAIN_VALUE);
+        MessageHeaders headers = new MessageHeaders(map);
+        Message<String> testMessage = new GenericMessage<>("value-consume", headers);
+        MvcResult mvcResult = restMockMvc
+            .perform(get("/api/<%= dasherizedBaseName %>-kafka/register"))
             .andExpect(status().isOk())
             .andExpect(request().asyncStarted())
             .andReturn();
-
         for (int i = 0; i < 100; i++) {
+            input.send(testMessage);
             Thread.sleep(100);
             String content = mvcResult.getResponse().getContentAsString();
             if (content.contains("data:value-consume")) {
+                restMockMvc
+                    .perform(get("/api/<%= dasherizedBaseName %>-kafka/unregister"));
                 return;
             }
         }
         fail("Expected content data:value-consume not received");
-<%_ } else { _%>
-        String value = client.get().uri("/api/<%= dasherizedBaseName %>-kafka/consume?topic=topic-consume")
-            .accept(MediaType.TEXT_EVENT_STREAM)
-            .exchange()
-            .expectStatus().isOk()
-            .expectHeader().contentTypeCompatibleWith(MediaType.TEXT_EVENT_STREAM)
-            .returnResult(String.class)
-            .getResponseBody().blockFirst(Duration.ofSeconds(10));
-
-        assertThat(value).isEqualTo("value-consume");
-<%_ } _%>
-    }
-
-    private Map<String, String> getProducerProps() {
-        Map<String, String> producerProps = new HashMap<>();
-        producerProps.put("key.serializer", "org.apache.kafka.common.serialization.StringSerializer");
-        producerProps.put("value.serializer", "org.apache.kafka.common.serialization.StringSerializer");
-        producerProps.put("bootstrap.servers", kafkaContainer.getBootstrapServers());
-        return producerProps;
-    }
-
-    private Map<String, String> getConsumerProps(String group) {
-        Map<String, String> consumerProps = new HashMap<>();
-        consumerProps.put("key.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
-        consumerProps.put("value.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
-        consumerProps.put("bootstrap.servers", kafkaContainer.getBootstrapServers());
-        consumerProps.put("auto.offset.reset", "earliest");
-        consumerProps.put("group.id", group);
-        return consumerProps;
     }
 }