mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-10-31 22:38:58 +00:00
[TESTS] tests.AddFixtures helper loads additional per-test fixtures
(cherry picked from commit93a844dd13
) (cherry picked from commit6d6d1a121c
) (cherry picked from commit8b101f2860
) (cherry picked from commit3e56212d6d
) (cherry picked from commit4f619bc585
) (cherry picked from commit06a47ea56e
) (cherry picked from commit5a4d56e77b
) (cherry picked from commit84b9d3a0c3
) (cherry picked from commit1eb2eca71c
) (cherry picked from commit11d0fe5400
) (cherry picked from commitc93b8b9d3c
) (cherry picked from commit679a7e2efa
) (cherry picked from commite31a3abb7d
) (cherry picked from commit72bedf68a7
) (cherry picked from commitef139ac06f
) (cherry picked from commit134bf83982
) (cherry picked from commitcaf5780c57
)
This commit is contained in:
parent
52e863ba5a
commit
27ee15e86e
3 changed files with 30 additions and 0 deletions
|
@ -7,6 +7,7 @@ package unittest
|
|||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"code.gitea.io/gitea/models/db"
|
||||
|
@ -28,6 +29,16 @@ func GetXORMEngine(engine ...*xorm.Engine) (x *xorm.Engine) {
|
|||
return db.DefaultContext.(*db.Context).Engine().(*xorm.Engine)
|
||||
}
|
||||
|
||||
func OverrideFixtures(opts FixturesOptions, engine ...*xorm.Engine) func() {
|
||||
old := fixturesLoader
|
||||
if err := InitFixtures(opts, engine...); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return func() {
|
||||
fixturesLoader = old
|
||||
}
|
||||
}
|
||||
|
||||
// InitFixtures initialize test fixtures for a test database
|
||||
func InitFixtures(opts FixturesOptions, engine ...*xorm.Engine) (err error) {
|
||||
e := GetXORMEngine(engine...)
|
||||
|
@ -37,6 +48,12 @@ func InitFixtures(opts FixturesOptions, engine ...*xorm.Engine) (err error) {
|
|||
} else {
|
||||
fixtureOptionFiles = testfixtures.Files(opts.Files...)
|
||||
}
|
||||
var fixtureOptionDirs []func(*testfixtures.Loader) error
|
||||
if opts.Dirs != nil {
|
||||
for _, dir := range opts.Dirs {
|
||||
fixtureOptionDirs = append(fixtureOptionDirs, testfixtures.Directory(filepath.Join(opts.Base, dir)))
|
||||
}
|
||||
}
|
||||
dialect := "unknown"
|
||||
switch e.Dialect().URI().DBType {
|
||||
case schemas.POSTGRES:
|
||||
|
@ -57,6 +74,7 @@ func InitFixtures(opts FixturesOptions, engine ...*xorm.Engine) (err error) {
|
|||
testfixtures.DangerousSkipTestDatabaseCheck(),
|
||||
fixtureOptionFiles,
|
||||
}
|
||||
loaderOptions = append(loaderOptions, fixtureOptionDirs...)
|
||||
|
||||
if e.Dialect().URI().DBType == schemas.POSTGRES {
|
||||
loaderOptions = append(loaderOptions, testfixtures.SkipResetSequences())
|
||||
|
|
|
@ -208,6 +208,8 @@ func MainTest(m *testing.M, testOpts ...*TestOptions) {
|
|||
type FixturesOptions struct {
|
||||
Dir string
|
||||
Files []string
|
||||
Dirs []string
|
||||
Base string
|
||||
}
|
||||
|
||||
// CreateTestEngine creates a memory database and loads the fixture data from fixturesDir
|
||||
|
|
|
@ -267,3 +267,13 @@ func PrintCurrentTest(t testing.TB, skip ...int) func() {
|
|||
func Printf(format string, args ...any) {
|
||||
testlogger.Printf(format, args...)
|
||||
}
|
||||
|
||||
func AddFixtures(dirs ...string) func() {
|
||||
return unittest.OverrideFixtures(
|
||||
unittest.FixturesOptions{
|
||||
Dir: filepath.Join(filepath.Dir(setting.AppPath), "models/fixtures/"),
|
||||
Base: filepath.Dir(setting.AppPath),
|
||||
Dirs: dirs,
|
||||
},
|
||||
)
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue