chore: add tests for migrator

server/version/version.go

@@ -29,11 +29,6 @@ func GetMinorVersion(version string) string {
 	return versionList[0] + "." + versionList[1]
 }
 
-func GetSchemaVersion(version string) string {
-	minorVersion := GetMinorVersion(version)
-	return minorVersion + ".0"
-}
-
 // IsVersionGreaterOrEqualThan returns true if version is greater than or equal to target.
 func IsVersionGreaterOrEqualThan(version, target string) bool {
 	return semver.Compare(fmt.Sprintf("v%s", version), fmt.Sprintf("v%s", target)) > -1

store/db/mysql/mysql.go

@@ -39,10 +39,6 @@ func NewDB(profile *profile.Profile) (store.Driver, error) {
 	return &driver, nil
 }
 
-func (*DB) Type() string {
-	return "mysql"
-}
-
 func (d *DB) GetDB() *sql.DB {
 	return d.db
 }

store/db/postgres/postgres.go

@@ -39,10 +39,6 @@ func NewDB(profile *profile.Profile) (store.Driver, error) {
 	return driver, nil
 }
 
-func (*DB) Type() string {
-	return "postgres"
-}
-
 func (d *DB) GetDB() *sql.DB {
 	return d.db
 }

store/db/sqlite/sqlite.go

@@ -50,10 +50,6 @@ func NewDB(profile *profile.Profile) (store.Driver, error) {
 	return &driver, nil
 }
 
-func (*DB) Type() string {
-	return "sqlite"
-}
-
 func (d *DB) GetDB() *sql.DB {
 	return d.db
 }

store/driver.go

@@ -11,10 +11,6 @@ type Driver interface {
 	GetDB() *sql.DB
 	Close() error
 
-	// Type returns the type of the driver.
-	// Supported types are: sqlite, mysql, postgres.
-	Type() string
-
 	// MigrationHistory model related methods.
 	FindMigrationHistoryList(ctx context.Context, find *FindMigrationHistory) ([]*MigrationHistory, error)
 	UpsertMigrationHistory(ctx context.Context, upsert *UpsertMigrationHistory) (*MigrationHistory, error)

test/store/migrator_test.go

+package teststore
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestGetCurrentSchemaVersion(t *testing.T) {
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	currentSchemaVersion, err := ts.GetCurrentSchemaVersion()
+	require.NoError(t, err)
+	require.Equal(t, "0.22.4", currentSchemaVersion)
+}

test/test.go

@@ -32,7 +32,7 @@ func GetTestingProfile(t *testing.T) *profile.Profile {
 
 	// Get a temporary directory for the test data.
 	dir := t.TempDir()
-	mode := "dev"
+	mode := "prod"
 	port := getUnusedPort()
 	driver := getDriverFromEnv()
 	dsn := os.Getenv("DSN")