aboutsummaryrefslogtreecommitdiff
path: root/internal/db/main_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/db/main_test.go')
-rw-r--r--internal/db/main_test.go14
1 files changed, 14 insertions, 0 deletions
diff --git a/internal/db/main_test.go b/internal/db/main_test.go
index cedd7cfc..d5598c58 100644
--- a/internal/db/main_test.go
+++ b/internal/db/main_test.go
@@ -17,6 +17,7 @@ import (
"gorm.io/gorm"
"gorm.io/gorm/logger"
"gorm.io/gorm/schema"
+ _ "modernc.org/sqlite"
log "unknwon.dev/clog/v2"
"gogs.io/gogs/internal/conf"
@@ -127,6 +128,19 @@ func initTestDB(t *testing.T, suite string, tables ...interface{}) *gorm.DB {
db.Exec(fmt.Sprintf(`DROP DATABASE %q`, dbName))
_ = sqlDB.Close()
}
+ case "sqlite":
+ dbName = filepath.Join(os.TempDir(), fmt.Sprintf("gogs-%s-%d.db", suite, time.Now().Unix()))
+ dbOpts = conf.DatabaseOpts{
+ Type: "sqlite",
+ Path: dbName,
+ }
+ cleanup = func(db *gorm.DB) {
+ sqlDB, err := db.DB()
+ if err == nil {
+ _ = sqlDB.Close()
+ }
+ _ = os.Remove(dbName)
+ }
default:
dbName = filepath.Join(os.TempDir(), fmt.Sprintf("gogs-%s-%d.db", suite, time.Now().Unix()))
dbOpts = conf.DatabaseOpts{