99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
|
package upper
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"go-common/app/job/main/tv/conf"
|
||
|
|
ugcMdl "go-common/app/job/main/tv/model/ugc"
|
||
|
|
|
||
|
|
. "github.com/smartystreets/goconvey/convey"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
ctx = context.TODO()
|
||
|
|
d *Dao
|
||
|
|
)
|
||
|
|
|
||
|
|
func WithDao(f func(d *Dao)) func() {
|
||
|
|
return func() {
|
||
|
|
dir, _ := filepath.Abs("../../cmd/tv-job-test.toml")
|
||
|
|
flag.Set("conf", dir)
|
||
|
|
conf.Init()
|
||
|
|
if d == nil {
|
||
|
|
d = New(conf.Conf)
|
||
|
|
}
|
||
|
|
f(d)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDao_LoadUpMeta(t *testing.T) {
|
||
|
|
Convey("TestDao_LoadUpMeta", t, WithDao(func(d *Dao) {
|
||
|
|
res, err := d.LoadUpMeta(ctx, 88895270)
|
||
|
|
So(err, ShouldBeNil)
|
||
|
|
fmt.Println(res)
|
||
|
|
}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDao_CountUP(t *testing.T) {
|
||
|
|
Convey("TestDao_CountUP", t, WithDao(func(d *Dao) {
|
||
|
|
res, err := d.CountUP(ctx)
|
||
|
|
So(err, ShouldBeNil)
|
||
|
|
fmt.Println(res)
|
||
|
|
}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDao_PickUppers(t *testing.T) {
|
||
|
|
Convey("TestDao_PickUppers", t, WithDao(func(d *Dao) {
|
||
|
|
res, myLast, err := d.PickUppers(ctx, 0, 50)
|
||
|
|
So(err, ShouldBeNil)
|
||
|
|
So(myLast, ShouldBeGreaterThan, 0)
|
||
|
|
So(len(res), ShouldBeGreaterThan, 0)
|
||
|
|
fmt.Println(myLast)
|
||
|
|
}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDao_RefreshUp(t *testing.T) {
|
||
|
|
Convey("TestDao_RefreshUp", t, WithDao(func(d *Dao) {
|
||
|
|
res, _, _ := d.PickUppers(ctx, 0, 50)
|
||
|
|
if len(res) > 0 {
|
||
|
|
err := d.RefreshUp(ctx, &ugcMdl.ReqSetUp{
|
||
|
|
MID: res[0],
|
||
|
|
Value: "Face666",
|
||
|
|
UpType: _upFace,
|
||
|
|
})
|
||
|
|
So(err, ShouldBeNil)
|
||
|
|
}
|
||
|
|
}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDao_TosyncUps(t *testing.T) {
|
||
|
|
Convey("TestDao_TosyncUps", t, WithDao(func(d *Dao) {
|
||
|
|
mids, err := d.TosyncUps(ctx)
|
||
|
|
So(err, ShouldBeNil)
|
||
|
|
fmt.Println(mids)
|
||
|
|
}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDao_FinsyncUps(t *testing.T) {
|
||
|
|
Convey("TestDao_FinsyncUps", t, WithDao(func(d *Dao) {
|
||
|
|
err := d.FinsyncUps(ctx, 88895270)
|
||
|
|
So(err, ShouldBeNil)
|
||
|
|
}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDao_IsSame(t *testing.T) {
|
||
|
|
Convey("TestDao_IsSame", t, WithDao(func(d *Dao) {
|
||
|
|
var u = &ugcMdl.Upper{
|
||
|
|
OriFace: "http://i0.hdslb.com/bfs/face/589568b0b0228d5a38b2e6f14b83bcb174637140.jpg",
|
||
|
|
}
|
||
|
|
var faceSame, _ = u.IsSame("", "http://i1.hdslb.com/bfs/face/589568b0b0228d5a38b2e6f14b83bcb174637140.jpg")
|
||
|
|
So(faceSame, ShouldBeTrue)
|
||
|
|
faceSame, _ = u.IsSame("", "http://i0.hdslb.com/bfs/face/xxx.jpg")
|
||
|
|
So(faceSame, ShouldBeFalse)
|
||
|
|
}))
|
||
|
|
}
|