From 66cdd0086c0a23fb4f7a452358bca07b881a6446 Mon Sep 17 00:00:00 2001
From: dylan <dylan@dlmw.ch>
Date: Tue, 14 Jan 2025 17:00:29 +0100
Subject: [PATCH] Fix tests

---
 cmd/web/testutils_test.go | 5 +++++
 1 file changed, 5 insertions(+)

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)
+}