package db

import (
	"go_dreamfactory/pb"
	"testing"

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

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

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

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

	// user.ServerId = 2
	user2, err := db.User_FindByAccount(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_Update(user)
	require.Nil(t, err)

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