diff options
Diffstat (limited to 'internal/db/main_test.go')
-rw-r--r-- | internal/db/main_test.go | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/internal/db/main_test.go b/internal/db/main_test.go index bc55a0d0..4f19da43 100644 --- a/internal/db/main_test.go +++ b/internal/db/main_test.go @@ -15,6 +15,7 @@ import ( _ "modernc.org/sqlite" log "unknwon.dev/clog/v2" + "gogs.io/gogs/internal/conf" "gogs.io/gogs/internal/testutil" ) @@ -37,6 +38,15 @@ func TestMain(m *testing.M) { // NOTE: AutoMigrate does not respect logger passed in gorm.Config. logger.Default = logger.Default.LogMode(level) + switch os.Getenv("GOGS_DATABASE_TYPE") { + case "mysql": + conf.UseMySQL = true + case "postgres": + conf.UsePostgreSQL = true + default: + conf.UseSQLite3 = true + } + os.Exit(m.Run()) } |