test(db-driver): unit test validateDbDriver

migrate.go

@@ -45,7 +45,7 @@ func Migrate(args []string) {
 	action := args[ACTION]
 	version := ""
 
-	if !validateDbDriver(driver) {
+	if !ValidateDbDriver(driver) {
 		log.Fatalf("Invalid database driver. Supported databases are: %s\n", DB_DRIVERS.String())
 	}
 	driver = GetSQLDriver(driver)
@@ -190,7 +190,7 @@ func startMigrate(migrationFiles []migrationFile, db *sql.DB, action string) {
 
 var DB_DRIVERS ValidData = []string{"mysql", "postgres", "sqlite3"}
 
-func validateDbDriver(inputDriver string) bool {
+func ValidateDbDriver(inputDriver string) bool {
 	for _, name := range DB_DRIVERS {
 		if name == inputDriver {
 			return true

migrate_test.go

@@ -0,0 +1,27 @@
+package gograte_test
+
+import (
+	"testing"
+
+	"github.com/Shin-Thant/gograte"
+)
+
+func TestValidateDbDriver(t *testing.T) {
+	// validate supported drivers
+	inputDrivers := []string{"sqlite3", "postgres", "mysql"}
+	for _, driver := range inputDrivers {
+		result := gograte.ValidateDbDriver(driver)
+		if !result {
+			t.Errorf("Expected true but got false for driver: %s", driver)
+		}
+	}
+
+	// validate unsupported drivers
+	unsupportedDrivers := []string{"sqlite", "pgx", "mysqlx", "mssql"}
+	for _, driver := range unsupportedDrivers {
+		result := gograte.ValidateDbDriver(driver)
+		if result {
+			t.Errorf("Expected false but got true for driver: %s", driver)
+		}
+	}
+}

status.go

@@ -13,7 +13,7 @@ func Status(args []string) {
 		log.Fatalf("Invalid number of arguments.")
 	}
 
-	if !validateDbDriver(driver) {
+	if !ValidateDbDriver(driver) {
 		log.Fatalf("Invalid database driver. Supported databases are: %s\n", DB_DRIVERS.String())
 	}
 	validatedURL, err := validateDbURL(dbURL)