41 lines
751 B
Go
41 lines
751 B
Go
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"go_dreamfactory/lego/sys/mgo"
|
|
"go_dreamfactory/pb"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
var db *DB
|
|
|
|
func TestMain(m *testing.M) {
|
|
imgo, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://10.0.0.9:10013"), mgo.SetMongodbDatabase("dreamfactory"))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
db = &DB{
|
|
mgo: imgo,
|
|
}
|
|
defer os.Exit(m.Run())
|
|
}
|
|
func TestMail(t *testing.T) {
|
|
|
|
mail := &pb.DBMailData{}
|
|
err := db.mgo.FindOneAndUpdate(
|
|
"mail",
|
|
bson.M{"_id": "62e784aa78d6d7bf12fb1f0e"},
|
|
bson.M{"$set": bson.M{
|
|
"check": true,
|
|
}},
|
|
options.FindOneAndUpdate().SetUpsert(true).SetReturnDocument(options.After),
|
|
).Decode(&mail)
|
|
fmt.Printf("%v", err)
|
|
}
|