diff options
Diffstat (limited to 'internal/webui/root_test.go')
| -rw-r--r-- | internal/webui/root_test.go | 5 |
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") |