openbilibili/app/service/main/resource/dao/ads/redis_test.go

package ads

import (
	"context"
	"testing"

	"github.com/smartystreets/goconvey/convey"
)

func TestAdsBuvidCache(t *testing.T) {
	var (
		buvidCache = map[string]map[int64]int64{
			"123456": {
				10097289: 1,
				10097290: 5,
			},
			"234567": {
				10097288: 3,
				10098523: 1,
			},
		}
		res int64
		err error
	)
	convey.Convey("AddBuvidCount - add cache", t, WithDao(func(d *Dao) {
		err = d.AddBuvidCount(context.Background(), buvidCache)
		convey.Convey("Error should be nil", func() {
			convey.So(err, convey.ShouldBeNil)
		})
		convey.Convey("BuvidCount - get cache", func() {
			convey.Convey("Case 1: faid = 10097290, buvid = 123456", func() {
				res, err = d.BuvidCount(context.Background(), 10097290, "123456")
				convey.Convey("Error should be nil, res should be 5", func() {
					convey.So(err, convey.ShouldBeNil)
					convey.So(res, convey.ShouldEqual, 5)
				})
			})
			convey.Convey("Case 2: faid = 10097288, buvid = 234567", func() {
				res, err := d.BuvidCount(context.Background(), 10097288, "234567")
				convey.Convey("Error should be nil, res should be 3", func() {
					convey.So(err, convey.ShouldBeNil)
					convey.So(res, convey.ShouldEqual, 3)
				})
			})
		})
		convey.Convey("keyBuvid", func() {
			key := d.keyBuvid("234567")
			convey.Convey("key should not be nil", func() {
				convey.So(key, convey.ShouldNotBeNil)
			})
			convey.Convey("ExistsAuth - expire cache", func() {
				res, err := d.ExistsAuth(context.Background(), key)
				convey.Convey("Error should be nil, res should be true", func() {
					convey.So(err, convey.ShouldBeNil)
					convey.So(res, convey.ShouldEqual, true)
				})
			})
		})
	}))
}