From 3c5e31b25a53268b413bc1e511b7486a2a1c80b9 Mon Sep 17 00:00:00 2001 From: Julien Dessaux Date: Wed, 8 Sep 2021 15:23:50 +0200 Subject: Renamed TrainStop to simply Stop --- pkg/database/train_stop_test.go | 130 ---------------------------------------- 1 file changed, 130 deletions(-) delete mode 100644 pkg/database/train_stop_test.go (limited to 'pkg/database/train_stop_test.go') diff --git a/pkg/database/train_stop_test.go b/pkg/database/train_stop_test.go deleted file mode 100644 index b3f9459..0000000 --- a/pkg/database/train_stop_test.go +++ /dev/null @@ -1,130 +0,0 @@ -package database - -import ( - "reflect" - "testing" - - "git.adyxax.org/adyxax/trains/pkg/model" - "github.com/DATA-DOG/go-sqlmock" - "github.com/stretchr/testify/assert" - "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") - require.NoError(t, err) - err = db.Migrate() - require.NoError(t, err) - // datasets - data1 := []model.TrainStop{ - model.TrainStop{Id: "first", Name: "firstName"}, - model.TrainStop{Id: "second", Name: "secondName"}, - } - data2 := []model.TrainStop{ - model.TrainStop{Id: "first", Name: "firstTest"}, - model.TrainStop{Id: "secondTest", Name: "secondTest"}, - model.TrainStop{Id: "thirdTest", Name: "thirdTest"}, - } - testCases := []struct { - name string - input []model.TrainStop - expectedError interface{} - }{ - {"Normal insert", data1, nil}, - {"Normal insert overwrite", data2, nil}, - } - for _, tc := range testCases { - t.Run(tc.name, func(t *testing.T) { - err := db.ReplaceAndImportTrainStops(tc.input) - if tc.expectedError != nil { - require.Error(t, err) - assert.Equalf(t, reflect.TypeOf(err), reflect.TypeOf(tc.expectedError), "Invalid error type. Got %s but expected %s", reflect.TypeOf(err), reflect.TypeOf(tc.expectedError)) - } else { - require.NoError(t, err) - } - }) - } -} - -func TestReplaceAndImportTrainStopsWithSQLMock(t *testing.T) { - // datasets - data1 := []model.TrainStop{ - model.TrainStop{Id: "first", Name: "firstName"}, - model.TrainStop{Id: "second", Name: "secondName"}, - } - // Transaction begin error - dbBeginError, _, err := sqlmock.New() - if err != nil { - t.Fatalf("an error '%s' was not expected when opening a stub database connection", err) - } - defer dbBeginError.Close() - // Query error cannot delete from - dbCannotDeleteFrom, mockCannotDeleteFrom, err := sqlmock.New() - if err != nil { - t.Fatalf("an error '%s' was not expected when opening a stub database connection", err) - } - defer dbCannotDeleteFrom.Close() - mockCannotDeleteFrom.ExpectBegin() - // Transaction commit error - dbCannotInsertError, mockCannotInsertError, err := sqlmock.New() - if err != nil { - t.Fatalf("an error '%s' was not expected when opening a stub database connection", err) - } - defer dbCannotInsertError.Close() - mockCannotInsertError.ExpectBegin() - mockCannotInsertError.ExpectExec(`DELETE FROM`).WillReturnResult(sqlmock.NewResult(1, 1)) - // Transaction commit error - dbCommitError, mockCommitError, err := sqlmock.New() - if err != nil { - t.Fatalf("an error '%s' was not expected when opening a stub database connection", err) - } - defer dbCommitError.Close() - mockCommitError.ExpectBegin() - mockCommitError.ExpectExec(`DELETE FROM`).WillReturnResult(sqlmock.NewResult(1, 1)) - mockCommitError.ExpectExec(`INSERT INTO`).WillReturnResult(sqlmock.NewResult(1, 1)) - mockCommitError.ExpectExec(`INSERT INTO`).WillReturnResult(sqlmock.NewResult(1, 1)) - // Test cases - testCases := []struct { - name string - db *DBEnv - expectedError interface{} - }{ - {"begin transaction error", &DBEnv{db: dbBeginError}, &TransactionError{}}, - {"query error cannot delete from", &DBEnv{db: dbCannotDeleteFrom}, &QueryError{}}, - {"query error cannot insert into", &DBEnv{db: dbCannotInsertError}, &QueryError{}}, - {"commit transaction error", &DBEnv{db: dbCommitError}, &TransactionError{}}, - } - for _, tc := range testCases { - t.Run(tc.name, func(t *testing.T) { - err := tc.db.ReplaceAndImportTrainStops(data1) - if tc.expectedError != nil { - require.Error(t, err) - assert.Equalf(t, reflect.TypeOf(err), reflect.TypeOf(tc.expectedError), "Invalid error type. Got %s but expected %s", reflect.TypeOf(err), reflect.TypeOf(tc.expectedError)) - } else { - require.NoError(t, err) - } - }) - } -} -- cgit v1.2.3