fix: update tests

Instabug · Nov 13, 2024 · 6f590cd · 6f590cd
1 parent 5661424
commit 6f590cd
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/android/src/test/java/com/instabug/reactlibrary/RNInstabugReactnativeModuleTest.java b/android/src/test/java/com/instabug/reactlibrary/RNInstabugReactnativeModuleTest.java
@@ -638,20 +638,20 @@ public void testWillRedirectToStore() {
 
     @Test
     public void testEnableAutoMasking() {
-
+        // when
         rnModule.setAutoMaskingEnabled(true);
 
-        verify(Instabug.class, times(1));
-        Instabug.setNetworkAutoMaskingState(Feature.State.ENABLED);
+        // then
+        mockInstabug.verify(() -> Instabug.setNetworkAutoMaskingState(Feature.State.ENABLED));
     }
-    
+
     @Test
     public void testDisableAutoMasking() {
-
+        // when
         rnModule.setAutoMaskingEnabled(false);
 
-        verify(Instabug.class, times(1));
-        Instabug.setNetworkAutoMaskingState(Feature.State.DISABLED);
+        // then
+        mockInstabug.verify(() -> Instabug.setNetworkAutoMaskingState(Feature.State.DISABLED));
     }
 
 }