Add tests

Author: FrederikBolding

packages/snaps-rpc-methods/jest.config.js

@@ -10,10 +10,10 @@ module.exports = deepmerge(baseConfig, {
   ],
   coverageThreshold: {
     global: {
-      branches: 95.37,
+      branches: 95.7,
       functions: 98.75,
-      lines: 98.92,
-      statements: 98.62,
+      lines: 98.99,
+      statements: 98.69,
     },
   },
 });

packages/snaps-rpc-methods/src/restricted/manageState.test.ts

@@ -393,6 +393,44 @@ describe('snap_manageState', () => {
         true,
       );
     });
+
+    it('throws an error on update if the new state is too large', async () => {
+      const clearSnapState = jest.fn().mockReturnValueOnce(true);
+      const getSnapState = jest.fn().mockReturnValueOnce(true);
+      const updateSnapState = jest.fn().mockReturnValueOnce(true);
+      const getSnap = jest.fn().mockReturnValue({ preinstalled: false });
+
+      const manageStateImplementation = getManageStateImplementation({
+        clearSnapState,
+        getSnapState,
+        updateSnapState,
+        getUnlockPromise: jest.fn(),
+        getSnap,
+      });
+
+      const newState = {
+        foo: 'foo'.repeat(21_500_000),
+      };
+
+      await expect(
+        manageStateImplementation({
+          context: { origin: 'snap-origin' },
+          method: 'snap_manageState',
+          params: {
+            operation: ManageStateOperation.UpdateState,
+            newState,
+          },
+        }),
+      ).rejects.toThrow(
+        'Invalid snap_manageState "newState" parameter: The new state must not exceed 64 MB in size.',
+      );
+
+      expect(updateSnapState).not.toHaveBeenCalledWith(
+        'snap-origin',
+        newState,
+        true,
+      );
+    });
   });
 
   describe('getValidatedParams', () => {

packages/snaps-rpc-methods/src/restricted/manageState.ts

@@ -234,6 +234,7 @@ export function getManageStateImplementation({
         return null;
       }
 
+      /* istanbul ignore next */
       default:
         throw rpcErrors.invalidParams(
           `Invalid ${method} operation: "${

packages/snaps-utils/src/json.test.ts

@@ -13,4 +13,9 @@ describe('getJsonSizeUnsafe', () => {
     const input = { foo: 'bar' };
     expect(getJsonSizeUnsafe(input)).toBe(13);
   });
+
+  it('calculates the size of the JSON input in bytes', () => {
+    const input = { foo: 'bar€' };
+    expect(getJsonSizeUnsafe(input, true)).toBe(16);
+  });
 });