BE: Chore: Cleanup api tests (#698)

Co-authored-by: Roman Zabaluev <gpg@haarolean.dev>
kafbat · Dec 10, 2024 · 318bcc9 · 318bcc9
1 parent 45324dd
commit 318bcc9
Show file tree

Hide file tree

Showing 27 changed files with 108 additions and 102 deletions.
diff --git a/api/src/test/java/io/kafbat/ui/KafkaConnectServiceTests.java b/api/src/test/java/io/kafbat/ui/KafkaConnectServiceTests.java
@@ -166,7 +166,7 @@ public void shouldReturnNotFoundForNonExistingConnectName() {
 
   @Test
   public void shouldRetrieveConnector() {
-    ConnectorDTO expected = (ConnectorDTO) new ConnectorDTO()
+    ConnectorDTO expected = new ConnectorDTO()
         .connect(connectName)
         .status(new ConnectorStatusDTO()
             .state(ConnectorStateDTO.RUNNING)
@@ -383,7 +383,7 @@ public void shouldValidateAndReturnErrorsOfConnectorPluginConfiguration() {
               .map(ConnectorPluginConfigDTO::getValue)
               .map(ConnectorPluginConfigValueDTO::getErrors)
               .filter(not(List::isEmpty))
-              .findFirst().get();
+              .findFirst().orElseThrow();
           assertEquals(
               "Invalid value 0 for configuration tasks.max: Value must be at least 1",
               error.get(0)

diff --git a/api/src/test/java/io/kafbat/ui/KafkaConsumerTests.java b/api/src/test/java/io/kafbat/ui/KafkaConsumerTests.java
@@ -13,6 +13,7 @@
 import io.kafbat.ui.producer.KafkaTestProducer;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.UUID;
 import java.util.stream.Stream;
 import lombok.extern.slf4j.Slf4j;
@@ -55,15 +56,17 @@ public void shouldDeleteRecords() {
       throw new RuntimeException(e);
     }
 
-    long count = webTestClient.get()
-        .uri("/api/clusters/{clusterName}/topics/{topicName}/messages/v2?mode=EARLIEST", LOCAL, topicName)
-        .accept(TEXT_EVENT_STREAM)
-        .exchange()
-        .expectStatus()
-        .isOk()
-        .expectBodyList(TopicMessageEventDTO.class)
-        .returnResult()
-        .getResponseBody()
+    long count = Objects.requireNonNull(
+        webTestClient.get()
+            .uri("/api/clusters/{clusterName}/topics/{topicName}/messages/v2?mode=EARLIEST", LOCAL, topicName)
+            .accept(TEXT_EVENT_STREAM)
+            .exchange()
+            .expectStatus()
+            .isOk()
+            .expectBodyList(TopicMessageEventDTO.class)
+            .returnResult()
+            .getResponseBody()
+        )
         .stream()
         .filter(e -> e.getType().equals(TopicMessageEventDTO.TypeEnum.MESSAGE))
         .count();
@@ -76,14 +79,16 @@ public void shouldDeleteRecords() {
         .expectStatus()
         .isOk();
 
-    count = webTestClient.get()
-        .uri("/api/clusters/{clusterName}/topics/{topicName}/messages/v2?mode=EARLIEST", LOCAL, topicName)
-        .exchange()
-        .expectStatus()
-        .isOk()
-        .expectBodyList(TopicMessageEventDTO.class)
-        .returnResult()
-        .getResponseBody()
+    count = Objects.requireNonNull(
+        webTestClient.get()
+            .uri("/api/clusters/{clusterName}/topics/{topicName}/messages/v2?mode=EARLIEST", LOCAL, topicName)
+            .exchange()
+            .expectStatus()
+            .isOk()
+            .expectBodyList(TopicMessageEventDTO.class)
+            .returnResult()
+            .getResponseBody()
+        )
         .stream()
         .filter(e -> e.getType().equals(TopicMessageEventDTO.TypeEnum.MESSAGE))
         .count();
@@ -120,7 +125,7 @@ public void shouldIncreasePartitionsUpTo10() {
         .returnResult()
         .getResponseBody();
 
-    assert response != null;
+    Assertions.assertNotNull(response);
     Assertions.assertEquals(10, response.getTotalPartitionsCount());
 
     TopicDetailsDTO topicDetails = webTestClient.get()
@@ -134,7 +139,7 @@ public void shouldIncreasePartitionsUpTo10() {
         .returnResult()
         .getResponseBody();
 
-    assert topicDetails != null;
+    Assertions.assertNotNull(topicDetails);
     Assertions.assertEquals(10, topicDetails.getPartitionCount());
   }
 
@@ -157,8 +162,6 @@ public void shouldReturn404ForNonExistingTopic() {
 
   @Test
   public void shouldReturnConfigsForBroker() {
-    var topicName = UUID.randomUUID().toString();
-
     List<BrokerConfigDTO> configs = webTestClient.get()
         .uri("/api/clusters/{clusterName}/brokers/{id}/configs",
             LOCAL,
@@ -171,7 +174,7 @@ public void shouldReturnConfigsForBroker() {
         .getResponseBody();
 
     Assertions.assertNotNull(configs);
-    assert !configs.isEmpty();
+    Assertions.assertFalse(configs.isEmpty());
     Assertions.assertNotNull(configs.get(0).getName());
     Assertions.assertNotNull(configs.get(0).getIsReadOnly());
     Assertions.assertNotNull(configs.get(0).getIsSensitive());
@@ -216,7 +219,7 @@ public void shouldRetrieveTopicConfig() {
             .getResponseBody();
 
     Assertions.assertNotNull(configs);
-    assert !configs.isEmpty();
+    Assertions.assertFalse(configs.isEmpty());
     Assertions.assertNotNull(configs.get(0).getName());
     Assertions.assertNotNull(configs.get(0).getIsReadOnly());
     Assertions.assertNotNull(configs.get(0).getIsSensitive());

diff --git a/api/src/test/java/io/kafbat/ui/config/auth/azure/AzureEntraLoginCallbackHandlerTest.java b/api/src/test/java/io/kafbat/ui/config/auth/azure/AzureEntraLoginCallbackHandlerTest.java
@@ -17,7 +17,6 @@
 import com.azure.core.credential.AccessToken;
 import com.azure.core.credential.TokenCredential;
 import com.azure.core.credential.TokenRequestContext;
-import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import javax.security.auth.callback.Callback;
@@ -59,9 +58,6 @@ public class AzureEntraLoginCallbackHandlerTest {
   @Mock
   private OAuthBearerTokenCallback oauthBearerTokenCallBack;
 
-  @Mock
-  private OAuthBearerToken oauthBearerToken;
-
   @Mock
   private TokenCredential tokenCredential;
 
@@ -77,12 +73,8 @@ public void beforeEach() {
   }
 
   @Test
-  public void shouldProvideTokenToCallbackWithSuccessfulTokenRequest()
-      throws UnsupportedCallbackException {
-    final Map<String, Object> configs = new HashMap<>();
-    configs.put(
-        "bootstrap.servers",
-        List.of("test-eh.servicebus.windows.net:9093"));
+  public void shouldProvideTokenToCallbackWithSuccessfulTokenRequest() throws UnsupportedCallbackException {
+    Map<String, Object> configs = Map.of("bootstrap.servers", List.of("test-eh.servicebus.windows.net:9093"));
 
     when(tokenCredential.getToken(any(TokenRequestContext.class))).thenReturn(Mono.just(accessToken));
     when(accessToken.getToken()).thenReturn(VALID_SAMPLE_TOKEN);
@@ -114,10 +106,7 @@ public void shouldProvideTokenToCallbackWithSuccessfulTokenRequest()
 
   @Test
   public void shouldProvideErrorToCallbackWithTokenError() throws UnsupportedCallbackException {
-    final Map<String, Object> configs = new HashMap<>();
-    configs.put(
-        "bootstrap.servers",
-        List.of("test-eh.servicebus.windows.net:9093"));
+    Map<String, Object> configs = Map.of("bootstrap.servers", List.of("test-eh.servicebus.windows.net:9093"));
 
     when(tokenCredential.getToken(any(TokenRequestContext.class)))
         .thenThrow(new RuntimeException("failed to acquire token"));
@@ -136,16 +125,13 @@ public void shouldProvideErrorToCallbackWithTokenError() throws UnsupportedCallb
 
   @Test
   public void shouldThrowExceptionWithNullBootstrapServers() {
-    final Map<String, Object> configs = new HashMap<>();
-
     assertThrows(IllegalArgumentException.class, () -> azureEntraLoginCallbackHandler.configure(
-        configs, null, null));
+        Map.of(), null, null));
   }
 
   @Test
   public void shouldThrowExceptionWithMultipleBootstrapServers() {
-    final Map<String, Object> configs = new HashMap<>();
-    configs.put("bootstrap.servers", List.of("server1", "server2"));
+    Map<String, Object> configs = Map.of("bootstrap.servers", List.of("server1", "server2"));
 
     assertThrows(IllegalArgumentException.class, () -> azureEntraLoginCallbackHandler.configure(
         configs, null, null));

diff --git a/api/src/test/java/io/kafbat/ui/config/auth/azure/AzureEntraOAuthBearerTokenTest.java b/api/src/test/java/io/kafbat/ui/config/auth/azure/AzureEntraOAuthBearerTokenTest.java
@@ -39,8 +39,7 @@ public class AzureEntraOAuthBearerTokenTest {
   void constructorShouldParseToken() {
     final AccessToken accessToken = new AccessToken(VALID_SAMPLE_TOKEN, OffsetDateTime.MIN);
 
-    final AzureEntraOAuthBearerToken azureOAuthBearerToken =
-        new AzureEntraOAuthBearerToken(accessToken);
+    final AzureEntraOAuthBearerToken azureOAuthBearerToken = new AzureEntraOAuthBearerToken(accessToken);
 
     assertThat(azureOAuthBearerToken, is(notNullValue()));
     assertThat(azureOAuthBearerToken.value(), is(VALID_SAMPLE_TOKEN));

diff --git a/api/src/test/java/io/kafbat/ui/container/KafkaConnectContainer.java b/api/src/test/java/io/kafbat/ui/container/KafkaConnectContainer.java
@@ -47,6 +47,6 @@ public KafkaConnectContainer withKafka(Network network, String bootstrapServers)
   }
 
   public String getTarget() {
-    return "http://" + getContainerIpAddress() + ":" + getMappedPort(CONNECT_PORT);
+    return "http://" + getHost() + ":" + getMappedPort(CONNECT_PORT);
   }
 }
diff --git a/api/src/test/java/io/kafbat/ui/container/KsqlDbContainer.java b/api/src/test/java/io/kafbat/ui/container/KsqlDbContainer.java
@@ -34,6 +34,6 @@ private KsqlDbContainer withKafka(Network network, String bootstrapServers) {
   }
 
   public String url() {
-    return "http://" + getContainerIpAddress() + ":" + getMappedPort(PORT);
+    return "http://" + getHost() + ":" + getMappedPort(PORT);
   }
 }
diff --git a/api/src/test/java/io/kafbat/ui/container/SchemaRegistryContainer.java b/api/src/test/java/io/kafbat/ui/container/SchemaRegistryContainer.java
@@ -28,7 +28,7 @@ public SchemaRegistryContainer withKafka(Network network, String bootstrapServer
   }
 
   public String getUrl() {
-    return "http://" + getContainerIpAddress() + ":" + getMappedPort(SCHEMA_PORT);
+    return "http://" + getHost() + ":" + getMappedPort(SCHEMA_PORT);
   }
 
   public SchemaRegistryClient schemaRegistryClient() {

diff --git a/api/src/test/java/io/kafbat/ui/emitter/CursorTest.java b/api/src/test/java/io/kafbat/ui/emitter/CursorTest.java
@@ -99,6 +99,7 @@ private Cursor assertCursor(PollingModeDTO expectedMode,
 
     var cursorPosition = registeredCursor.consumerPosition();
     assertThat(cursorPosition).isNotNull();
+    assertThat(cursorPosition.offsets()).isNotNull();
     assertThat(cursorPosition.topic()).isEqualTo(TOPIC);
     assertThat(cursorPosition.partitions()).isEqualTo(List.of());
     assertThat(cursorPosition.pollingMode()).isEqualTo(expectedMode);
@@ -111,6 +112,7 @@ private void waitMgsgEmitted(AbstractEmitter emitter, int expectedMsgsCnt) {
     List<TopicMessageEventDTO> events = Flux.create(emitter)
         .collectList()
         .block();
+    assertThat(events).isNotNull();
     assertThat(events.stream().filter(m -> m.getType() == TopicMessageEventDTO.TypeEnum.MESSAGE).count())
         .isEqualTo(expectedMsgsCnt);
   }

diff --git a/api/src/test/java/io/kafbat/ui/emitter/TailingEmitterTest.java b/api/src/test/java/io/kafbat/ui/emitter/TailingEmitterTest.java
@@ -106,7 +106,7 @@ private Flux<TopicMessageEventDTO> createTailingFlux(
       String query) {
     var cluster = applicationContext.getBean(ClustersStorage.class)
         .getClusterByName(LOCAL)
-        .get();
+        .orElseThrow();
 
     return applicationContext.getBean(MessagesService.class)
         .loadMessages(cluster, topicName,

diff --git a/api/src/test/java/io/kafbat/ui/serdes/SerdesInitializerTest.java b/api/src/test/java/io/kafbat/ui/serdes/SerdesInitializerTest.java
@@ -5,7 +5,6 @@
 import static org.mockito.Mockito.any;
 import static org.mockito.Mockito.anyString;
 import static org.mockito.Mockito.eq;
-import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
@@ -145,6 +144,8 @@ private void verifyAutoConfigured(SerdeInstance serde) {
   }
 
   private void verifyPatternsMatch(ClustersProperties.SerdeConfig config, SerdeInstance serde) {
+    assertThat(serde.topicKeyPattern).isNotNull();
+    assertThat(serde.topicValuePattern).isNotNull();
     assertThat(serde.topicKeyPattern.pattern()).isEqualTo(config.getTopicKeysPattern());
     assertThat(serde.topicValuePattern.pattern()).isEqualTo(config.getTopicValuesPattern());
   }

diff --git a/api/src/test/java/io/kafbat/ui/serdes/builtin/ConsumerOffsetsSerdeTest.java b/api/src/test/java/io/kafbat/ui/serdes/builtin/ConsumerOffsetsSerdeTest.java
@@ -168,6 +168,7 @@ private boolean isGroupMetadataMessage(DeserializeResult key, DeserializeResult
   }
 
   @SneakyThrows
+  @SuppressWarnings("unchecked")
   private Map<String, Object> toMapFromJsom(DeserializeResult result) {
     return new JsonMapper().readValue(result.getResult(), Map.class);
   }

diff --git a/api/src/test/java/io/kafbat/ui/serdes/builtin/ProtobufFileSerdeTest.java b/api/src/test/java/io/kafbat/ui/serdes/builtin/ProtobufFileSerdeTest.java
@@ -226,7 +226,7 @@ void createConfigureFillsDescriptorMappingsWhenProtoFilesListProvided() throws E
     }
 
     @Test
-    void createConfigureFillsDescriptorMappingsWhenProtoFileDirProvided() throws Exception {
+    void createConfigureFillsDescriptorMappingsWhenProtoFileDirProvided() {
       PropertyResolver resolver = mock(PropertyResolver.class);
       when(resolver.getProperty("protobufFilesDir", String.class))
           .thenReturn(Optional.of(protoFilesDir()));

diff --git a/api/src/test/java/io/kafbat/ui/service/BrokerServiceTest.java b/api/src/test/java/io/kafbat/ui/service/BrokerServiceTest.java
@@ -15,7 +15,7 @@ class BrokerServiceTest extends AbstractIntegrationTest {
 
   @Test
   void getBrokersReturnsFilledBrokerDto() {
-    var localCluster = clustersStorage.getClusterByName(LOCAL).get();
+    var localCluster = clustersStorage.getClusterByName(LOCAL).orElseThrow();
     StepVerifier.create(brokerService.getBrokers(localCluster))
         .expectNextMatches(b -> b.getId().equals(1)
             && b.getHost().equals(kafka.getHost())

diff --git a/api/src/test/java/io/kafbat/ui/service/ConfigTest.java b/api/src/test/java/io/kafbat/ui/service/ConfigTest.java
@@ -89,6 +89,8 @@ private Optional<BrokerConfigDTO> getConfig(String name) {
         .returnResult()
         .getResponseBody();
 
+    assertThat(configs).isNotNull();
+
     return configs.stream()
         .filter(c -> c.getName().equals(name))
         .findAny();

diff --git a/api/src/test/java/io/kafbat/ui/service/LogDirsTest.java b/api/src/test/java/io/kafbat/ui/service/LogDirsTest.java
@@ -38,7 +38,7 @@ public void testAllBrokers() {
 
     BrokerTopicLogdirsDTO topic = dir.getTopics().stream()
         .filter(t -> t.getName().equals("__consumer_offsets"))
-        .findAny().get();
+        .findAny().orElseThrow();
 
     assertThat(topic.getPartitions()).hasSize(1);
     assertThat(topic.getPartitions().get(0).getBroker()).isEqualTo(1);
@@ -63,7 +63,7 @@ public void testOneBrokers() {
 
     BrokerTopicLogdirsDTO topic = dir.getTopics().stream()
         .filter(t -> t.getName().equals("__consumer_offsets"))
-        .findAny().get();
+        .findAny().orElseThrow();
 
     assertThat(topic.getPartitions()).hasSize(1);
     assertThat(topic.getPartitions().get(0).getBroker()).isEqualTo(1);
@@ -99,6 +99,7 @@ public void testChangeDirToWrongDir() {
         .returnResult()
         .getResponseBody();
 
+    assertThat(dirs).isNotNull();
     assertThat(dirs.getMessage())
         .isEqualTo(new LogDirNotFoundApiException().getMessage());
 
@@ -116,6 +117,7 @@ public void testChangeDirToWrongDir() {
         .returnResult()
         .getResponseBody();
 
+    assertThat(dirs).isNotNull();
     assertThat(dirs.getMessage())
         .isEqualTo(new TopicOrPartitionNotFoundException().getMessage());
   }

diff --git a/api/src/test/java/io/kafbat/ui/service/MessagesServiceTest.java b/api/src/test/java/io/kafbat/ui/service/MessagesServiceTest.java
@@ -10,7 +10,6 @@
 import io.kafbat.ui.model.KafkaCluster;
 import io.kafbat.ui.model.PollingModeDTO;
 import io.kafbat.ui.model.SmartFilterTestExecutionDTO;
-import io.kafbat.ui.model.SmartFilterTestExecutionResultDTO;
 import io.kafbat.ui.model.TopicMessageDTO;
 import io.kafbat.ui.model.TopicMessageEventDTO;
 import io.kafbat.ui.producer.KafkaTestProducer;
@@ -48,7 +47,7 @@ void init() {
     cluster = applicationContext
         .getBean(ClustersStorage.class)
         .getClusterByName(LOCAL)
-        .get();
+        .orElseThrow();
   }
 
   @AfterEach

diff --git a/api/src/test/java/io/kafbat/ui/service/OffsetsResetServiceTest.java b/api/src/test/java/io/kafbat/ui/service/OffsetsResetServiceTest.java
@@ -45,7 +45,7 @@ public class OffsetsResetServiceTest extends AbstractIntegrationTest {
 
   @BeforeEach
   void init() {
-    cluster = applicationContext.getBean(ClustersStorage.class).getClusterByName(LOCAL).get();
+    cluster = applicationContext.getBean(ClustersStorage.class).getClusterByName(LOCAL).orElseThrow();
     offsetsResetService = new OffsetsResetService(applicationContext.getBean(AdminClientService.class));
     createTopic(new NewTopic(topic, PARTITIONS, (short) 1));
     createConsumerGroup();