Fix tests

2025-01-14 17:00:29 +01:00
parent c095e9ae0b
commit 66cdd0086c
1 changed files with 5 additions and 0 deletions
--- a/cmd/web/testutils_test.go
+++ b/cmd/web/testutils_test.go
@ -90,3 +90,8 @@ func (v *mockVoteModel) Exists(voterIdentity string, electionID int) (bool, erro
 	args := v.Called(voterIdentity, electionID)
 	return args.Bool(0), args.Error(1)
 }
+
+func (v *mockVoteModel) GetByElection(electionID int) (*[]models.Vote, error) {
+	args := v.Called(electionID)
+	return args.Get(0).(*[]models.Vote), args.Error(2)
+}