public
/
bilibili-go-common
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1 Commit
1 Branch
0 Tags
34 MiB
 
 
 
Branch: master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
bilibili-go-common/app/interface/main/creative/dao/data/overview_test.go

395 lines
12 KiB
Raw Permalink Blame History

package data
import (
"context"
"encoding/binary"
"reflect"
"testing"
"go-common/library/ecode"
"github.com/bouk/monkey"
"github.com/smartystreets/goconvey/convey"
"github.com/tsuna/gohbase/hrpc"
hbase "go-common/library/database/hbase.v2"
)
func TestDataViewerBase(t *testing.T) {
var (
c = context.TODO()
mid = int64(2089809)
dt = "dt"
err error
)
convey.Convey("1", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
return nil, ecode.CreativeDataErr
})
defer guard.Unpatch()
_, err = d.ViewerBase(c, mid, dt)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
res := &hrpc.Result{}
return res, nil
})
defer guard.Unpatch()
_, err = d.ViewerBase(c, mid, dt)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("3", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
bs := make([]byte, 4)
binary.LittleEndian.PutUint32(bs, 123)
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("f"),
Qualifier: []byte("male"),
Value: bs,
})
return res, nil
})
defer guard.Unpatch()
_, err = d.ViewerBase(c, mid, dt)
ctx.Convey("3", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("4", t, func(ctx convey.C) {
g1 := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
bs := make([]byte, 4)
binary.LittleEndian.PutUint32(bs, 123)
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("g"),
Qualifier: []byte("female"),
Value: bs,
})
return res, nil
})
defer g1.Unpatch()
_, err = d.ViewerBase(c, mid, dt)
ctx.Convey("41", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
g2 := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
bs := make([]byte, 4)
binary.LittleEndian.PutUint32(bs, 123)
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("g"),
Qualifier: []byte("male"),
Value: bs,
})
return res, nil
})
defer g2.Unpatch()
_, err = d.ViewerBase(c, mid, dt)
ctx.Convey("42", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
}
func TestDataViewerTrend(t *testing.T) {
var (
c = context.TODO()
mid = int64(2089809)
dt = "dt"
err error
)
convey.Convey("1", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
return nil, ecode.CreativeDataErr
})
defer guard.Unpatch()
_, err = d.ViewerTrend(c, mid, dt)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
res := &hrpc.Result{}
return res, nil
})
defer guard.Unpatch()
_, err = d.ViewerTrend(c, mid, dt)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("3", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
bs := make([]byte, 4)
binary.LittleEndian.PutUint32(bs, 123)
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("fs"),
Qualifier: []byte("male"),
Value: bs,
})
return res, nil
})
defer guard.Unpatch()
_, err = d.ViewerTrend(c, mid, dt)
ctx.Convey("3", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("4", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
bs := make([]byte, 4)
binary.LittleEndian.PutUint32(bs, 123)
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("gs"),
Qualifier: []byte("male"),
Value: bs,
})
return res, nil
})
defer guard.Unpatch()
_, err = d.ViewerTrend(c, mid, dt)
ctx.Convey("4", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
}
func TestDataRelationFansDay(t *testing.T) {
var (
c = context.TODO()
mid = int64(2089809)
)
convey.Convey("RelationFansDay", t, func(ctx convey.C) {
_, err := d.RelationFansDay(c, mid)
ctx.Convey("Then err should be nil.res should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
}
func TestDataRelationFansHistory(t *testing.T) {
var (
c = context.TODO()
mid = int64(0)
month = ""
)
convey.Convey("RelationFansHistory", t, func(ctx convey.C) {
_, err := d.RelationFansHistory(c, mid, month)
ctx.Convey("Then err should be nil.res should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
}
func TestDataRelationFansMonth(t *testing.T) {
var (
c = context.TODO()
mid = int64(0)
)
convey.Convey("RelationFansMonth", t, func(ctx convey.C) {
_, err := d.RelationFansMonth(c, mid)
ctx.Convey("Then err should be nil.res should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
}
func TestDataViewerActionHour(t *testing.T) {
var (
c = context.TODO()
mid = int64(2089809)
dt = "dt"
err error
)
convey.Convey("1", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
return nil, ecode.CreativeDataErr
})
defer guard.Unpatch()
_, err = d.ViewerActionHour(c, mid, dt)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
res := &hrpc.Result{}
return res, nil
})
defer guard.Unpatch()
_, err = d.ViewerActionHour(c, mid, dt)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("3", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
bs := make([]byte, 4)
binary.LittleEndian.PutUint32(bs, 123)
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("fs"),
Qualifier: []byte("male"),
Value: bs,
})
return res, nil
})
defer guard.Unpatch()
_, err = d.ViewerActionHour(c, mid, dt)
ctx.Convey("3", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("4", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
bs := make([]byte, 4)
binary.LittleEndian.PutUint32(bs, 123)
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("gs"),
Qualifier: []byte("male"),
Value: bs,
})
return res, nil
})
defer guard.Unpatch()
_, err = d.ViewerActionHour(c, mid, dt)
ctx.Convey("4", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
}
func TestDataUpIncr(t *testing.T) {
var (
c = context.TODO()
mid = int64(0)
ty = int8(2)
now = ""
err error
)
convey.Convey("1", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
return nil, ecode.CreativeDataErr
})
defer guard.Unpatch()
_, err = d.UpIncr(c, mid, ty, now)
ctx.Convey("Then err should be nil.res should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
res := &hrpc.Result{}
return res, nil
})
defer guard.Unpatch()
_, err = d.UpIncr(c, mid, ty, now)
ctx.Convey("Then err should be nil.res should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("3", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
bs := make([]byte, 4)
binary.LittleEndian.PutUint32(bs, 123)
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("u"),
Qualifier: []byte("male"),
Value: bs,
})
return res, nil
})
defer guard.Unpatch()
_, err = d.UpIncr(c, mid, ty, now)
ctx.Convey("3", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
}
func TestDataThirtyDayArchive(t *testing.T) {
var (
c = context.TODO()
mid = int64(0)
ty = int8(2)
err error
)
convey.Convey("1", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
return nil, ecode.CreativeDataErr
})
defer guard.Unpatch()
_, err = d.ThirtyDayArchive(c, mid, ty)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
res := &hrpc.Result{}
return res, nil
})
defer guard.Unpatch()
_, err = d.ThirtyDayArchive(c, mid, ty)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("3", t, func(ctx convey.C) {
guard := monkey.PatchInstanceMethod(reflect.TypeOf(d.hbase), "GetStr", func(_ *hbase.Client, _ context.Context, _, _ string, _ ...func(hrpc.Call) error) (*hrpc.Result, error) {
res := &hrpc.Result{
Cells: make([]*hrpc.Cell, 0),
}
res.Cells = append(res.Cells, &hrpc.Cell{
Family: []byte("u"),
Qualifier: []byte("20181111"),
Value: []byte("200"),
})
return res, nil
})
defer guard.Unpatch()
_, err = d.ThirtyDayArchive(c, mid, ty)
ctx.Convey("3", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
}
func TestDataparseKeyValue(t *testing.T) {
var (
k = ""
v = ""
)
convey.Convey("parseKeyValue", t, func(ctx convey.C) {
_, _, err := parseKeyValue(k, v)
ctx.Convey("Then err should be nil.timestamp,value should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
}