@@ -20,7 +20,7 @@ import { type EthereumProvider } from "hardhat/types";
2020
2121import { deployFreeForAllSignUpPolicy } from "../ts/deploy" ;
2222import { type IVerifyingKeyStruct } from "../ts/types" ;
23- import { asHex , getDefaultSigner , getBlockTimestamp } from "../ts/utils" ;
23+ import { asHex , getBlockTimestamp , getSigners } from "../ts/utils" ;
2424import {
2525 type Tally ,
2626 type MACI ,
@@ -52,6 +52,7 @@ describe("VoteTally", function test() {
5252 this . timeout ( 900000 ) ; // 15 minutes
5353
5454 let signer : Signer ;
55+ let userSigners : Signer [ ] ;
5556 let maciContract : MACI ;
5657 let pollContract : PollContract ;
5758 let tallyContract : Tally ;
@@ -75,7 +76,8 @@ describe("VoteTally", function test() {
7576
7677 users = [ new Keypair ( ) , new Keypair ( ) ] ;
7778
78- signer = await getDefaultSigner ( ) ;
79+ const signers = await getSigners ( ) ;
80+ [ signer , ...userSigners ] = signers ;
7981
8082 const startTime = await getBlockTimestamp ( signer ) ;
8183
@@ -239,10 +241,9 @@ describe("VoteTally", function test() {
239241 // signup on chain
240242
241243 // eslint-disable-next-line no-await-in-loop
242- await maciContract . signUp (
243- users [ i ] . publicKey . asContractParam ( ) ,
244- AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 1 ] ) ,
245- ) ;
244+ await maciContract
245+ . connect ( userSigners [ i ] )
246+ . signUp ( users [ i ] . publicKey . asContractParam ( ) , AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 1 ] ) ) ;
246247 }
247248
248249 const [ pollPolicyContract ] = await deployFreeForAllSignUpPolicy ( { } , signer , true ) ;
@@ -340,14 +341,16 @@ describe("VoteTally", function test() {
340341
341342 // join on chain
342343 // eslint-disable-next-line no-await-in-loop
343- await pollContract . joinPoll (
344- nullifier ,
345- pollKeys [ i ] . publicKey . asContractParam ( ) ,
346- i ,
347- [ 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 ] ,
348- AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 1 ] ) ,
349- AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 0 ] ) ,
350- ) ;
344+ await pollContract
345+ . connect ( userSigners [ i ] )
346+ . joinPoll (
347+ nullifier ,
348+ pollKeys [ i ] . publicKey . asContractParam ( ) ,
349+ i ,
350+ [ 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 ] ,
351+ AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 1 ] ) ,
352+ AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 0 ] ) ,
353+ ) ;
351354 }
352355
353356 await timeTravel ( signer . provider ! as unknown as EthereumProvider , updatedDuration ) ;
@@ -569,10 +572,9 @@ describe("VoteTally", function test() {
569572 // signup on chain
570573
571574 // eslint-disable-next-line no-await-in-loop
572- await maciContract . signUp (
573- users [ i ] . publicKey . asContractParam ( ) ,
574- AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 1 ] ) ,
575- ) ;
575+ await maciContract
576+ . connect ( userSigners [ i ] )
577+ . signUp ( users [ i ] . publicKey . asContractParam ( ) , AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 1 ] ) ) ;
576578 }
577579
578580 const [ pollPolicyContract ] = await deployFreeForAllSignUpPolicy ( { } , signer , true ) ;
@@ -672,14 +674,16 @@ describe("VoteTally", function test() {
672674
673675 // join on chain
674676 // eslint-disable-next-line no-await-in-loop
675- await pollContract . joinPoll (
676- nullifier ,
677- pollKeys [ i ] . publicKey . asContractParam ( ) ,
678- i ,
679- [ 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 ] ,
680- AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 1 ] ) ,
681- AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 0 ] ) ,
682- ) ;
677+ await pollContract
678+ . connect ( userSigners [ i ] )
679+ . joinPoll (
680+ nullifier ,
681+ pollKeys [ i ] . publicKey . asContractParam ( ) ,
682+ i ,
683+ [ 0 , 0 , 0 , 0 , 0 , 0 , 0 , 0 ] ,
684+ AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 1 ] ) ,
685+ AbiCoder . defaultAbiCoder ( ) . encode ( [ "uint256" ] , [ 0 ] ) ,
686+ ) ;
683687 }
684688
685689 await timeTravel ( signer . provider ! as unknown as EthereumProvider , duration ) ;
0 commit comments