MCPcopy Create free account
hub / github.com/QuantumNous/new-api / openTokenControllerTestDB

Function openTokenControllerTestDB

controller/token_test.go:74–97  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

72}
73
74func openTokenControllerTestDB(t *testing.T) *gorm.DB {
75 t.Helper()
76
77 gin.SetMode(gin.TestMode)
78 common.SetDatabaseTypes(common.DatabaseTypeSQLite, common.DatabaseTypeSQLite)
79 common.RedisEnabled = false
80
81 dsn := fmt.Sprintf("file:%s?mode=memory&cache=shared", strings.ReplaceAll(t.Name(), "/", "_"))
82 db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
83 if err != nil {
84 t.Fatalf("failed to open sqlite db: %v", err)
85 }
86 model.DB = db
87 model.LOG_DB = db
88
89 t.Cleanup(func() {
90 sqlDB, err := db.DB()
91 if err == nil {
92 _ = sqlDB.Close()
93 }
94 })
95
96 return db
97}
98
99func migrateTokenControllerTestDB(t *testing.T, db *gorm.DB) {
100 t.Helper()

Calls 3

SetDatabaseTypesFunction · 0.92
OpenMethod · 0.80
CloseMethod · 0.45

Tested by

no test coverage detected