Add chain id mock tu

massalabs · Dec 22, 2023 · adeba37 · adeba37
1 parent 35ec5e6
commit adeba37
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/assembly/__tests__/vm-mock.spec.ts b/assembly/__tests__/vm-mock.spec.ts
@@ -14,6 +14,7 @@ import {
 import { changeCallStack, resetStorage } from '../vm-mock/storage';
 import {
   mockAdminContext,
+  mockSetChainId,
   setDeployContext,
   setLocalContext,
 } from '../vm-mock/env';
@@ -290,3 +291,18 @@ describe('Testing mocked Context', () => {
     );
   });
 });
+
+describe('Testing mocked Chain id', () => {
+  beforeEach(() => {
+    mockSetChainId(9_000_000);
+  });
+
+  afterAll(() => {
+    mockSetChainId(77);
+  });
+
+  it('chain id mock value', () => {
+    let chainId = env.chainId();
+    expect(env.chainId()).toBe(9_000_000);
+  });
+});
diff --git a/vm-mock/vm.js b/vm-mock/vm.js
@@ -731,7 +731,7 @@ export default function createMockedABI(
 
         const pubKeyBuf = getArrayBuffer(publicKeyPtr);
         if (pubKeyBuf.byteLength !== 64) {
-          console.log('Invalid publickey length. Expected 64 bytes uncompressed secp256k1 public key');
+          console.log('Invalid public key length. Expected 64 bytes uncompressed secp256k1 public key');
           throw new Error();
         }