forked from vikunja/vikunja
konrad
d28f005552
Fix limit for databases other than sqlite go mod tidy && go mod vendor Remove unneeded break statements Make everything work with the new xorm version Fix xorm logging Fix lint Fix redis init Fix using id field Fix database init for testing Change default database log level Add xorm logger Use const for postgres go mod tidy Merge branch 'master' into update/xorm # Conflicts: # go.mod # go.sum # vendor/modules.txt go mod vendor Fix loading fixtures for postgres Go mod vendor1 Update xorm to version 1 Co-authored-by: kolaente <k@knt.li> Reviewed-on: vikunja/api#323
116 lines
3.4 KiB
Go
116 lines
3.4 KiB
Go
// Copyright 2018-2020 Vikunja and contriubtors. All rights reserved.
|
|
//
|
|
// This file is part of Vikunja.
|
|
//
|
|
// Vikunja is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// Vikunja is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Vikunja. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package db
|
|
|
|
import (
|
|
"code.vikunja.io/api/pkg/config"
|
|
"fmt"
|
|
"github.com/go-testfixtures/testfixtures/v3"
|
|
"github.com/stretchr/testify/assert"
|
|
"path/filepath"
|
|
"testing"
|
|
"xorm.io/xorm/schemas"
|
|
)
|
|
|
|
var fixtures *testfixtures.Loader
|
|
|
|
// InitFixtures initialize test fixtures for a test database
|
|
func InitFixtures(tablenames ...string) (err error) {
|
|
|
|
var testfiles func(loader *testfixtures.Loader) error
|
|
dir := filepath.Join(config.ServiceRootpath.GetString(), "pkg", "db", "fixtures")
|
|
|
|
// If fixture table names are specified, load them
|
|
// Otherwise, load all fixtures
|
|
if len(tablenames) > 0 {
|
|
for i, name := range tablenames {
|
|
tablenames[i] = filepath.Join(dir, name+".yml")
|
|
}
|
|
testfiles = testfixtures.Files(tablenames...)
|
|
} else {
|
|
testfiles = testfixtures.Directory(dir)
|
|
}
|
|
|
|
loaderOptions := []func(loader *testfixtures.Loader) error{
|
|
testfixtures.Database(x.DB().DB),
|
|
testfixtures.Dialect(config.DatabaseType.GetString()),
|
|
testfixtures.DangerousSkipTestDatabaseCheck(),
|
|
testfiles,
|
|
}
|
|
|
|
if config.DatabaseType.GetString() == "postgres" {
|
|
loaderOptions = append(loaderOptions, testfixtures.SkipResetSequences())
|
|
}
|
|
|
|
fixtures, err = testfixtures.New(loaderOptions...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// LoadFixtures load fixtures for a test database
|
|
func LoadFixtures() error {
|
|
err := fixtures.Load()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Copied from https://github.com/go-gitea/gitea/blob/master/models/test_fixtures.go#L39
|
|
// Now if we're running postgres we need to tell it to update the sequences
|
|
if x.Dialect().URI().DBType == schemas.POSTGRES {
|
|
results, err := x.QueryString(`SELECT 'SELECT SETVAL(' ||
|
|
quote_literal(quote_ident(PGT.schemaname) || '.' || quote_ident(S.relname)) ||
|
|
', COALESCE(MAX(' ||quote_ident(C.attname)|| '), 1) ) FROM ' ||
|
|
quote_ident(PGT.schemaname)|| '.'||quote_ident(T.relname)|| ';'
|
|
FROM pg_class AS S,
|
|
pg_depend AS D,
|
|
pg_class AS T,
|
|
pg_attribute AS C,
|
|
pg_tables AS PGT
|
|
WHERE S.relkind = 'S'
|
|
AND S.oid = D.objid
|
|
AND D.refobjid = T.oid
|
|
AND D.refobjid = C.attrelid
|
|
AND D.refobjsubid = C.attnum
|
|
AND T.relname = PGT.tablename
|
|
ORDER BY S.relname;`)
|
|
if err != nil {
|
|
fmt.Printf("Failed to generate sequence update: %v\n", err)
|
|
return err
|
|
}
|
|
for _, r := range results {
|
|
for _, value := range r {
|
|
_, err = x.Exec(value)
|
|
if err != nil {
|
|
fmt.Printf("Failed to update sequence: %s Error: %v\n", value, err)
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return err
|
|
}
|
|
|
|
// LoadAndAssertFixtures loads all fixtures defined before and asserts they are correctly loaded
|
|
func LoadAndAssertFixtures(t *testing.T) {
|
|
err := LoadFixtures()
|
|
assert.NoError(t, err)
|
|
}
|