1 files changed, 0 insertions, 130 deletions
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)
-			}
-		})
-	}
-}