1 files changed, 5 insertions, 0 deletions
diff --git a/internal/webui/root_test.go b/internal/webui/root_test.go
index cd2da0f..945f04b 100644
--- a/internal/webui/root_test.go
+++ b/internal/webui/root_test.go
@@ -12,6 +12,7 @@ import (
 
 type NavitiaMockClient struct {
 	departures []model.Departure
+	trainStops []model.TrainStop
 	err        error
 }
 
@@ -19,6 +20,10 @@ func (c *NavitiaMockClient) GetDepartures(trainStop string) (departures []model.
 	return c.departures, c.err
 }
 
+func (c *NavitiaMockClient) GetTrainStops() (trainStops []model.TrainStop, err error) {
+	return c.trainStops, c.err
+}
+
 func TestRootHandler(t *testing.T) {
 	// test environment setup
 	dbEnv, err := database.InitDB("sqlite3", "file::memory:?_foreign_keys=on")