mirror of https://github.com/dtm-labs/dtm.git
csharpjavadistributed-transactionsdtmgogolangmicroservicenodejsphpdatabasesagaseatatcctransactiontransactionsxapythondistributed
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.3 KiB
55 lines
1.3 KiB
package test
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/dtm-labs/dtm/client/dtmcli"
|
|
"github.com/dtm-labs/dtm/client/dtmcli/dtmimp"
|
|
"github.com/dtm-labs/dtm/client/dtmgrpc"
|
|
"github.com/dtm-labs/dtm/dtmsvr/storage/sql"
|
|
"github.com/dtm-labs/dtm/dtmutil"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGeneralDB(t *testing.T) {
|
|
if conf.Store.IsDB() {
|
|
testSQL(t)
|
|
testDbAlone(t)
|
|
}
|
|
}
|
|
|
|
func testSQL(t *testing.T) {
|
|
conf := conf.Store.GetDBConf()
|
|
conf.Host = "127.0.0.1" // use a new host to trigger SetDBConn called
|
|
db := dtmutil.DbGet(conf, sql.SetDBConn)
|
|
err := func() (rerr error) {
|
|
defer dtmimp.P2E(&rerr)
|
|
db.Must().Exec("select a")
|
|
return nil
|
|
}()
|
|
assert.NotEqual(t, nil, err)
|
|
}
|
|
|
|
func testDbAlone(t *testing.T) {
|
|
db, err := dtmimp.StandaloneDB(conf.Store.GetDBConf())
|
|
assert.Nil(t, err)
|
|
_, err = dtmimp.DBExec(conf.Store.Driver, db, "select 1")
|
|
assert.Equal(t, nil, err)
|
|
_, err = dtmimp.DBExec(conf.Store.Driver, db, "")
|
|
assert.Equal(t, nil, err)
|
|
db.Close()
|
|
_, err = dtmimp.DBExec(conf.Store.Driver, db, "select 1")
|
|
assert.NotEqual(t, nil, err)
|
|
}
|
|
|
|
func TestMustGenGid(t *testing.T) {
|
|
dtmgrpc.MustGenGid(dtmutil.DefaultGrpcServer)
|
|
dtmcli.MustGenGid(dtmutil.DefaultHTTPServer)
|
|
}
|
|
|
|
func MaySkipMongo(t *testing.T) {
|
|
if os.Getenv("SKIP_MONGO") != "" {
|
|
t.Skip("skipping test with mongo")
|
|
}
|
|
}
|
|
|