diff --git a/cmd/web/testutils_test.go b/cmd/web/testutils_test.go index 0f1a0d7..9ef1fa6 100644 --- 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) +}