@maci-protocol/testing 0.0.0-ci.8b07a49 → 0.0.0-ci.8bb6ea8

package/build/ts/__tests__/e2e.nonQv.test.js

@@ -4,61 +4,57 @@ const core_1 = require("@maci-protocol/core");
 const crypto_1 = require("@maci-protocol/crypto");
 const domainobjs_1 = require("@maci-protocol/domainobjs");
 const sdk_1 = require("@maci-protocol/sdk");
+const chai_1 = require("chai");
 const constants_1 = require("../constants");
 const utils_1 = require("../utils");
 /**
  Test scenarios:
     1 signup, 1 message with quadratic voting disabled
     1 signup, 1 relayed message
+    1 signup, 2 valid messages
+    1 signup, 2 valid and 2 invalid messages
+    2 signups, 2 different messages
  */
 describe("e2e tests with non quadratic voting", function test() {
     const useWasm = (0, sdk_1.isArm)();
     this.timeout(900000);
     let maciAddresses;
     let initialVoiceCreditProxyContractAddress;
-    let verifierContractAddress;
     let signer;
-    let vkRegistryAddress;
     const generateProofsArgs = {
         outputDir: constants_1.testProofsDirPath,
         tallyFile: constants_1.testTallyFilePath,
-        tallyZkey: constants_1.tallyVotesTestNonQvZkeyPath,
-        processZkey: constants_1.processMessageTestNonQvZkeyPath,
+        voteTallyZkey: constants_1.testVoteTallyNonQvZkeyPath,
+        messageProcessorZkey: constants_1.testProcessMessageNonQvZkeyPath,
         pollId: 0n,
         rapidsnark: constants_1.testRapidsnarkPath,
-        processWitgen: constants_1.testProcessMessagesNonQvWitnessPath,
-        processDatFile: constants_1.testProcessMessagesNonQvWitnessDatPath,
-        tallyWitgen: constants_1.testTallyVotesNonQvWitnessPath,
-        tallyDatFile: constants_1.testTallyVotesNonQvWitnessDatPath,
-        coordinatorPrivateKey: constants_1.coordinatorPrivKey,
-        processWasm: constants_1.testProcessMessagesNonQvWasmPath,
-        tallyWasm: constants_1.testTallyVotesNonQvWasmPath,
+        messageProcessorWitnessGenerator: constants_1.testProcessMessagesNonQvWitnessPath,
+        messageProcessorWitnessDatFile: constants_1.testProcessMessagesNonQvWitnessDatPath,
+        voteTallyWitnessGenerator: constants_1.testVoteTallyNonQvWitnessPath,
+        voteTallyWitnessDatFile: constants_1.testVoteTallyNonQvWitnessDatPath,
+        coordinatorPrivateKey: constants_1.coordinatorPrivateKey,
+        messageProcessorWasm: constants_1.testProcessMessagesNonQvWasmPath,
+        voteTallyWasm: constants_1.testVoteTallyNonQvWasmPath,
         useWasm,
-        useQuadraticVoting: false,
+        mode: sdk_1.EMode.NON_QV,
     };
-    // before all tests we deploy the vk registry contract and set the verifying keys
+    // before all tests we deploy the verifying keys registry contract and set the verifying keys
     before(async () => {
         signer = await (0, sdk_1.getDefaultSigner)();
-        // we deploy the vk registry contract
-        vkRegistryAddress = await (0, sdk_1.deployVkRegistryContract)({ signer });
-        const [initialVoiceCreditProxy] = await (0, sdk_1.deployConstantInitialVoiceCreditProxy)({ amount: constants_1.DEFAULT_INITIAL_VOICE_CREDITS }, signer, undefined, true);
+        const constantInitialVoiceCreditProxyFactory = await (0, sdk_1.deployConstantInitialVoiceCreditProxyFactory)(signer, true);
+        const initialVoiceCreditProxy = await (0, sdk_1.deployConstantInitialVoiceCreditProxy)({ amount: constants_1.DEFAULT_INITIAL_VOICE_CREDITS }, constantInitialVoiceCreditProxyFactory, signer);
         initialVoiceCreditProxyContractAddress = await initialVoiceCreditProxy.getAddress();
-        const verifier = await (0, sdk_1.deployVerifier)(signer, true);
-        verifierContractAddress = await verifier.getAddress();
-        // we deploy the vk registry contract
-        vkRegistryAddress = await (0, sdk_1.deployVkRegistryContract)({ signer });
-        // we set the verifying keys
-        await (0, sdk_1.setVerifyingKeys)({ ...(await (0, constants_1.verifyingKeysArgs)(signer, sdk_1.EMode.NON_QV)), vkRegistryAddress });
     });
     describe("1 signup, 1 message", () => {
         after(async () => {
             await (0, utils_1.clean)();
         });
         const user = new domainobjs_1.Keypair();
+        const voteWeight = 9n;
         before(async () => {
-            const [signupPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)(signer, true);
+            const [signupPolicy, , signupPolicyFactory, signupCheckerFactory] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({}, signer, true);
             const signupPolicyContractAddress = await signupPolicy.getAddress();
-            const [pollPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)(signer, true);
+            const [pollPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({ policy: signupPolicyFactory, checker: signupCheckerFactory }, signer, true);
             const pollPolicyContractAddress = await pollPolicy.getAddress();
             // deploy the smart contracts
             maciAddresses = await (0, sdk_1.deployMaci)({
@@ -66,6 +62,12 @@ describe("e2e tests with non quadratic voting", function test() {
                 signer,
                 signupPolicyAddress: signupPolicyContractAddress,
             });
+            // we set the verifying keys
+            const { verifyingKeysRegistryContractAddress } = maciAddresses;
+            await (0, sdk_1.setVerifyingKeys)({
+                ...(await (0, constants_1.verifyingKeysArgs)(signer, [sdk_1.EMode.NON_QV])),
+                verifyingKeysRegistryAddress: verifyingKeysRegistryContractAddress,
+            });
             const startDate = await (0, sdk_1.getBlockTimestamp)(signer);
             // deploy a poll contract
             await (0, sdk_1.deployPoll)({
@@ -75,8 +77,6 @@ describe("e2e tests with non quadratic voting", function test() {
                 pollEndTimestamp: startDate + constants_1.pollDuration,
                 relayers: [await signer.getAddress()],
                 maciAddress: maciAddresses.maciContractAddress,
-                verifierContractAddress,
-                vkRegistryContractAddress: vkRegistryAddress,
                 policyContractAddress: pollPolicyContractAddress,
                 initialVoiceCreditProxyContractAddress,
                 mode: sdk_1.EMode.NON_QV,
@@ -85,22 +85,37 @@ describe("e2e tests with non quadratic voting", function test() {
         it("should signup one user", async () => {
             await (0, sdk_1.signup)({
                 maciAddress: maciAddresses.maciContractAddress,
-                maciPubKey: user.pubKey.serialize(),
+                maciPublicKey: user.publicKey.serialize(),
                 sgData: constants_1.DEFAULT_SG_DATA,
                 signer,
             });
         });
+        it("should join one user", async () => {
+            await (0, sdk_1.joinPoll)({
+                maciAddress: maciAddresses.maciContractAddress,
+                privateKey: user.privateKey.serialize(),
+                pollId: 0n,
+                pollJoiningZkey: constants_1.testPollJoiningZkeyPath,
+                useWasm,
+                pollWasm: constants_1.testPollJoiningWasmPath,
+                pollWitnessGenerator: constants_1.testPollJoiningWitnessPath,
+                rapidsnark: constants_1.testRapidsnarkPath,
+                sgDataArg: constants_1.DEFAULT_SG_DATA,
+                ivcpDataArg: constants_1.DEFAULT_IVCP_DATA,
+                signer,
+            });
+        });
         it("should publish one message", async () => {
             await (0, sdk_1.publish)({
-                pubkey: user.pubKey.serialize(),
+                publicKey: user.publicKey.serialize(),
                 stateIndex: 1n,
                 voteOptionIndex: 0n,
                 nonce: 1n,
                 pollId: 0n,
-                newVoteWeight: 9n,
+                newVoteWeight: voteWeight,
                 maciAddress: maciAddresses.maciContractAddress,
-                salt: (0, crypto_1.genRandomSalt)(),
-                privateKey: user.privKey.serialize(),
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: user.privateKey.serialize(),
                 signer,
             });
         });
@@ -111,7 +126,7 @@ describe("e2e tests with non quadratic voting", function test() {
                 ...generateProofsArgs,
                 signer,
                 maciAddress: maciAddresses.maciContractAddress,
-                useQuadraticVoting: false,
+                mode: sdk_1.EMode.NON_QV,
             });
             await (0, sdk_1.proveOnChain)({ ...constants_1.proveOnChainArgs, maciAddress: maciAddresses.maciContractAddress, signer });
             await (0, sdk_1.verify)({
@@ -119,6 +134,10 @@ describe("e2e tests with non quadratic voting", function test() {
                 tallyData: tallyFileData,
                 maciAddress: tallyFileData.maci,
             });
+            (0, chai_1.expect)(tallyFileData.totalSpentVoiceCredits.spent.toString()).to.eq(voteWeight.toString());
+            tallyFileData.results.tally.forEach((result, index) => {
+                (0, chai_1.expect)(result.toString()).to.eq(index === 0 ? voteWeight.toString() : "0");
+            });
         });
     });
     describe("1 signup, 1 relayed message", () => {
@@ -126,10 +145,11 @@ describe("e2e tests with non quadratic voting", function test() {
             await (0, utils_1.clean)();
         });
         const user = new domainobjs_1.Keypair();
+        const voteWeight = 9n;
         before(async () => {
-            const [signupPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)(signer, true);
+            const [signupPolicy, , signupPolicyFactory, signupCheckerFactory] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({}, signer, true);
             const signupPolicyContractAddress = await signupPolicy.getAddress();
-            const [pollPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)(signer, true);
+            const [pollPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({ policy: signupPolicyFactory, checker: signupCheckerFactory }, signer, true);
             const pollPolicyContractAddress = await pollPolicy.getAddress();
             // deploy the smart contracts
             maciAddresses = await (0, sdk_1.deployMaci)({
@@ -137,6 +157,12 @@ describe("e2e tests with non quadratic voting", function test() {
                 signer,
                 signupPolicyAddress: signupPolicyContractAddress,
             });
+            // we set the verifying keys
+            const { verifyingKeysRegistryContractAddress } = maciAddresses;
+            await (0, sdk_1.setVerifyingKeys)({
+                ...(await (0, constants_1.verifyingKeysArgs)(signer, [sdk_1.EMode.NON_QV])),
+                verifyingKeysRegistryAddress: verifyingKeysRegistryContractAddress,
+            });
             const startDate = await (0, sdk_1.getBlockTimestamp)(signer);
             // deploy a poll contract
             await (0, sdk_1.deployPoll)({
@@ -146,8 +172,6 @@ describe("e2e tests with non quadratic voting", function test() {
                 pollEndTimestamp: startDate + constants_1.pollDuration,
                 relayers: [await signer.getAddress()],
                 maciAddress: maciAddresses.maciContractAddress,
-                verifierContractAddress,
-                vkRegistryContractAddress: vkRegistryAddress,
                 policyContractAddress: pollPolicyContractAddress,
                 initialVoiceCreditProxyContractAddress,
                 mode: sdk_1.EMode.NON_QV,
@@ -156,31 +180,46 @@ describe("e2e tests with non quadratic voting", function test() {
         it("should signup one user", async () => {
             await (0, sdk_1.signup)({
                 maciAddress: maciAddresses.maciContractAddress,
-                maciPubKey: user.pubKey.serialize(),
+                maciPublicKey: user.publicKey.serialize(),
                 sgData: constants_1.DEFAULT_SG_DATA,
                 signer,
             });
         });
+        it("should join one user", async () => {
+            await (0, sdk_1.joinPoll)({
+                maciAddress: maciAddresses.maciContractAddress,
+                privateKey: user.privateKey.serialize(),
+                pollId: 0n,
+                pollJoiningZkey: constants_1.testPollJoiningZkeyPath,
+                useWasm,
+                pollWasm: constants_1.testPollJoiningWasmPath,
+                pollWitnessGenerator: constants_1.testPollJoiningWitnessPath,
+                rapidsnark: constants_1.testRapidsnarkPath,
+                sgDataArg: constants_1.DEFAULT_SG_DATA,
+                ivcpDataArg: constants_1.DEFAULT_IVCP_DATA,
+                signer,
+            });
+        });
         it("should relay one message", async () => {
             const { message, ephemeralKeypair } = (0, sdk_1.generateVote)({
                 pollId: 0n,
                 voteOptionIndex: 0n,
-                salt: (0, crypto_1.genRandomSalt)(),
+                salt: (0, crypto_1.generateRandomSalt)(),
                 nonce: 1n,
-                privateKey: user.privKey,
+                privateKey: user.privateKey,
                 stateIndex: 1n,
-                voteWeight: 9n,
-                coordinatorPubKey: constants_1.coordinatorKeypair.pubKey,
+                voteWeight,
+                coordinatorPublicKey: constants_1.coordinatorKeypair.publicKey,
                 maxVoteOption: BigInt(core_1.VOTE_OPTION_TREE_ARITY ** constants_1.deployPollArgs.voteOptionTreeDepth),
-                newPubKey: user.pubKey,
+                newPublicKey: user.publicKey,
             });
             const messages = [
                 {
                     maciAddress: maciAddresses.maciContractAddress,
                     poll: 0,
                     data: message.data.map(String),
-                    publicKey: ephemeralKeypair.pubKey.asArray().map(String),
-                    hash: message.hash(ephemeralKeypair.pubKey).toString(),
+                    publicKey: ephemeralKeypair.publicKey.asArray().map(String),
+                    hash: message.hash(ephemeralKeypair.publicKey).toString(),
                 },
             ];
             await (0, utils_1.relayTestMessages)({ messages, signer, pollId: 0, maciAddress: maciAddresses.maciContractAddress });
@@ -194,7 +233,244 @@ describe("e2e tests with non quadratic voting", function test() {
                 signer,
                 maciAddress: maciAddresses.maciContractAddress,
                 ipfsMessageBackupFiles,
-                useQuadraticVoting: false,
+                mode: sdk_1.EMode.NON_QV,
+            });
+            await (0, sdk_1.proveOnChain)({ ...constants_1.proveOnChainArgs, maciAddress: maciAddresses.maciContractAddress, signer });
+            await (0, sdk_1.verify)({
+                ...(await (0, constants_1.verifyArgs)(signer)),
+                tallyData: tallyFileData,
+                maciAddress: tallyFileData.maci,
+            });
+            (0, chai_1.expect)(tallyFileData.totalSpentVoiceCredits.spent.toString()).to.eq(voteWeight.toString());
+            tallyFileData.results.tally.forEach((result, index) => {
+                (0, chai_1.expect)(result.toString()).to.eq(index === 0 ? voteWeight.toString() : "0");
+            });
+        });
+    });
+    describe("1 signup, 2 valid messages", () => {
+        after(async () => {
+            await (0, utils_1.clean)();
+        });
+        const user = new domainobjs_1.Keypair();
+        before(async () => {
+            const [signupPolicy, , signupPolicyFactory, signupCheckerFactory] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({}, signer, true);
+            const signupPolicyContractAddress = await signupPolicy.getAddress();
+            const [pollPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({ policy: signupPolicyFactory, checker: signupCheckerFactory }, signer, true);
+            const pollPolicyContractAddress = await pollPolicy.getAddress();
+            // deploy the smart contracts
+            maciAddresses = await (0, sdk_1.deployMaci)({
+                ...constants_1.deployArgs,
+                signer,
+                signupPolicyAddress: signupPolicyContractAddress,
+            });
+            // we set the verifying keys
+            const { verifyingKeysRegistryContractAddress } = maciAddresses;
+            await (0, sdk_1.setVerifyingKeys)({
+                ...(await (0, constants_1.verifyingKeysArgs)(signer, [sdk_1.EMode.NON_QV])),
+                verifyingKeysRegistryAddress: verifyingKeysRegistryContractAddress,
+            });
+            const startDate = await (0, sdk_1.getBlockTimestamp)(signer);
+            // deploy a poll contract
+            await (0, sdk_1.deployPoll)({
+                ...constants_1.deployPollArgs,
+                signer,
+                pollStartTimestamp: startDate,
+                pollEndTimestamp: startDate + constants_1.pollDuration,
+                relayers: [await signer.getAddress()],
+                maciAddress: maciAddresses.maciContractAddress,
+                policyContractAddress: pollPolicyContractAddress,
+                initialVoiceCreditProxyContractAddress,
+                mode: sdk_1.EMode.NON_QV,
+            });
+        });
+        it("should signup one user", async () => {
+            await (0, sdk_1.signup)({
+                maciAddress: maciAddresses.maciContractAddress,
+                maciPublicKey: user.publicKey.serialize(),
+                sgData: constants_1.DEFAULT_SG_DATA,
+                signer,
+            });
+        });
+        it("should join one user", async () => {
+            await (0, sdk_1.joinPoll)({
+                maciAddress: maciAddresses.maciContractAddress,
+                privateKey: user.privateKey.serialize(),
+                pollId: 0n,
+                pollJoiningZkey: constants_1.testPollJoiningZkeyPath,
+                useWasm,
+                pollWasm: constants_1.testPollJoiningWasmPath,
+                pollWitnessGenerator: constants_1.testPollJoiningWitnessPath,
+                rapidsnark: constants_1.testRapidsnarkPath,
+                sgDataArg: constants_1.DEFAULT_SG_DATA,
+                ivcpDataArg: constants_1.DEFAULT_IVCP_DATA,
+                signer,
+            });
+        });
+        it("should publish two messages", async () => {
+            await (0, sdk_1.publish)({
+                publicKey: user.publicKey.serialize(),
+                stateIndex: 1n,
+                voteOptionIndex: 0n,
+                nonce: 2n,
+                pollId: 0n,
+                newVoteWeight: BigInt(constants_1.DEFAULT_INITIAL_VOICE_CREDITS),
+                maciAddress: maciAddresses.maciContractAddress,
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: user.privateKey.serialize(),
+                signer,
+            });
+            await (0, sdk_1.publish)({
+                publicKey: user.publicKey.serialize(),
+                stateIndex: 1n,
+                voteOptionIndex: 1n,
+                nonce: 1n,
+                pollId: 0n,
+                newVoteWeight: BigInt(constants_1.DEFAULT_INITIAL_VOICE_CREDITS),
+                maciAddress: maciAddresses.maciContractAddress,
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: user.privateKey.serialize(),
+                signer,
+            });
+        });
+        it("should generate zk-SNARK proofs and verify them", async () => {
+            await (0, sdk_1.timeTravel)({ seconds: constants_1.pollDuration, signer });
+            await (0, sdk_1.mergeSignups)({ ...constants_1.mergeSignupsArgs, maciAddress: maciAddresses.maciContractAddress, signer });
+            const { tallyData: tallyFileData } = await (0, sdk_1.generateProofs)({
+                ...generateProofsArgs,
+                signer,
+                maciAddress: maciAddresses.maciContractAddress,
+                mode: sdk_1.EMode.NON_QV,
+            });
+            await (0, sdk_1.proveOnChain)({ ...constants_1.proveOnChainArgs, maciAddress: maciAddresses.maciContractAddress, signer });
+            await (0, sdk_1.verify)({
+                ...(await (0, constants_1.verifyArgs)(signer)),
+                tallyData: tallyFileData,
+                maciAddress: tallyFileData.maci,
+            });
+            (0, chai_1.expect)(tallyFileData.totalSpentVoiceCredits.spent.toString()).to.eq(constants_1.DEFAULT_INITIAL_VOICE_CREDITS.toString());
+            tallyFileData.results.tally.forEach((result, index) => {
+                (0, chai_1.expect)(result.toString()).to.eq(index === 1 ? constants_1.DEFAULT_INITIAL_VOICE_CREDITS.toString() : "0");
+            });
+        });
+    });
+    describe("1 signup, 2 valid and 1 invalid message", () => {
+        after(async () => {
+            await (0, utils_1.clean)();
+        });
+        const user = new domainobjs_1.Keypair();
+        const voteWeight = 9n;
+        before(async () => {
+            const [signupPolicy, , signupPolicyFactory, signupCheckerFactory] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({}, signer, true);
+            const signupPolicyContractAddress = await signupPolicy.getAddress();
+            const [pollPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({ policy: signupPolicyFactory, checker: signupCheckerFactory }, signer, true);
+            const pollPolicyContractAddress = await pollPolicy.getAddress();
+            // deploy the smart contracts
+            maciAddresses = await (0, sdk_1.deployMaci)({
+                ...constants_1.deployArgs,
+                signer,
+                signupPolicyAddress: signupPolicyContractAddress,
+            });
+            // we set the verifying keys
+            const { verifyingKeysRegistryContractAddress } = maciAddresses;
+            await (0, sdk_1.setVerifyingKeys)({
+                ...(await (0, constants_1.verifyingKeysArgs)(signer, [sdk_1.EMode.NON_QV])),
+                verifyingKeysRegistryAddress: verifyingKeysRegistryContractAddress,
+            });
+            const startDate = await (0, sdk_1.getBlockTimestamp)(signer);
+            // deploy a poll contract
+            await (0, sdk_1.deployPoll)({
+                ...constants_1.deployPollArgs,
+                signer,
+                pollStartTimestamp: startDate,
+                pollEndTimestamp: startDate + constants_1.pollDuration,
+                relayers: [await signer.getAddress()],
+                maciAddress: maciAddresses.maciContractAddress,
+                policyContractAddress: pollPolicyContractAddress,
+                initialVoiceCreditProxyContractAddress,
+                mode: sdk_1.EMode.NON_QV,
+            });
+        });
+        it("should signup one user", async () => {
+            await (0, sdk_1.signup)({
+                maciAddress: maciAddresses.maciContractAddress,
+                maciPublicKey: user.publicKey.serialize(),
+                sgData: constants_1.DEFAULT_SG_DATA,
+                signer,
+            });
+        });
+        it("should join one user", async () => {
+            await (0, sdk_1.joinPoll)({
+                maciAddress: maciAddresses.maciContractAddress,
+                privateKey: user.privateKey.serialize(),
+                pollId: 0n,
+                pollJoiningZkey: constants_1.testPollJoiningZkeyPath,
+                useWasm,
+                pollWasm: constants_1.testPollJoiningWasmPath,
+                pollWitnessGenerator: constants_1.testPollJoiningWitnessPath,
+                rapidsnark: constants_1.testRapidsnarkPath,
+                sgDataArg: constants_1.DEFAULT_SG_DATA,
+                ivcpDataArg: constants_1.DEFAULT_IVCP_DATA,
+                signer,
+            });
+        });
+        it("should publish two valid and two invalid messages", async () => {
+            await (0, sdk_1.publish)({
+                publicKey: user.publicKey.serialize(),
+                stateIndex: 1n,
+                voteOptionIndex: 0n,
+                nonce: 1n,
+                pollId: 0n,
+                newVoteWeight: voteWeight,
+                maciAddress: maciAddresses.maciContractAddress,
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: user.privateKey.serialize(),
+                signer,
+            });
+            await (0, sdk_1.publish)({
+                publicKey: user.publicKey.serialize(),
+                stateIndex: 1n,
+                voteOptionIndex: 1n,
+                nonce: 1n,
+                pollId: 0n,
+                newVoteWeight: BigInt(constants_1.DEFAULT_INITIAL_VOICE_CREDITS + 10),
+                maciAddress: maciAddresses.maciContractAddress,
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: user.privateKey.serialize(),
+                signer,
+            });
+            await (0, sdk_1.publish)({
+                publicKey: user.publicKey.serialize(),
+                stateIndex: 1n,
+                voteOptionIndex: 1n,
+                nonce: 1n,
+                pollId: 0n,
+                newVoteWeight: -voteWeight,
+                maciAddress: maciAddresses.maciContractAddress,
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: user.privateKey.serialize(),
+                signer,
+            });
+            await (0, sdk_1.publish)({
+                publicKey: user.publicKey.serialize(),
+                stateIndex: 1n,
+                voteOptionIndex: 1n,
+                nonce: 1n,
+                pollId: 0n,
+                newVoteWeight: voteWeight,
+                maciAddress: maciAddresses.maciContractAddress,
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: user.privateKey.serialize(),
+                signer,
+            });
+        });
+        it("should generate zk-SNARK proofs and verify them", async () => {
+            await (0, sdk_1.timeTravel)({ seconds: constants_1.pollDuration, signer });
+            await (0, sdk_1.mergeSignups)({ ...constants_1.mergeSignupsArgs, maciAddress: maciAddresses.maciContractAddress, signer });
+            const { tallyData: tallyFileData } = await (0, sdk_1.generateProofs)({
+                ...generateProofsArgs,
+                signer,
+                maciAddress: maciAddresses.maciContractAddress,
+                mode: sdk_1.EMode.NON_QV,
             });
             await (0, sdk_1.proveOnChain)({ ...constants_1.proveOnChainArgs, maciAddress: maciAddresses.maciContractAddress, signer });
             await (0, sdk_1.verify)({
@@ -202,6 +478,127 @@ describe("e2e tests with non quadratic voting", function test() {
                 tallyData: tallyFileData,
                 maciAddress: tallyFileData.maci,
             });
+            (0, chai_1.expect)(tallyFileData.totalSpentVoiceCredits.spent.toString()).to.eq(voteWeight.toString());
+            tallyFileData.results.tally.forEach((result, index) => {
+                (0, chai_1.expect)(result.toString()).to.eq(index === 1 ? voteWeight.toString() : "0");
+            });
+        });
+    });
+    describe("2 signups, 2 different messages", () => {
+        after(async () => {
+            await (0, utils_1.clean)();
+        });
+        const users = [new domainobjs_1.Keypair(), new domainobjs_1.Keypair()];
+        const voteWeight = 9n;
+        before(async () => {
+            const [signupPolicy, , signupPolicyFactory, signupCheckerFactory] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({}, signer, true);
+            const signupPolicyContractAddress = await signupPolicy.getAddress();
+            const [pollPolicy] = await (0, sdk_1.deployFreeForAllSignUpPolicy)({ policy: signupPolicyFactory, checker: signupCheckerFactory }, signer, true);
+            const pollPolicyContractAddress = await pollPolicy.getAddress();
+            // deploy the smart contracts
+            maciAddresses = await (0, sdk_1.deployMaci)({
+                ...constants_1.deployArgs,
+                signer,
+                signupPolicyAddress: signupPolicyContractAddress,
+            });
+            // we set the verifying keys
+            const { verifyingKeysRegistryContractAddress } = maciAddresses;
+            await (0, sdk_1.setVerifyingKeys)({
+                ...(await (0, constants_1.verifyingKeysArgs)(signer, [sdk_1.EMode.NON_QV])),
+                verifyingKeysRegistryAddress: verifyingKeysRegistryContractAddress,
+            });
+            const startDate = await (0, sdk_1.getBlockTimestamp)(signer);
+            // deploy a poll contract
+            await (0, sdk_1.deployPoll)({
+                ...constants_1.deployPollArgs,
+                signer,
+                pollStartTimestamp: startDate,
+                pollEndTimestamp: startDate + constants_1.pollDuration,
+                relayers: [await signer.getAddress()],
+                maciAddress: maciAddresses.maciContractAddress,
+                policyContractAddress: pollPolicyContractAddress,
+                initialVoiceCreditProxyContractAddress,
+                mode: sdk_1.EMode.NON_QV,
+            });
+        });
+        it("should signup two users", async () => {
+            await (0, sdk_1.signup)({
+                maciAddress: maciAddresses.maciContractAddress,
+                maciPublicKey: users[0].publicKey.serialize(),
+                sgData: constants_1.DEFAULT_SG_DATA,
+                signer,
+            });
+            await (0, sdk_1.signup)({
+                maciAddress: maciAddresses.maciContractAddress,
+                maciPublicKey: users[1].publicKey.serialize(),
+                sgData: constants_1.DEFAULT_SG_DATA,
+                signer,
+            });
+        });
+        it("should join two users", async () => {
+            // eslint-disable-next-line @typescript-eslint/prefer-for-of
+            for (let index = 0; index < users.length; index += 1) {
+                // eslint-disable-next-line no-await-in-loop
+                await (0, sdk_1.joinPoll)({
+                    maciAddress: maciAddresses.maciContractAddress,
+                    privateKey: users[index].privateKey.serialize(),
+                    pollId: 0n,
+                    pollJoiningZkey: constants_1.testPollJoiningZkeyPath,
+                    useWasm,
+                    pollWasm: constants_1.testPollJoiningWasmPath,
+                    pollWitnessGenerator: constants_1.testPollJoiningWitnessPath,
+                    rapidsnark: constants_1.testRapidsnarkPath,
+                    sgDataArg: constants_1.DEFAULT_SG_DATA,
+                    ivcpDataArg: constants_1.DEFAULT_IVCP_DATA,
+                    signer,
+                });
+            }
+        });
+        it("should publish two different messages", async () => {
+            await (0, sdk_1.publish)({
+                publicKey: users[0].publicKey.serialize(),
+                stateIndex: 1n,
+                voteOptionIndex: 0n,
+                nonce: 1n,
+                pollId: 0n,
+                newVoteWeight: voteWeight,
+                maciAddress: maciAddresses.maciContractAddress,
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: users[0].privateKey.serialize(),
+                signer,
+            });
+            await (0, sdk_1.publish)({
+                publicKey: users[1].publicKey.serialize(),
+                stateIndex: 2n,
+                voteOptionIndex: 1n,
+                nonce: 1n,
+                pollId: 0n,
+                newVoteWeight: voteWeight,
+                maciAddress: maciAddresses.maciContractAddress,
+                salt: (0, crypto_1.generateRandomSalt)(),
+                privateKey: users[1].privateKey.serialize(),
+                signer,
+            });
+        });
+        it("should generate zk-SNARK proofs and verify them", async () => {
+            await (0, sdk_1.timeTravel)({ seconds: constants_1.pollDuration, signer });
+            await (0, sdk_1.mergeSignups)({ ...constants_1.mergeSignupsArgs, maciAddress: maciAddresses.maciContractAddress, signer });
+            const { tallyData: tallyFileData } = await (0, sdk_1.generateProofs)({
+                ...generateProofsArgs,
+                signer,
+                maciAddress: maciAddresses.maciContractAddress,
+                mode: sdk_1.EMode.NON_QV,
+            });
+            await (0, sdk_1.proveOnChain)({ ...constants_1.proveOnChainArgs, maciAddress: maciAddresses.maciContractAddress, signer });
+            await (0, sdk_1.verify)({
+                ...(await (0, constants_1.verifyArgs)(signer)),
+                tallyData: tallyFileData,
+                maciAddress: tallyFileData.maci,
+            });
+            (0, chai_1.expect)(tallyFileData.totalSpentVoiceCredits.spent.toString()).to.eq((voteWeight * 2n).toString());
+            tallyFileData.results.tally.forEach((result, index) => {
+                (0, chai_1.expect)(result.toString()).to.eq(index < 2 ? voteWeight.toString() : "0");
+            });
         });
     });
 });