add: unit tests to InstabugUtils.spec.ts,

remove: debug console logs

Author: AndrewAminInstabug

src/utils/AppStatesHandler.ts

@@ -6,7 +6,6 @@ let subscription: any = null;
 export const addAppStateListener = (handleAppStateChange: (state: AppStateStatus) => void) => {
   if (!subscription) {
     subscription = AppState.addEventListener('change', handleAppStateChange);
-    console.log('Andrew: AppState listener added');
   }
 };
 
@@ -16,6 +15,5 @@ export const removeAppStateListener = () => {
   if (subscription) {
     subscription.remove();
     subscription = null;
-    console.log('Andrew: AppState listener removed');
   }
 };

test/utils/AppStatesHandler.spec.ts

@@ -0,0 +1,60 @@
+import { AppState } from 'react-native';
+import { addAppStateListener, removeAppStateListener } from '../../src/utils/AppStatesHandler';
+
+jest.mock('react-native', () => ({
+  AppState: {
+    addEventListener: jest.fn(),
+  },
+}));
+
+describe('AppState Listener', () => {
+  const mockHandleAppStateChange = jest.fn();
+  const mockRemove = jest.fn();
+
+  beforeEach(() => {
+    jest.clearAllMocks();
+    (AppState.addEventListener as jest.Mock).mockReturnValue({ remove: mockRemove });
+  });
+
+  afterEach(() => {
+    removeAppStateListener(); // Ensure no leftover subscriptions between tests
+  });
+
+  it('should add an AppState listener if none exists', () => {
+    addAppStateListener(mockHandleAppStateChange);
+
+    expect(AppState.addEventListener).toHaveBeenCalledTimes(1);
+    expect(AppState.addEventListener).toHaveBeenCalledWith('change', mockHandleAppStateChange);
+  });
+
+  it('should not add another listener if one already exists', () => {
+    addAppStateListener(mockHandleAppStateChange);
+    addAppStateListener(mockHandleAppStateChange);
+
+    expect(AppState.addEventListener).toHaveBeenCalledTimes(1); // Only called once
+  });
+
+  it('should remove the AppState listener if one exists', () => {
+    addAppStateListener(mockHandleAppStateChange);
+    removeAppStateListener();
+
+    expect(mockRemove).toHaveBeenCalledTimes(1); // The remove function is called
+  });
+
+  it('should do nothing if removeAppStateListener is called without an existing subscription', () => {
+    removeAppStateListener();
+
+    expect(mockRemove).not.toHaveBeenCalled(); // No remove is called
+  });
+
+  it('should handle multiple add/remove calls properly', () => {
+    addAppStateListener(mockHandleAppStateChange);
+    removeAppStateListener();
+
+    addAppStateListener(mockHandleAppStateChange);
+    removeAppStateListener();
+
+    expect(AppState.addEventListener).toHaveBeenCalledTimes(2); // Listener is added twice
+    expect(mockRemove).toHaveBeenCalledTimes(2); // Listener is removed twice
+  });
+});

test/utils/InstabugUtils.spec.ts

@@ -13,13 +13,21 @@ import InstabugUtils, {
   registerFilteringListener,
   registerObfuscationListener,
   reportNetworkLog,
+  resetNativeObfuscationListener,
   sendCrashReport,
+  updateNetworkLogSnapshot,
 } from '../../src/utils/InstabugUtils';
 
-import { NativeNetworkLogger, NetworkListenerType } from '../../src/native/NativeNetworkLogger';
+import {
+  NativeNetworkLogger,
+  NetworkListenerType,
+  NetworkLoggerEmitter,
+} from '../../src/native/NativeNetworkLogger';
 import { NativeInstabug } from '../../src/native/NativeInstabug';
 import { NativeAPM } from '../../src/native/NativeAPM';
 
+jest.mock('../../src/modules/NetworkLogger');
+
 describe('Test global error handler', () => {
   beforeEach(() => {
     Instabug.init({ token: '', invocationEvents: [InvocationEvent.none] });
@@ -339,15 +347,33 @@ describe('reportNetworkLog', () => {
   });
 });
 
-jest.mock('../../src/modules/NetworkLogger');
-
 describe('test registerNetworkLogsListener usage', () => {
   beforeEach(() => {
     jest.clearAllMocks(); // Clear all mocks before each test
   });
+
+  const network: NetworkLogger.NetworkData = {
+    id: '',
+    url: 'https://api.instabug.com',
+    requestBody: '',
+    requestHeaders: { 'content-type': 'application/json' },
+    method: 'GET',
+    responseBody: '',
+    responseCode: 200,
+    responseHeaders: { 'content-type': 'application/json' },
+    contentType: 'application/json',
+    duration: 0,
+    requestBodySize: 0,
+    responseBodySize: 0,
+    errorDomain: '',
+    errorCode: 0,
+    startTime: 0,
+    serverErrorMessage: '',
+    requestContentType: 'application/json',
+  };
+
   it('registerObfuscationListener should call NetworkLogger.registerNetworkLogsListener() with  NetworkListenerType = NetworkListenerType.obfuscation', () => {
     registerObfuscationListener();
-
     expect(NetworkLogger.registerNetworkLogsListener).toBeCalledTimes(1);
     expect(NetworkLogger.registerNetworkLogsListener).toBeCalledWith(
       NetworkListenerType.obfuscation,
@@ -376,4 +402,35 @@ describe('test registerNetworkLogsListener usage', () => {
       expect.any(Function),
     );
   });
+
+  it('should call NetworkLoggerEmitter.removeAllListeners when call resetNativeObfuscationListener', () => {
+    jest.spyOn(NetworkLoggerEmitter, 'removeAllListeners').mockImplementation();
+    resetNativeObfuscationListener();
+    expect(NetworkLoggerEmitter.removeAllListeners).toBeCalledTimes(1);
+  });
+
+  it('should call NativeNetworkLogger.resetNetworkLogsListener when call resetNativeObfuscationListener on android platform', () => {
+    Platform.OS = 'android';
+    jest.spyOn(NativeNetworkLogger, 'resetNetworkLogsListener').mockImplementation();
+    jest.spyOn(NetworkLoggerEmitter, 'removeAllListeners').mockImplementation();
+    resetNativeObfuscationListener();
+    expect(NativeNetworkLogger.resetNetworkLogsListener).toBeCalledTimes(1);
+    expect(NetworkLoggerEmitter.removeAllListeners).toBeCalledTimes(1);
+  });
+
+  it('should call  NativeNetworkLogger.updateNetworkLogSnapshot when call updateNetworkLogSnapshot with correct parameters', () => {
+    jest.spyOn(NativeNetworkLogger, 'updateNetworkLogSnapshot').mockImplementation();
+
+    updateNetworkLogSnapshot(network);
+    expect(NativeNetworkLogger.updateNetworkLogSnapshot).toBeCalledTimes(1);
+    expect(NativeNetworkLogger.updateNetworkLogSnapshot).toHaveBeenCalledWith(
+      network.url,
+      network.id,
+      network.requestBody,
+      network.responseBody,
+      network.responseCode ?? 200,
+      network.requestHeaders,
+      network.responseHeaders,
+    );
+  });
 });