diff options
Diffstat (limited to 'pkg/database/train_stop_test.go')
| -rw-r--r-- | pkg/database/train_stop_test.go | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/pkg/database/train_stop_test.go b/pkg/database/train_stop_test.go index f593d51..b3f9459 100644 --- a/pkg/database/train_stop_test.go +++ b/pkg/database/train_stop_test.go @@ -10,6 +10,27 @@ import ( "github.com/stretchr/testify/require" ) +func TestCountTrainStops(t *testing.T) { + trainStops := []model.TrainStop{ + model.TrainStop{Id: "id1", Name: "name1"}, + model.TrainStop{Id: "id2", Name: "name2"}, + } + // test db setup + db, err := InitDB("sqlite3", "file::memory:?_foreign_keys=on") + require.NoError(t, err) + // check sql error + i, err := db.CountTrainStops() + require.Error(t, err) + assert.Equalf(t, reflect.TypeOf(err), reflect.TypeOf(&QueryError{}), "Invalid error type. Got %s but expected %s", reflect.TypeOf(err), reflect.TypeOf(&QueryError{})) + // normal check + err = db.Migrate() + require.NoError(t, err) + err = db.ReplaceAndImportTrainStops(trainStops) + i, err = db.CountTrainStops() + require.NoError(t, err) + assert.Equal(t, i, len(trainStops)) +} + func TestReplaceAndImportTrainStops(t *testing.T) { // test db setup db, err := InitDB("sqlite3", "file::memory:?_foreign_keys=on") |