feat: add tests and make RawArgs more flexible

__tests__/utils/stark.test.ts

@@ -0,0 +1,65 @@
+import fs from 'fs';
+
+import { RawArgs, json, stark } from '../../src';
+import { toBigInt, toHex } from '../../src/utils/number';
+
+const compiledAccount = json.parse(fs.readFileSync('./__mocks__/Account.json').toString('ascii'));
+
+describe('stark', () => {
+  describe('compressProgram()', () => {
+    test('compresses a contract program', () => {
+      const inputProgram = compiledAccount.program;
+
+      const compressed = stark.compressProgram(inputProgram);
+
+      expect(compressed).toMatchSnapshot();
+    });
+    test('works with strings', () => {
+      const inputProgram = json.stringify(compiledAccount.program);
+
+      const compressed = stark.compressProgram(inputProgram);
+
+      expect(compressed).toMatchSnapshot();
+    });
+  });
+
+  describe('compileCalldata() ', () => {
+    test('compiles BigNumberish[] calldata', () => {
+      const callData = ['0x000', 2n, 10000];
+
+      const compiled = stark.compileCalldata(callData);
+
+      expect(compiled).toEqual(['0', '2', '10000']);
+    });
+
+    test('compiles Object type calldata', () => {
+      const callData = {
+        a: '0x000',
+        b: 2n,
+        c: 10000,
+        d: [1n, 2n, '0x3'],
+      };
+
+      const compiled = stark.compileCalldata(callData);
+
+      expect(compiled).toEqual(['0', '2', '10000', '3', '1', '2', '3']);
+    });
+
+    test('compiles Struct type calldata', () => {
+      const calldata: RawArgs = {
+        a: {
+          type: 'struct',
+          x: 1n,
+          y: 2n,
+          z: 3n,
+        },
+        b: 10000000000,
+        c: ['1', '2', toBigInt(3), toHex(4)],
+      };
+
+      const compiled = stark.compileCalldata(calldata);
+
+      expect(compiled).toEqual(['1', '2', '3', '10000000000', '4', '1', '2', '3', '4']);
+    });
+  });
+});

__tests__/utils/utils.test.ts

@@ -1,33 +1,13 @@
-import fs from 'fs';
-
 import { pedersen } from '@noble/curves/stark';
 
-import { constants, hash, json, number, stark } from '../../src';
+import { constants, hash, number, stark } from '../../src';
 import { Block } from '../../src/provider/utils';
 
 const { IS_BROWSER } = constants;
 
-const compiledAccount = json.parse(fs.readFileSync('./__mocks__/Account.json').toString('ascii'));
-
 test('isNode', () => {
   expect(IS_BROWSER).toBe(false);
 });
-describe('compressProgram()', () => {
-  test('compresses a contract program', () => {
-    const inputProgram = compiledAccount.program;
-
-    const compressed = stark.compressProgram(inputProgram);
-
-    expect(compressed).toMatchSnapshot();
-  });
-  test('works with strings', () => {
-    const inputProgram = json.stringify(compiledAccount.program);
-
-    const compressed = stark.compressProgram(inputProgram);
-
-    expect(compressed).toMatchSnapshot();
-  });
-});
 
 describe('hexToDecimalString()', () => {
   test('parse 0xa23', () => {

src/types/lib.ts

@@ -12,7 +12,10 @@ export type RawCalldata = BigNumberish[];
 export type AllowArray<T> = T | T[];
 export type RawArgs =
   | {
-      [inputName: string]: string | string[] | { type: 'struct'; [k: string]: BigNumberish };
+      [inputName: string]:
+        | BigNumberish
+        | BigNumberish[]
+        | { type: 'struct'; [k: string]: BigNumberish };
     }
   | BigNumberish[];