Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/models/gorm_test.go
akuma06 2646bc2db8 This is a prelimenary work
Showing how we can remove services, preventing cyclic imports and lessing the number of imports.
Now db is in models. Db and models are highly tightened, according to go standards, you should put them in the same package.
In models, there are folders separating the different methods used to modify the models. For example, if you want to create a user, you have to use /models (for the user struct) and /models/user (for creating a user.
However, if you want to delete a torrent, you just have to import /models and do torrent.Delete(definitely bool).

By the way packages in models are the plural name of a model. For example, you have torrent.go for a torrent model and its package torrents for db stuff related functions (Find, Create, Some helpers)
2017-06-29 00:44:07 +02:00

98 lignes
2,4 Kio
Go

package models
import (
"fmt"
"path"
"testing"
"github.com/NyaaPantsu/nyaa/config"
"github.com/azhao12345/gorm"
)
// run before config/parse.go:init()
var _ = func() (_ struct{}) {
config.ConfigPath = path.Join("..", config.ConfigPath)
config.DefaultConfigPath = path.Join("..", config.DefaultConfigPath)
config.Parse()
return
}()
type errorLogger struct {
t *testing.T
}
func (logger *errorLogger) Print(values ...interface{}) {
if len(values) > 1 {
message := gorm.LogFormatter(values...)
level := values[0]
if level == "log" {
logger.t.Error(message...)
}
fmt.Println(message...)
}
}
func TestGormInitSqlite(t *testing.T) {
config.Conf.DBType = SqliteType
config.Conf.DBParams = ":memory:?cache=shared&mode=memory"
config.Conf.DBLogMode = "detailed"
db, err := GormInit(config.Conf, &errorLogger{t})
if err != nil {
t.Errorf("failed to initialize database: %v", err)
return
}
if db == nil {
return
}
err = db.Close()
if err != nil {
t.Errorf("failed to close database: %v", err)
}
}
// This test requires a running postgres instance. To run it in CI build add these settings in the .travis.yml
// services:
// - postgresql
// before_script:
// - psql -c "CREATE DATABASE nyaapantsu;" -U postgres
// - psql -c "CREATE USER nyaapantsu WITH PASSWORD 'nyaapantsu';" -U postgres
//
// Then enable the test by setting this variable to "true" via ldflags:
// go test ./... -v -ldflags="-X github.com/NyaaPantsu/nyaa/db.testPostgres=true"
var testPostgres = "false"
func TestGormInitPostgres(t *testing.T) {
if testPostgres != "true" {
t.Skip("skip", testPostgres)
}
config.Conf.DBType = "postgres"
config.Conf.DBParams = "host=localhost user=nyaapantsu dbname=nyaapantsu sslmode=disable password=nyaapantsu"
config.Conf.DBLogMode = "detailed"
config.Conf.Models.CommentsTableName = "comments"
config.Conf.Models.FilesTableName = "files"
config.Conf.Models.NotificationsTableName = "notifications"
config.Conf.Models.ReportsTableName = "torrent_reports"
config.Conf.Models.TorrentsTableName = "torrents"
config.Conf.Models.UploadsOldTableName = "user_uploads_old"
config.Conf.Models.LastOldTorrentID = 90000
db, err := GormInit(config.Conf, &errorLogger{t})
if err != nil {
t.Errorf("failed to initialize database: %v", err)
}
if db == nil {
return
}
err = db.Close()
if err != nil {
t.Errorf("failed to close database: %v", err)
}
}