feat: update add new key test

Author: 99Kies

docs/AddNewKey-Test-Summary.md

@@ -14,6 +14,7 @@
     "test:stateTree": "pnpm run mocha-test tests/state-tree.e2e.test.ts",
     "test:batch-publish": "pnpm run mocha-test tests/batch-publish.e2e.test.ts",
     "test:key-rotation": "pnpm run mocha-test tests/key-rotation.e2e.test.ts",
+    "test:add-new-key": "pnpm run mocha-test tests/add-new-key.e2e.test.ts",
     "generate:vectors": "ts-node poseidon-test/generate-vectors.ts",
     "pretest:poseidon": "pnpm run generate:vectors",
     "test:poseidon": "pnpm run mocha-test tests/poseidon-consistency.e2e.test.ts",

e2e/package.json

@@ -119,6 +119,25 @@ export class AmaciContractClient extends BaseContractClient {
     });
   }
 
+  /**
+   * Pre add new key (uses pre-uploaded deactivate data)
+   */
+  async preAddNewKey(
+    pubkey: { x: string; y: string },
+    nullifier: string,
+    d: [string, string, string, string],
+    proof: { a: string; b: string; c: string }
+  ): Promise<any> {
+    return await this.execute({
+      pre_add_new_key: {
+        pubkey,
+        nullifier,
+        d,
+        groth16_proof: proof
+      }
+    });
+  }
+
   /**
    * Publish message (vote)
    */
@@ -269,6 +288,24 @@ export class AmaciContractClient extends BaseContractClient {
   async getProcessedDMsgCount(): Promise<any> {
     return await this.query({ get_processed_d_msg_count: {} });
   }
+
+  /**
+   * Upload deactivate message (for pre-add-new-key)
+   */
+  async uploadDeactivateMessage(deactivateMessage: string[][]): Promise<any> {
+    return await this.execute({
+      upload_deactivate_message: {
+        deactivate_message: deactivateMessage
+      }
+    });
+  }
+
+  /**
+   * Query: Get deactivate message
+   */
+  async getDeactivateMessage(): Promise<string[][]> {
+    return await this.query({ get_deactivate_message: {} });
+  }
 }
 
 /**