dtm/common/types_test.go
2021-07-31 16:46:02 +08:00

42 lines
744 B
Go

package common
import (
"testing"
"github.com/go-playground/assert/v2"
)
type testConfig struct {
DB map[string]string `yaml:"DB"`
}
var config = testConfig{}
func init() {
InitConfig(GetProjectDir(), &config)
config.DB["database"] = ""
}
func TestDb(t *testing.T) {
db := DbGet(config.DB)
err := func() (rerr error) {
defer P2E(&rerr)
dbr := db.NoMust().Exec("select a")
assert.NotEqual(t, nil, dbr.Error)
db.Must().Exec("select a")
return nil
}()
assert.NotEqual(t, nil, err)
sdb := db.ToSQLDB()
db = SQLDB2DB(sdb)
}
func TestDbAlone(t *testing.T) {
db := DbAlone(config.DB)
_, err := DbExec(db, "select 1")
assert.Equal(t, nil, err)
db.Close()
_, err = DbExec(db, "select 1")
assert.NotEqual(t, nil, err)
}