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.
74 lines
1.4 KiB
74 lines
1.4 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"flag" |
|
"go-common/app/service/live/live-dm/conf" |
|
"path/filepath" |
|
"testing" |
|
) |
|
|
|
func init() { |
|
dir, _ := filepath.Abs("../cmd/test.toml") |
|
flag.Set("conf", dir) |
|
var err error |
|
if err = conf.Init(); err != nil { |
|
panic(err) |
|
} |
|
InitAPI() |
|
InitGrpc(conf.Conf) |
|
} |
|
|
|
//group=qa01 DEPLOY_ENV=uat go test -run TestLimitPerSec |
|
func TestLimitPerSec(t *testing.T) { |
|
l := LimitCheckInfo{ |
|
UID: 111, |
|
RoomID: 222, |
|
Msg: "6666", |
|
Dao: New(conf.Conf), |
|
MsgType: 0, |
|
Conf: &LimitConf{ |
|
DmNum: 20, |
|
DMPercent: 25, |
|
}, |
|
} |
|
if err := l.LimitPerSec(context.TODO()); err != nil { |
|
t.Error("每秒限制错误:", err) |
|
} |
|
} |
|
|
|
//group=qa01 DEPLOY_ENV=uat go test -run TestLimitSameMsg |
|
func TestLimitSameMsg(t *testing.T) { |
|
l := LimitCheckInfo{ |
|
UID: 111, |
|
RoomID: 222, |
|
Msg: "6666", |
|
Dao: New(conf.Conf), |
|
MsgType: 0, |
|
Conf: &LimitConf{ |
|
DmNum: 20, |
|
DMPercent: 25, |
|
}, |
|
} |
|
if err := l.LimitSameMsg(context.TODO()); err != nil { |
|
t.Error("5秒相同发言错误:", err) |
|
} |
|
} |
|
|
|
//group=qa01 DEPLOY_ENV=uat go test -run TestLimitRoomPerSecond |
|
func TestLimitRoomPerSecond(t *testing.T) { |
|
l := LimitCheckInfo{ |
|
UID: 111, |
|
RoomID: 222, |
|
Msg: "6666", |
|
Dao: New(conf.Conf), |
|
MsgType: 0, |
|
Conf: &LimitConf{ |
|
DmNum: 20, |
|
DMPercent: 25, |
|
}, |
|
} |
|
if err := l.LimitRoomPerSecond(context.TODO()); err != nil { |
|
t.Error("每秒20弹幕错误:", err) |
|
} |
|
}
|
|
|