[test] Update capacity test (#80)

axiom-crypto · Apr 9, 2024 · a1fb2e2 · a1fb2e2
2 parents 107c94a + 805d2d6
commit a1fb2e2
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 25 deletions.
diff --git a/client/test/integration/capacityDataQuery.test.ts b/client/test/integration/capacityDataQuery.test.ts
@@ -39,4 +39,33 @@ describe("On-chain Data Query scenarios", () => {
     const receipt = await axiom.sendQuery();
     expect(receipt.status).toBe('success');
   }, 180000);
+
+  test("256-capacity query", async () => {
+    const { circuit, compiledCircuit, inputs } = await generateCircuit("capacityDataQuery/size256Header");
+
+    const axiom = new Axiom({
+      circuit: circuit,
+      compiledCircuit: compiledCircuit,
+      chainId: "11155111",  // Sepolia
+      provider: process.env.PROVIDER_URI_SEPOLIA as string,
+      privateKey: process.env.PRIVATE_KEY_SEPOLIA as string,
+      callback: {
+        target: "0x4A4e2D8f3fBb3525aD61db7Fc843c9bf097c362e",
+      },
+      options: {
+        overrides: {
+          validateBuild: false,
+        },
+      },
+      capacity: {
+        maxOutputs: 256,
+        maxSubqueries: 256,
+      },
+    });
+    await axiom.init();
+    await axiom.prove(inputs);
+    const receipt = await axiom.sendQuery();
+    // Transaction will be sent successfully but fulfill tx may not succeed
+    expect(receipt.status).toBe('success');
+  }, 180000);
 });
diff --git a/...apacityDataQuery/size129Header.circuit.ts → ...apacityDataQuery/size256Header.circuit.ts b/...apacityDataQuery/size129Header.circuit.ts → ...apacityDataQuery/size256Header.circuit.ts
@@ -4,9 +4,16 @@ export interface CircuitInputs {}
 
 export const defaultInputs = {};
 
+export const config = {
+  capacity: {
+    maxOutputs: 256,
+    maxSubqueries: 256,
+  },
+};
+
 export const circuit = async (inputs: CircuitInputs) => {
   const blockNumber = 5000000;
-  for (let i = 0; i < 129; i++) {
+  for (let i = 0; i < 256; i++) {
     const header = await getHeader(blockNumber + i).receiptsRoot();
     addToCallback(header);
   }

diff --git a/client/test/integration/computeQuery.test.ts b/client/test/integration/computeQuery.test.ts
@@ -52,28 +52,4 @@ describe("Build ComputeQuery with DataQuery", () => {
     expect(args.args[0]).toBe(chainIdOverride);
     expect(args.address).toBe(addressOverride);
   }, 90000);
-
-  test("simple computeQuery with non-default capacity", async () => {
-    const { circuit, compiledCircuit, inputs } = await generateCircuit("computeQuery/simpleWithCapacity");
-
-    const axiom = new Axiom({
-      circuit: circuit,
-      compiledCircuit: compiledCircuit,
-      chainId: "11155111",  // Sepolia
-      provider: process.env.PROVIDER_URI_SEPOLIA as string,
-      privateKey: process.env.PRIVATE_KEY_SEPOLIA as string,
-      callback: {
-        target: "0x4A4e2D8f3fBb3525aD61db7Fc843c9bf097c362e",
-      },
-      capacity: {
-        maxOutputs: 256,
-        maxSubqueries: 256,
-      },
-    });
-    await axiom.init();
-    await axiom.prove(inputs);
-    const receipt = await axiom.sendQuery();
-    // Transaction will be sent successfully but fulfill tx may not succeed
-    expect(receipt.status).toBe('success');
-  }, 90000);
 });