tests() add tests on createInstance rejection

zama-ai · Jul 15, 2023 · 3c53cc8 · 3c53cc8
1 parent 69ab0d7
commit 3c53cc8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/sdk/index.test.ts b/src/sdk/index.test.ts
@@ -26,6 +26,14 @@ describe('token', () => {
     expect(instance.hasKeypair).toBeDefined();
   });
 
+  it('fails to create an instance', async () => {
+    await expect(createInstance({ chainId: BigInt(1234) as any, publicKey: tfhePublicKey })).rejects.toThrow(
+      'chainId must be a number'
+    );
+
+    await expect(createInstance({ chainId: 9000, publicKey: 43 as any })).rejects.toThrow('publicKey must be a string');
+  });
+
   it('creates an instance with keypairs', async () => {
     const keypair = sodium.crypto_box_keypair('hex');
 

diff --git a/src/sdk/index.ts b/src/sdk/index.ts
@@ -43,8 +43,8 @@ export type FhevmInstanceParams = {
 export const createInstance = async (params: FhevmInstanceParams): Promise<FhevmInstance> => {
   await sodium.ready;
   const { chainId, publicKey, keypairs } = params;
-  if (typeof chainId !== 'number') throw new Error('chainId must be a number.');
-  if (typeof publicKey !== 'string') throw new Error('publicKey must be a string.');
+  if (typeof chainId !== 'number') throw new Error('chainId must be a number');
+  if (typeof publicKey !== 'string') throw new Error('publicKey must be a string');
   const buff = fromHexString(publicKey);
   const tfheCompactPublicKey = TfheCompactPublicKey.deserialize(buff);