test(stress): add stress test with ceremony params

privacy-scaling-explorations · Jan 12, 2024 · 978dc10 · 978dc10
1 parent dcca295
commit 978dc10
Show file tree

Hide file tree

Showing 6 changed files with 434 additions and 81 deletions.
diff --git a/.github/scripts/ceremony-param-tests-c-witness.sh b/.github/scripts/ceremony-param-tests-c-witness.sh
diff --git a/.github/workflows/nightly-ceremony.yml b/.github/workflows/nightly-ceremony.yml
@@ -1,8 +1,9 @@
 name: Nightly Ceremony
 
 on:
-  schedule:
-    - cron: 0 0 * * *
+  push:
+    branches:
+      - test/stress-test
 
 env:
   STATE_TREE_DEPTH: ${{ vars.STATE_TREE_DEPTH }}
@@ -14,7 +15,7 @@ jobs:
     steps:
       - uses: actions/checkout@v4
         with:
-          ref: dev
+          ref: test/stress-test
       - uses: pnpm/action-setup@v2
         with:
           version: latest
@@ -57,8 +58,8 @@ jobs:
       - name: Download ceremony artifacts
         run: ./.github/scripts/download-ceremony-artifacts.sh
 
-      - name: Run e2e tests
-        run: ./.github/scripts/ceremony-param-tests-c-witness.sh
+      - name: Run stress tests
+        run: pnpm run test:stress
 
       - name: Stop Hardhat
         if: always()

diff --git a/cli/package.json b/cli/package.json
@@ -14,7 +14,7 @@
     "build": "tsc -p tsconfig.build.json",
     "postbuild": "cp package.json ./build",
     "types": "tsc -p tsconfig.json --noEmit",
-    "test": "nyc ts-mocha --exit tests/**/*.test.ts",
+    "test": "nyc ts-mocha --exit tests/e2e/*.test.ts tests/unit/*.test.ts",
     "test:e2e": "ts-mocha --exit tests/e2e.test.ts",
     "test:e2e-subsidy": "ts-mocha --exit tests/e2e.subsidy.test.ts",
     "test:keyChange": "ts-mocha --exit tests/keyChange.test.ts",
@@ -24,7 +24,8 @@
     "test:genKeypair": "ts-mocha --exit tests/unit/genKeyPair.test.ts",
     "test:timeTravel": "ts-mocha --exit tests/unit/timeTravel.test.ts",
     "test:fundWallet": "ts-mocha --exit tests/unit/fundWallet.test.ts",
-    "test:utils": "ts-mocha --exit tests/unit/utils.test.ts"
+    "test:utils": "ts-mocha --exit tests/unit/utils.test.ts",
+    "test:stress": "ts-mocha --exit tests/stress/*.test.ts"
   },
   "dependencies": {
     "@commander-js/extra-typings": "^11.1.0",

diff --git a/cli/tests/constants.ts b/cli/tests/constants.ts
@@ -2,14 +2,22 @@ import { Keypair } from "maci-domainobjs";
 
 import { homedir } from "os";
 
+// config params
 export const STATE_TREE_DEPTH = 10;
 export const INT_STATE_TREE_DEPTH = 1;
 export const MSG_TREE_DEPTH = 2;
 export const VOTE_OPTION_TREE_DEPTH = 2;
 export const MSG_BATCH_DEPTH = 1;
+export const pollDuration = 90;
+export const maxMessages = 25;
+export const maxVoteOptions = 25;
+
+// coordinator keypair
 const coordinatorKeypair = new Keypair();
 export const coordinatorPubKey = coordinatorKeypair.pubKey.serialize();
 export const coordinatorPrivKey = coordinatorKeypair.privKey.serialize();
+
+// local paths to zkeys, wasm, and witnesses
 export const processMessageTestZkeyPath = "./zkeys/ProcessMessages_10-2-1-2_test.0.zkey";
 export const tallyVotesTestZkeyPath = "./zkeys/TallyVotes_10-1-2_test.0.zkey";
 export const subsidyTestZkeyPath = "./zkeys/SubsidyPerBatch_10-1-2_test.0.zkey";
@@ -24,6 +32,9 @@ export const testProcessMessagesWasmPath =
 export const testTallyVotesWasmPath = "./zkeys/TallyVotes_10-1-2_test_js/TallyVotes_10-1-2_test.wasm";
 export const testSubsidyWasmPath = "./zkeys/SubsidyPerBatch_10-1-2_test_js/SubsidyPerBatch_10-1-2_test.wasm";
 export const testRapidsnarkPath = `${homedir()}/rapidsnark/build/prover`;
-export const pollDuration = 90;
-export const maxMessages = 25;
-export const maxVoteOptions = 25;
+export const ceremonyProcessMessageZkeyPath = "./zkeys/processMessages_6-8-2-3.zkey";
+export const ceremonyTallyVotesZkeyPath = "./zkeys/tallyVotes_6-2-3.zkey";
+export const cermeonyProcessMessagesWitnessPath = "./zkeys/processMessages_6-8-2-3";
+export const ceremonyTallyVotesWitnessPath = "./zkeys/tallyVotes_6-2-3";
+export const ceremonyProcessMessagesWasmPath = "./zkeys/processMessages_6-8-2-3.wasm";
+export const ceremonyTallyVotesWasmPath = "./zkeys/tallyVotes_6-2-3.wasm";