fix tests

Author: annie-mac

sdk/cosmos/azure-cosmos-kafka-connect/src/test/java/com/azure/cosmos/kafka/connect/implementation/CosmosClientCacheConfigTest.java

@@ -15,7 +15,7 @@
  */
 public class CosmosClientCacheConfigTest {
 
-    @Test
+    @Test(groups = "unit")
     void shouldBeEqualWhenAllValuesAreSame() {
         CosmosAadAuthConfig authConfig =
             new CosmosAadAuthConfig(
@@ -48,7 +48,7 @@ void shouldBeEqualWhenAllValuesAreSame() {
             .hasSameHashCodeAs(config2);
     }
 
-    @Test
+    @Test(groups = "unit")
     void shouldNotBeEqualWhenValuesAreDifferent() {
         CosmosAadAuthConfig authConfig1 =
             new CosmosAadAuthConfig(
@@ -88,7 +88,7 @@ void shouldNotBeEqualWhenValuesAreDifferent() {
             .doesNotHaveSameHashCodeAs(config2);
     }
 
-    @Test
+    @Test(groups = "unit")
     void shouldSerializeToStringCorrectly() {
         CosmosAadAuthConfig authConfig =
             new CosmosAadAuthConfig(
@@ -118,7 +118,7 @@ void shouldSerializeToStringCorrectly() {
             .contains("context1");
     }
 
-    @Test
+    @Test(groups = "unit")
     void shouldHandleNullValuesCorrectly() {
         CosmosAadAuthConfig authConfig =
             new CosmosAadAuthConfig(
@@ -146,7 +146,7 @@ void shouldHandleNullValuesCorrectly() {
             .contains("||true||");
     }
 
-    @Test
+    @Test(groups = "unit")
     void shouldNotBeEqualToNull() {
         CosmosAadAuthConfig authConfig =
             new CosmosAadAuthConfig(
@@ -168,7 +168,7 @@ void shouldNotBeEqualToNull() {
         assertThat(config).isNotNull();
     }
 
-    @Test
+    @Test(groups = "unit")
     void shouldNotBeEqualToDifferentClass() {
         CosmosAadAuthConfig authConfig =
             new CosmosAadAuthConfig(

sdk/cosmos/azure-cosmos-kafka-connect/src/test/java/com/azure/cosmos/kafka/connect/implementation/CosmosClientCacheMetadataTest.java

@@ -4,38 +4,53 @@
 package com.azure.cosmos.kafka.connect.implementation;
 
 import com.azure.cosmos.CosmosAsyncClient;
-import org.mockito.Mockito;
-import org.testng.annotations.BeforeMethod;
+import com.azure.cosmos.CosmosBridgeInternal;
+import com.azure.cosmos.CosmosClientBuilder;
+import com.azure.cosmos.implementation.RxDocumentClientImpl;
+import com.azure.cosmos.kafka.connect.KafkaCosmosTestConfigurations;
+import com.azure.cosmos.kafka.connect.KafkaCosmosTestSuiteBase;
+import org.testng.annotations.AfterClass;
+import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
 
 import java.time.Instant;
 
 import static org.assertj.core.api.Assertions.assertThat;
-import static org.mockito.Mockito.verify;
 
 /**
  * Test for CosmosClientCacheMetadata
  */
-public class CosmosClientCacheMetadataTest {
-    private CosmosAsyncClient mockClient;
-    private Instant createdTime;
-    private CosmosClientCacheMetadata metadata;
+public class CosmosClientCacheMetadataTest extends KafkaCosmosTestSuiteBase {
+    private CosmosAsyncClient cosmosClient;
 
-    @BeforeMethod
+    @BeforeClass(groups = "kafka-emulator")
     void setUp() {
-        mockClient = Mockito.mock(CosmosAsyncClient.class);
-        createdTime = Instant.now();
-        metadata = new CosmosClientCacheMetadata(mockClient, createdTime);
+        cosmosClient = new CosmosClientBuilder()
+            .endpoint(KafkaCosmosTestConfigurations.HOST)
+            .key(KafkaCosmosTestConfigurations.MASTER_KEY)
+            .buildAsyncClient();
     }
 
-    @Test
+    @AfterClass(groups = "kafka-emulator")
+    void cleanup() {
+        if (this.cosmosClient != null) {
+            this.cosmosClient.close();
+        }
+    }
+
+    @Test(groups = "kafka-emulator")
     void shouldInitializeWithRefCountOne() {
+        Instant createdTime = Instant.now();
+        CosmosClientCacheMetadata metadata = new CosmosClientCacheMetadata(cosmosClient, createdTime);
+
         assertThat(metadata.getRefCount()).isEqualTo(1);
         assertThat(metadata.getLastAccessed()).isEqualTo(createdTime);
     }
 
-    @Test
+    @Test(groups = "kafka-emulator")
     void shouldIncrementRefCount() {
+        CosmosClientCacheMetadata metadata = new CosmosClientCacheMetadata(cosmosClient, Instant.now());
+
         long initialCount = metadata.getRefCount();
         metadata.incrementRefCount();
 
@@ -44,8 +59,10 @@ void shouldIncrementRefCount() {
             .isEqualTo(2);
     }
 
-    @Test
+    @Test(groups = "kafka-emulator")
     void shouldDecrementRefCount() {
+        CosmosClientCacheMetadata metadata = new CosmosClientCacheMetadata(cosmosClient, Instant.now());
+
         metadata.incrementRefCount(); // ref count = 2
         long countBeforeDecrement = metadata.getRefCount();
         metadata.decrementRefCount();
@@ -55,8 +72,10 @@ void shouldDecrementRefCount() {
             .isEqualTo(1);
     }
 
-    @Test
+    @Test(groups = "kafka-emulator")
     void shouldUpdateLastAccessedTime() {
+        CosmosClientCacheMetadata metadata = new CosmosClientCacheMetadata(cosmosClient, Instant.now());
+
         Instant originalLastAccessed = metadata.getLastAccessed();
         // Sleep briefly to ensure time difference
         try {
@@ -72,14 +91,23 @@ void shouldUpdateLastAccessedTime() {
             .isNotEqualTo(originalLastAccessed);
     }
 
-    @Test
+    @Test(groups = "kafka-emulator")
     void shouldReturnClient() {
-        assertThat(metadata.getClient()).isSameAs(mockClient);
+        CosmosClientCacheMetadata metadata = new CosmosClientCacheMetadata(cosmosClient, Instant.now());
+
+        assertThat(metadata.getClient()).isSameAs(cosmosClient);
     }
 
-    @Test
+    @Test(groups = "kafka-emulator")
     void shouldCloseClientOnClose() {
+        CosmosAsyncClient cosmosAsyncClient = new CosmosClientBuilder()
+            .endpoint(KafkaCosmosTestConfigurations.HOST)
+            .key(KafkaCosmosTestConfigurations.MASTER_KEY)
+            .buildAsyncClient();
+        CosmosClientCacheMetadata metadata = new CosmosClientCacheMetadata(cosmosAsyncClient, Instant.now());
         metadata.close();
-        verify(mockClient).close();
+
+        assertThat(((RxDocumentClientImpl)CosmosBridgeInternal.getAsyncDocumentClient(cosmosAsyncClient)).isClosed())
+            .isTrue();
     }
 }