Fix reports service test

datacite · Apr 5, 2023 · 0df1f03 · 0df1f03
1 parent 1d6ebce
commit 0df1f03
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/internal/app/reports/service_test.go b/internal/app/reports/service_test.go
@@ -4,6 +4,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/datacite/keeshond/internal/app/event"
 	"github.com/datacite/keeshond/internal/app/stats"
 )
 
@@ -92,6 +93,13 @@ func (m *MockStatsService) Aggregate(repoId string, query stats.Query) stats.Agg
 	}
 }
 
+// Mock LastEvent
+func (m *MockStatsService) LastEvent(repoId string) (event.Event, bool) {
+	var e event.Event
+
+	return e, true
+}
+
 // Test that the service can generate a dataset usage report
 func TestGenerateDatasetUsageReport(t *testing.T) {
 	// Create a mock stats service