adding unit test

app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.test.ts

@@ -0,0 +1,56 @@
+import { PendingJsonRpcResponse } from '@metamask/utils';
+import { JsonRpcEngineEndCallback } from 'json-rpc-engine';
+import getProviderState, {
+  GetProviderState,
+  ProviderStateHandlerResult,
+} from './get-provider-state';
+import { HandlerRequestType } from './types';
+
+describe('getProviderState', () => {
+  let mockEnd: JsonRpcEngineEndCallback;
+  let mockGetProviderState: GetProviderState;
+
+  beforeEach(() => {
+    mockEnd = jest.fn();
+    mockGetProviderState = jest.fn().mockResolvedValue({
+      chainId: '0x539',
+      isUnlocked: true,
+      networkVersion: '',
+      accounts: [],
+    });
+  });
+
+  it('should handle valid input for type ERC721 correctly', async () => {
+    const req: HandlerRequestType = {
+      origin: 'testOrigin',
+      params: [],
+      id: '22',
+      jsonrpc: '2.0',
+      method: 'metamask_getProviderState',
+    };
+
+    const res: PendingJsonRpcResponse<ProviderStateHandlerResult> = {
+      id: '22',
+      jsonrpc: '2.0',
+      result: {
+        chainId: '0x539',
+        isUnlocked: true,
+        networkVersion: '',
+        accounts: [],
+      },
+    };
+
+    await getProviderState.implementation(req, res, jest.fn(), mockEnd, {
+      getProviderState: mockGetProviderState,
+    });
+
+    expect(mockGetProviderState).toHaveBeenCalledWith(req.origin);
+    expect(res.result).toStrictEqual({
+      chainId: '0x539',
+      isUnlocked: true,
+      networkVersion: '',
+      accounts: [],
+    });
+    expect(mockEnd).toHaveBeenCalled();
+  });
+});

app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.ts

@@ -16,14 +16,14 @@ import {
  * @property networkVersion - The current network ID.
  * @property accounts - List of permitted accounts for the specified origin.
  */
-type ProviderStateHandlerResult = {
+export type ProviderStateHandlerResult = {
   chainId: string;
   isUnlocked: boolean;
   networkVersion: string;
   accounts: string[];
 };
 
-type GetProviderState = (
+export type GetProviderState = (
   origin: OriginString,
 ) => Promise<ProviderStateHandlerResult>;