package db

import (
	"go_dreamfactory/pb"
	"log"
	"os"
	"testing"

	"go_dreamfactory/lego/sys/mgo"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

var db *DB

func TestMain(m *testing.M) {
	imgo, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://admin:123456@10.0.0.9:27018"), mgo.SetMongodbDatabase("dreamfactory"))
	if err != nil {
		log.Fatal(err)
	}

	db = &DB{
		mgo: imgo,
	}
	defer os.Exit(m.Run())
}

func TestCreate(t *testing.T) {
	user := &pb.DB_UserData{
		Account:  "legu3",
		NiceName: "乐谷3",
		ServerId: 1,
	}

	err := db.User_CreateUser(user)
	require.Nil(t, err)
}

func TestFindOne(t *testing.T) {
	user, err := db.User_FindUserById("629eb3f4132dc4bb26139659")
	require.Nil(t, err)
	assert.Equal(t, "legu3", user.Account)

	// user.ServerId = 2
	user2, err := db.User_FindUserByAccount(user)
	require.Nil(t, err)
	assert.Equal(t, "legu3", user2.Account)
	assert.Equal(t, int32(1), user2.ServerId)
}

func TestUpdate(t *testing.T) {
	user := &pb.DB_UserData{
		UserId: primitive.NewObjectID().Hex(),
	}
	err := db.User_UpdateUser(user)
	require.Nil(t, err)

	assert.Equal(t, "NiceName", "")
}