44 lines
862 B
Go
44 lines
862 B
Go
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{
|
|
Binduid: "legu131",
|
|
Name: "legu131",
|
|
Sid: 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.Binduid)
|
|
|
|
// user.ServerId = 2
|
|
user2, err := db.User_FindByAccount(user)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, "legu3", user2.Binduid)
|
|
assert.Equal(t, int32(1), user2.Sid)
|
|
}
|
|
|
|
func TestUpdate(t *testing.T) {
|
|
user := &pb.DB_UserData{
|
|
Uuid: primitive.NewObjectID().Hex(),
|
|
}
|
|
err := db.User_Update(user)
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, "NiceName", "")
|
|
}
|