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)
}