aboutsummaryrefslogtreecommitdiff
path: root/pkg/database
diff options
context:
space:
mode:
Diffstat (limited to 'pkg/database')
-rw-r--r--pkg/database/stop_test.go12
1 files changed, 6 insertions, 6 deletions
diff --git a/pkg/database/stop_test.go b/pkg/database/stop_test.go
index d3db75e..2922cbe 100644
--- a/pkg/database/stop_test.go
+++ b/pkg/database/stop_test.go
@@ -11,7 +11,7 @@ import (
)
func TestCountStops(t *testing.T) {
- trainStops := []model.Stop{
+ stops := []model.Stop{
model.Stop{Id: "id1", Name: "name1"},
model.Stop{Id: "id2", Name: "name2"},
}
@@ -25,14 +25,14 @@ func TestCountStops(t *testing.T) {
// normal check
err = db.Migrate()
require.NoError(t, err)
- err = db.ReplaceAndImportStops(trainStops)
+ err = db.ReplaceAndImportStops(stops)
i, err = db.CountStops()
require.NoError(t, err)
- assert.Equal(t, i, len(trainStops))
+ assert.Equal(t, i, len(stops))
}
func TestGetStop(t *testing.T) {
- trainStops := []model.Stop{
+ stops := []model.Stop{
model.Stop{Id: "id1", Name: "name1"},
model.Stop{Id: "id2", Name: "name2"},
}
@@ -41,11 +41,11 @@ func TestGetStop(t *testing.T) {
require.NoError(t, err)
err = db.Migrate()
require.NoError(t, err)
- err = db.ReplaceAndImportStops(trainStops)
+ err = db.ReplaceAndImportStops(stops)
// normal check
stop, err := db.GetStop("id1")
require.NoError(t, err)
- assert.Equal(t, stop, &trainStops[0])
+ assert.Equal(t, stop, &stops[0])
// error check
stop, err = db.GetStop("non_existent")
require.Error(t, err)