From 57bbf301c2e2a380ee4c50f951ff1ec712956a77 Mon Sep 17 00:00:00 2001
From: Niranjana Binoy <43930900+NiranjanaBinoy@users.noreply.github.com>
Date: Thu, 26 Sep 2024 22:36:29 -0400
Subject: [PATCH] adding unit test

---
 .../handlers/get-provider-state.test.ts       | 56 +++++++++++++++++++
 .../handlers/get-provider-state.ts            |  4 +-
 2 files changed, 58 insertions(+), 2 deletions(-)
 create mode 100644 app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.test.ts

diff --git a/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.test.ts b/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.test.ts
new file mode 100644
index 000000000000..807a3daf939d
--- /dev/null
+++ b/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();
+  });
+});
diff --git a/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.ts b/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.ts
index c507a17f8c8b..0763913c20cb 100644
--- a/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.ts
+++ b/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>;