refactor(db): migrate `Follow` off `user.go` (#7203)

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())
 }