dtm/common/types_test.go
2021-08-21 04:06:59 +08:00

31 lines
620 B
Go

package common
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/yedf/dtm/dtmcli"
)
func TestDb(t *testing.T) {
db := DbGet(DtmConfig.DB)
err := func() (rerr error) {
defer dtmcli.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)
}
func TestDbAlone(t *testing.T) {
db, err := dtmcli.SdbAlone(DtmConfig.DB)
assert.Nil(t, err)
_, err = dtmcli.DBExec(db, "select 1")
assert.Equal(t, nil, err)
db.Close()
_, err = dtmcli.DBExec(db, "select 1")
assert.NotEqual(t, nil, err)
}