61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
package db
|
|
|
|
import (
|
|
"go_dreamfactory/pb"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/liwei1dao/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",
|
|
Email: "1111@legu.com",
|
|
}
|
|
|
|
err := db.User_CreateUser(user)
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestFindOne(t *testing.T) {
|
|
user, err := db.User_FindUserById("")
|
|
require.Nil(t, err)
|
|
assert.Equal(t, "legu1", user.Account)
|
|
|
|
user2, err := db.User_FindUserByAccount("legu1")
|
|
require.Nil(t, err)
|
|
assert.Equal(t, "legu1", user2.Account)
|
|
|
|
}
|
|
|
|
func TestUpdate(t *testing.T) {
|
|
user := &pb.DB_UserData{
|
|
UserId: primitive.NewObjectID().String(),
|
|
Email: "new@qq.com",
|
|
}
|
|
err := db.User_UpdateUser(user)
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, "new@qq.com", user.Email)
|
|
}
|