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.
256 lines
9.1 KiB
256 lines
9.1 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"encoding/json" |
|
"go-common/app/service/bbq/topic/internal/model" |
|
"go-common/library/log" |
|
"math/rand" |
|
"testing" |
|
|
|
"github.com/smartystreets/goconvey/convey" |
|
) |
|
|
|
func TestDaoInsertTopicVideo(t *testing.T) { |
|
convey.Convey("InsertTopicVideo", t, func(convCtx convey.C) { |
|
var ( |
|
ctx = context.Background() |
|
svid = rand.Int63() % 1000000 |
|
topicIDs = []int64{1} |
|
) |
|
convCtx.Convey("When everything goes positive", func(convCtx convey.C) { |
|
rowsAffected, err := d.InsertTopicVideo(ctx, svid, topicIDs) |
|
convCtx.Convey("Then err should be nil.rowsAffected should not be nil.", func(convCtx convey.C) { |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(rowsAffected, convey.ShouldEqual, 1) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoUpdateVideoScore(t *testing.T) { |
|
convey.Convey("UpdateVideoScore", t, func(convCtx convey.C) { |
|
var ( |
|
ctx = context.Background() |
|
svid = int64(1) |
|
score = float64(1.0) |
|
) |
|
convCtx.Convey("When everything goes positive", func(convCtx convey.C) { |
|
err := d.UpdateVideoScore(ctx, svid, score) |
|
convCtx.Convey("Then err should be nil.", func(convCtx convey.C) { |
|
convCtx.So(err, convey.ShouldBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoUpdateVideoState(t *testing.T) { |
|
convey.Convey("UpdateVideoState", t, func(convCtx convey.C) { |
|
var ( |
|
ctx = context.Background() |
|
svid = int64(1) |
|
state = int32(1) |
|
) |
|
convCtx.Convey("When everything goes positive", func(convCtx convey.C) { |
|
err := d.UpdateVideoState(ctx, svid, state) |
|
convCtx.Convey("Then err should be nil.", func(convCtx convey.C) { |
|
convCtx.So(err, convey.ShouldBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoListTopicVideos(t *testing.T) { |
|
convey.Convey("ListTopicVideos", t, func(convCtx convey.C) { |
|
var ( |
|
ctx = context.Background() |
|
topicID = int64(1) |
|
) |
|
res, hasMore, err := d.ListTopicVideos(ctx, topicID, "", "", model.TopicVideoSize) |
|
log.V(1).Infow(ctx, "res", res) |
|
convCtx.Convey("Then err should be nil.res,hasMore should not be nil.", func(convCtx convey.C) { |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(hasMore, convey.ShouldNotBeNil) |
|
convCtx.So(res, convey.ShouldNotBeNil) |
|
}) |
|
|
|
originStickList, _ := d.GetStickTopicVideo(ctx, topicID) |
|
newStickList := []int64{1, 2, 3, 4, 5, 6} |
|
d.SetStickTopicVideo(ctx, topicID, newStickList) |
|
var data []byte |
|
|
|
convCtx.Convey("cursor_in_rank && direction_next", func(convCtx convey.C) { |
|
data, _ = json.Marshal(model.CursorValue{Offset: 0, StickRank: 2}) |
|
res, hasMore, err = d.ListTopicVideos(ctx, topicID, "", string(data), model.TopicVideoSize) |
|
log.V(1).Infow(ctx, "res", res, "has_more", hasMore) |
|
|
|
convCtx.So(res[0].Svid, convey.ShouldEqual, 3) |
|
convCtx.So(res[1].Svid, convey.ShouldEqual, 4) |
|
convCtx.So(res[2].Svid, convey.ShouldEqual, 5) |
|
convCtx.So(res[3].Svid, convey.ShouldEqual, 6) |
|
// 检验cursor值是否符合要求 |
|
var unmarshalCursor model.CursorValue |
|
json.Unmarshal([]byte(res[3].CursorValue), &unmarshalCursor) |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(unmarshalCursor.StickRank, convey.ShouldEqual, 6) |
|
convCtx.So(unmarshalCursor.Offset, convey.ShouldEqual, 0) |
|
json.Unmarshal([]byte(res[4].CursorValue), &unmarshalCursor) |
|
convCtx.So(unmarshalCursor.StickRank, convey.ShouldEqual, 0) |
|
convCtx.So(unmarshalCursor.Offset, convey.ShouldEqual, 1) |
|
|
|
convCtx.So(len(res), convey.ShouldEqual, model.TopicVideoSize+4) |
|
convCtx.So(hasMore, convey.ShouldBeTrue) |
|
convCtx.So(res, convey.ShouldNotBeNil) |
|
}) |
|
|
|
convCtx.Convey("cursor_in_rank && direction_prev", func(convCtx convey.C) { |
|
data, _ = json.Marshal(model.CursorValue{Offset: 0, StickRank: 4}) |
|
res, hasMore, err = d.ListTopicVideos(ctx, topicID, string(data), "", model.TopicVideoSize) |
|
log.V(1).Infow(ctx, "res", res, "has_more", hasMore) |
|
|
|
convCtx.So(res[0].Svid, convey.ShouldEqual, 3) |
|
convCtx.So(res[1].Svid, convey.ShouldEqual, 2) |
|
convCtx.So(res[2].Svid, convey.ShouldEqual, 1) |
|
convCtx.So(len(res), convey.ShouldEqual, 3) |
|
convCtx.So(hasMore, convey.ShouldBeFalse) |
|
|
|
// 边缘情况，选择了rank=1的视频 |
|
data, _ = json.Marshal(model.CursorValue{Offset: 0, StickRank: 1}) |
|
res, hasMore, err = d.ListTopicVideos(ctx, topicID, string(data), "", model.TopicVideoSize) |
|
log.V(1).Infow(ctx, "res", res, "has_more", hasMore) |
|
|
|
convCtx.So(len(res), convey.ShouldEqual, 0) |
|
convCtx.So(hasMore, convey.ShouldBeFalse) |
|
}) |
|
|
|
convCtx.Convey("direction_prev", func(convCtx convey.C) { |
|
data, _ = json.Marshal(model.CursorValue{Offset: 1, StickRank: 0}) |
|
res, hasMore, err = d.ListTopicVideos(ctx, topicID, string(data), "", model.TopicVideoSize) |
|
log.V(1).Infow(ctx, "res", res, "has_more", hasMore) |
|
|
|
convCtx.So(len(res), convey.ShouldEqual, len(newStickList)) |
|
convCtx.So(hasMore, convey.ShouldBeFalse) |
|
|
|
data, _ = json.Marshal(model.CursorValue{Offset: 5, StickRank: 0}) |
|
res, hasMore, err = d.ListTopicVideos(ctx, topicID, string(data), "", model.TopicVideoSize) |
|
log.V(1).Infow(ctx, "res", res, "has_more", hasMore) |
|
// 检验cursor值是否符合要求 |
|
var unmarshalCursor model.CursorValue |
|
json.Unmarshal([]byte(res[0].CursorValue), &unmarshalCursor) |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(unmarshalCursor.StickRank, convey.ShouldEqual, 0) |
|
convCtx.So(unmarshalCursor.Offset, convey.ShouldEqual, 4) |
|
json.Unmarshal([]byte(res[1].CursorValue), &unmarshalCursor) |
|
convCtx.So(unmarshalCursor.StickRank, convey.ShouldEqual, 0) |
|
convCtx.So(unmarshalCursor.Offset, convey.ShouldEqual, 3) |
|
}) |
|
|
|
convCtx.Convey("direction_next", func(convCtx convey.C) { |
|
data, _ = json.Marshal(model.CursorValue{Offset: 1, StickRank: 0}) |
|
res, hasMore, err = d.ListTopicVideos(ctx, topicID, "", string(data), model.TopicVideoSize) |
|
log.V(1).Infow(ctx, "res", res, "has_more", hasMore) |
|
|
|
convCtx.So(len(res), convey.ShouldEqual, int(model.TopicVideoSize)) |
|
convCtx.So(hasMore, convey.ShouldBeTrue) |
|
|
|
data, _ = json.Marshal(model.CursorValue{Offset: 5, StickRank: 0}) |
|
res, hasMore, err = d.ListTopicVideos(ctx, topicID, "", string(data), model.TopicVideoSize) |
|
log.V(1).Infow(ctx, "res", res, "has_more", hasMore) |
|
// 检验cursor值是否符合要求 |
|
var unmarshalCursor model.CursorValue |
|
json.Unmarshal([]byte(res[0].CursorValue), &unmarshalCursor) |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(unmarshalCursor.StickRank, convey.ShouldEqual, 0) |
|
convCtx.So(unmarshalCursor.Offset, convey.ShouldEqual, 6) |
|
json.Unmarshal([]byte(res[1].CursorValue), &unmarshalCursor) |
|
convCtx.So(unmarshalCursor.StickRank, convey.ShouldEqual, 0) |
|
convCtx.So(unmarshalCursor.Offset, convey.ShouldEqual, 7) |
|
|
|
}) |
|
|
|
// 恢复原来的置顶话题 |
|
if len(originStickList) > 0 { |
|
d.SetStickTopicVideo(ctx, topicID, originStickList) |
|
} |
|
|
|
}) |
|
} |
|
|
|
func TestDaogetStickTopicVideo(t *testing.T) { |
|
convey.Convey("GetStickTopicVideo", t, func(convCtx convey.C) { |
|
var ( |
|
ctx = context.Background() |
|
topicID = int64(1) |
|
) |
|
convCtx.Convey("When everything goes positive", func(convCtx convey.C) { |
|
list, err := d.GetStickTopicVideo(ctx, topicID) |
|
log.V(1).Infow(ctx, "list", list) |
|
convCtx.Convey("Then err should be nil.list should not be nil.", func(convCtx convey.C) { |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(list, convey.ShouldNotBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoStickTopicVideo(t *testing.T) { |
|
convey.Convey("StickTopicVideo", t, func(convCtx convey.C) { |
|
var ( |
|
ctx = context.Background() |
|
opTopicID = int64(1) |
|
opSvid = int64(1) |
|
op = int64(1) |
|
) |
|
|
|
originStickList, _ := d.GetStickTopicVideo(ctx, opTopicID) |
|
|
|
convCtx.Convey("common stick operate", func(convCtx convey.C) { |
|
err := d.StickTopicVideo(ctx, opTopicID, opSvid, op) |
|
newStickList, _ := d.GetStickTopicVideo(ctx, opTopicID) |
|
log.V(1).Infow(ctx, "new_stick_list", newStickList) |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(newStickList[0], convey.ShouldEqual, 1) |
|
}) |
|
|
|
convCtx.Convey("common cancel stick operate", func(convCtx convey.C) { |
|
err := d.StickTopicVideo(ctx, opTopicID, opSvid, 0) |
|
newCancelStickList, _ := d.GetStickTopicVideo(ctx, opTopicID) |
|
log.V(1).Infow(ctx, "new_cancel_stick_list", newCancelStickList) |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(newCancelStickList[0], convey.ShouldNotEqual, 1) |
|
}) |
|
|
|
convCtx.Convey("stick num test", func(convCtx convey.C) { |
|
for i := 1; i < model.MaxStickTopicVideoNum+3; i++ { |
|
d.StickTopicVideo(ctx, opTopicID, int64(i), 1) |
|
} |
|
list, _ := d.GetStickTopicVideo(ctx, opTopicID) |
|
log.V(1).Infow(ctx, "list", list) |
|
convCtx.So(len(list), convey.ShouldEqual, model.MaxStickTopicVideoNum) |
|
}) |
|
|
|
// 恢复原来的置顶话题 |
|
if len(originStickList) > 0 { |
|
d.SetStickTopicVideo(ctx, opTopicID, originStickList) |
|
} |
|
|
|
}) |
|
} |
|
|
|
func TestDaoGetVideoTopic(t *testing.T) { |
|
convey.Convey("GetVideoTopic", t, func(convCtx convey.C) { |
|
var ( |
|
ctx = context.Background() |
|
svid = int64(1547635456050324977) |
|
) |
|
convCtx.Convey("When everything goes positive", func(convCtx convey.C) { |
|
res, err := d.GetVideoTopic(ctx, svid) |
|
data, _ := json.Marshal(res) |
|
log.V(1).Infow(ctx, "res", string(data)) |
|
convCtx.Convey("Then err should be nil.list should not be nil.", func(convCtx convey.C) { |
|
convCtx.So(err, convey.ShouldBeNil) |
|
convCtx.So(res, convey.ShouldNotBeNil) |
|
}) |
|
}) |
|
}) |
|
}
|
|
|