vikunja-api/pkg/models/unit_tests.go

95 lines
2.7 KiB
Go
Raw Normal View History

// Vikunja is a todo-list application to facilitate your life.
// Copyright 2018-2019 Vikunja and contributors. All rights reserved.
2018-11-26 21:17:33 +01:00
//
// This program 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.
2018-11-26 21:17:33 +01:00
//
// This program 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.
2018-11-26 21:17:33 +01:00
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
2018-11-26 21:17:33 +01:00
2018-06-10 11:11:41 +02:00
package models
import (
2019-03-29 18:54:35 +01:00
"code.vikunja.io/api/pkg/config"
_ "code.vikunja.io/api/pkg/config" // To trigger its init() which initializes the config
2019-10-16 22:52:29 +02:00
"code.vikunja.io/api/pkg/db"
"code.vikunja.io/api/pkg/log"
2018-10-31 13:42:38 +01:00
"code.vikunja.io/api/pkg/mail"
2018-06-10 11:11:41 +02:00
"fmt"
"github.com/go-xorm/xorm"
2019-12-07 23:28:45 +01:00
"github.com/stretchr/testify/assert"
2018-06-10 11:11:41 +02:00
"gopkg.in/testfixtures.v2"
"os"
"path/filepath"
2019-12-07 23:28:45 +01:00
"testing"
2018-06-10 11:11:41 +02:00
)
2019-04-21 20:18:17 +02:00
// SetupTests takes care of seting up the db, fixtures etc.
// This is an extra function to be able to call the fixtures setup from the integration tests.
func SetupTests(pathToRoot string) {
2018-06-10 11:11:41 +02:00
var err error
2018-12-18 17:01:46 +01:00
fixturesDir := filepath.Join(pathToRoot, "pkg", "models", "fixtures")
2018-06-10 11:11:41 +02:00
if err = createTestEngine(fixturesDir); err != nil {
2019-07-20 20:12:10 +02:00
log.Fatalf("Error creating test engine: %v\n", err)
2018-06-10 11:11:41 +02:00
}
2018-10-28 17:11:13 +01:00
// Start the pseudo mail queue
mail.StartMailDaemon()
// Create test database
2019-10-16 22:52:29 +02:00
if err = db.LoadFixtures(); err != nil {
2019-07-20 20:12:10 +02:00
log.Fatalf("Error preparing test database: %v", err.Error())
}
2018-06-10 11:11:41 +02:00
}
func createTestEngine(fixturesDir string) error {
var err error
2019-03-29 18:54:35 +01:00
var fixturesHelper testfixtures.Helper = &testfixtures.SQLite{}
2018-12-18 17:01:46 +01:00
// If set, use the config we provided instead of normal
if os.Getenv("VIKUNJA_TESTS_USE_CONFIG") == "1" {
2019-10-16 22:52:29 +02:00
x, err = db.CreateTestEngine()
2018-12-18 17:01:46 +01:00
if err != nil {
2019-10-16 22:52:29 +02:00
return fmt.Errorf("error getting test engine: %v", err)
2018-12-18 17:01:46 +01:00
}
2019-03-29 18:54:35 +01:00
2019-10-16 22:52:29 +02:00
err = initSchema(x)
2019-03-29 18:54:35 +01:00
if err != nil {
return err
}
if config.DatabaseType.GetString() == "mysql" {
2019-03-29 18:54:35 +01:00
fixturesHelper = &testfixtures.MySQL{}
}
2018-12-18 17:01:46 +01:00
} else {
2019-10-16 22:52:29 +02:00
x, err = db.CreateTestEngine()
2018-12-18 17:01:46 +01:00
if err != nil {
2019-10-16 22:52:29 +02:00
return fmt.Errorf("error getting test engine: %v", err)
2018-12-18 17:01:46 +01:00
}
// Sync dat shit
2019-10-16 22:52:29 +02:00
err = initSchema(x)
if err != nil {
2018-12-18 17:01:46 +01:00
return fmt.Errorf("sync database struct error: %v", err)
}
2018-06-10 14:24:33 +02:00
}
2018-06-10 11:11:41 +02:00
2019-10-16 22:52:29 +02:00
return db.InitFixtures(fixturesHelper, fixturesDir)
}
func initSchema(tx *xorm.Engine) error {
return tx.Sync2(GetTables()...)
2018-06-10 11:11:41 +02:00
}
2019-12-07 23:28:45 +01:00
func initFixtures(t *testing.T) {
// Init db fixtures
err := db.LoadFixtures()
assert.NoError(t, err)
}