update tests

harness · May 15, 2024 · 5797e54 · 5797e54
1 parent 1c93f16
commit 5797e54
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/test/java/io/harness/cf/client/common/SdkCodesTest.java b/src/test/java/io/harness/cf/client/common/SdkCodesTest.java
@@ -2,6 +2,7 @@
 
 import static io.harness.cf.client.common.SdkCodes.*;
 import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
+import static org.mockito.Mockito.when;
 
 import io.harness.cf.client.api.BaseConfig;
 import io.harness.cf.client.dto.Target;
@@ -14,7 +15,8 @@ class SdkCodesTest {
   void testAllLogs() {
     assertDoesNotThrow(
         () -> {
-          BaseConfig baseConfig = Mockito.mock(BaseConfig.class);
+          BaseConfig no6001logs = Mockito.mock(BaseConfig.class);
+          when(no6001logs.isSdkCode6001Suppressed()).thenReturn(true);
 
           errorMissingSdkKey();
           infoPollStarted(123);
@@ -34,11 +36,17 @@ void testAllLogs() {
           warnStreamDisconnected(null);
           warnPostMetricsFailed(null);
           warnPostMetricsFailed("error 3");
-          warnDefaultVariationServed("id1", null, null, baseConfig);
-          warnDefaultVariationServed("id1", null, "defaultVal", baseConfig);
+          warnDefaultVariationServed("id1_6001visible", null, null, Mockito.mock(BaseConfig.class));
+          warnDefaultVariationServed(
+              "id1_6001visible", null, "defaultVal", Mockito.mock(BaseConfig.class));
 
           Target target = Target.builder().identifier("test").isPrivate(false).build();
-          warnDefaultVariationServed("id2", target, "defaultVal2", baseConfig);
+          warnDefaultVariationServed(
+              "id2_6001visible", target, "defaultVal2", Mockito.mock(BaseConfig.class));
+
+          warnDefaultVariationServed("id1_6001notvisible", null, null, no6001logs);
+          warnDefaultVariationServed("id1_6001notvisible", null, "defaultVal", no6001logs);
+          warnDefaultVariationServed("id2_6001notvisible", target, "defaultVal2", no6001logs);
         });
   }
 }