mdbc/insert_scope_test.go
2022-02-23 16:59:45 +08:00

68 lines
1.4 KiB
Go

package mdbc
import (
"testing"
"github.com/sirupsen/logrus"
)
func TestInsertScope_One(t *testing.T) {
cfg := &Config{
URI: "mongodb://mdbc:mdbc@10.0.0.135:27117/admin",
MinPoolSize: 32,
ConnTimeout: 10,
}
cfg.RegistryBuilder = RegisterTimestampCodec(nil)
client, err := ConnInit(cfg)
if err != nil {
logrus.Fatalf("get err: %+v", err)
}
Init(client).InitDB(client.Database("mdbc"))
var m = NewModel(&ModelSchedTask{})
var record = ModelSchedTask{}
record.TaskState = uint32(TaskState_TaskStateCompleted)
record.Id = "insertffddddfdfdsdnkodsanfkasdf"
res, err := m.SetDebug(true).Insert().One(&record)
if err != nil {
panic(err)
}
logrus.Infof("res: %+v", res.GetString())
}
func TestInsertScope_Many(t *testing.T) {
cfg := &Config{
URI: "mongodb://admin:admin@10.0.0.135:27017/admin",
MinPoolSize: 32,
ConnTimeout: 10,
}
cfg.RegistryBuilder = RegisterTimestampCodec(nil)
client, err := ConnInit(cfg)
if err != nil {
logrus.Fatalf("get err: %+v", err)
}
Init(client).InitDB(client.Database("mdbc"))
var m = NewModel(&ModelSchedTask{})
var list = []interface{}{
&ModelSchedTask{
Id: "insertManyaaaajkfnjaksdf",
},
&ModelSchedTask{
Id: "insertManybbbwdsfsadnasjkf",
},
}
res, err := m.SetDebug(true).Insert().Many(list)
if err != nil {
panic(err)
}
logrus.Infof("res: %+v", res.GetListString())
}