cc66ceb484
- use clearer method names, comments; - define dtm usage info; - determine action by switch-case instead of if-else, because only os.Args[1] needs to be checked; - use time layout '2006-01-02 15:04:05.999' to format milliseconds; - change filenames dtmsvr/main.go to dtmsvr/dtmsvr.go; - simplify code; - use `()` to think less about operator precedence; xxx
31 lines
624 B
Go
31 lines
624 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.StandaloneDB(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)
|
|
}
|