gitee.com/KyleChenSource/lib-robot@v1.0.2/robottest/testcaseimp/brutetest.go

package testcaseimp

/*
	暴力测试
	配置 [
		// 根据循序进行测试
			{
			'name': // 协议名
			'id': // 协议id, 可以为0
			'notest':
			'data': { //发送的协议
				%INT8%  16,32,64:[min:max:...valid value]
				%RAND_UINT8%  16,32,64:[min:max:...valid value]
				%RAND_FLOAT32%  64:[min:max:...valid value]
				%RAND_BOOL%
				%RAND_STRING% :[min:max:...valid value]
				%RAND_BYTES[m:n]% :[min:max:...valid value]
				%RAND_MSG:msgName%
				%RAND_ARRAY:msgName[minLen:maxLen]% :[min:max:...valid value]
			}
		]
	}
*/
import (
	"fmt"
	"time"

	"gitee.com/KyleChenSource/lib-robot/robottest/common"
	"gitee.com/KyleChenSource/lib-robot/robottest/common/brutetest"
	"gitee.com/KyleChenSource/lib-robot/robottest/robot/statistic"
	"gitee.com/KyleChenSource/lib-robot/robottest/robot/testcase"

	"gitee.com/KyleChenSource/lib-robot/robottest/protos"

	"gitee.com/KyleChenSource/lib-robot/robottest/log"
)

const (
	prePath = "./brutetest"
)

type BruteTestFactory struct {
}

type BruteTestCnf struct {
	testcase.TestcaseActionConfig_
	Cid   int
	Index int
}

func (this *BruteTestFactory) Cnf() testcase.TestcaseActionCnf {
	return &BruteTestCnf{}
}

func (this *BruteTestFactory) New(id testcase.TestActionID, tf testcase.ITestFlow, cnf testcase.TestcaseActionCnf) (testcase.ITestcaseAction, error) {
	bruteCnf := cnf.(*BruteTestCnf)
	name := fmt.Sprintf("brute_%d_%d", bruteCnf.Cid, bruteCnf.Index)
	b, ok := tf.TestCtx().Get(name)
	if ok {
		brute := b.(*BruteTest)
		brute._id = id
		brute._tf = tf

		return b.(*BruteTest), nil
	}

	n := &BruteTest{
		_id:  id,
		_tf:  tf,
		_cnf: cnf.(*BruteTestCnf),
	}
	tf.TestCtx().Set(name, n)
	return n, nil
}

func init() {
	testcase.TESTCASEACTION_MGR.Register("BruteTest", &BruteTestFactory{})
}

type BruteTest struct {
	_id       testcase.TestActionID
	_tf       testcase.ITestFlow
	_cnf      *BruteTestCnf
	_gen      *brutetest.BruteMsgGenerator
	_index    int // 当前协议的次数记录
	_data     protos.JsonString
	_timeSend int64 // 发送时间
	_timer    *common.TimeInterface
}

// 启动
// 目前仅仅支持定时器和消息两种驱动模式
func (this *BruteTest) Start() error {
	if this._gen == nil {
		cnf := brutetest.BrutetestCaselistMap[this._cnf.Cid][this._cnf.Index]
		this._gen = cnf.Generator(0, 0)
		if this._gen == nil {
			return fmt.Errorf("%s %s Generator nil", this._tf.LogPre(), cnf.MsgName)
		}

		log.LogInfo("%s %s Generator. Count:%d",
			this._tf.LogPre(),
			cnf.MsgName,
			this._gen.Count())
	} else {
		log.LogInfo("%s %s Generator. Count:%d index:%d",
			this._tf.LogPre(),
			this._gen.BruteCnf().MsgName,
			this._gen.Count(),
			this._index)
	}

	// 初始化配置
	this.msgSend()
	return nil
}

// 停止，需要删除目前所有的驱动注册
func (this *BruteTest) Stop() {
	if this._timer != nil {
		this._tf.TestCtx().TimerDel(this._timer)
		this._timer = nil
	}
}

func (this *BruteTest) ID() testcase.TestActionID {
	return this._id
}

func (this *BruteTest) Name() string {
	return this._cnf.ActionName()
}

func (this *BruteTest) Testflow() testcase.ITestFlow {
	return this._tf
}

func (this *BruteTest) OnMsg(protoId protos.ProtoID, header protos.JsonString, data protos.JsonString) {
	tCost := time.Now().UnixMilli() - this._timeSend
	log.LogInfo("MsgResponse OK. %s action:%d OnMsg:%d index:%d header:%s data:%s timeCost:%d",
		this._tf.LogPre(),
		this._id,
		protoId,
		this._index-1,
		header, data, tCost)

	if this._timer != nil {
		this._tf.TestCtx().TimerDel(this._timer)
		this._timer = nil
	}

	this._tf.TestCtx().MsgHandleUnreg(protos.ProtoID(this._gen.BruteCnf().ResponeId), this)
	statistic.Record(&statistic.StatisticEvent{
		Type: statistic.StatisticType_Msg_Success,
		ID:   int(protoId),
		Time: tCost,
	})

	this.msgSend()
}

func (this *BruteTest) msgSend() {
	ok, data := this._gen.Next()
	if !ok {
		this._tf.OnActionEnd(this, nil, this._cnf.FailContinue)
		return
	}

	this._data = protos.JsonString(data)
	err := this._tf.TestCtx().Send(protos.ProtoID(this._gen.BruteCnf().MsgId), this._data)
	if err != nil {
		this._tf.OnActionEnd(this, fmt.Errorf("SendMsg %d(%s) err:%s",
			this._gen.BruteCnf().MsgId,
			this._gen.BruteCnf().MsgName,
			err.Error()),
			this._cnf.FailContinue)
		return
	}

	log.LogInfo("%s action:%d msgSend:%d(%s) index:%d data:%s", this._tf.LogPre(), this._id,
		this._gen.BruteCnf().MsgId,
		this._gen.BruteCnf().MsgName,
		this._index,
		data,
	)
	this._index++

	this._timeSend = time.Now().UnixMilli()
	// 发送成功，开始等待需要的回包
	if this._gen.BruteCnf().ResponeId != 0 {
		this._tf.TestCtx().MsgHandleReg(protos.ProtoID(this._gen.BruteCnf().ResponeId), this)

		if this._gen.BruteCnf().Timeout != 0 {
			this._timer, err = this._tf.TestCtx().TimerAdd(this._gen.BruteCnf().Timeout,
				this.OnTimerResponseTimeout,
				nil)
		}
	} else if this._gen.BruteCnf().Timeout != 0 {
		statistic.Record(&statistic.StatisticEvent{
			Type: statistic.StatisticType_Msg_Success,
			ID:   int(this._gen.BruteCnf().MsgId),
			Time: 0,
		})
		this._timer, err = this._tf.TestCtx().TimerAdd(this._gen.BruteCnf().Timeout,
			this.OnTimerNext,
			nil)
	} else {
		statistic.Record(&statistic.StatisticEvent{
			Type: statistic.StatisticType_Msg_Success,
			ID:   int(this._gen.BruteCnf().MsgId),
			Time: 0,
		})
		this._timer, err = this._tf.TestCtx().TimerAdd(10,
			this.OnTimerNext,
			nil)
	}
}

func (this *BruteTest) OnTimerNext(t *common.TimeInterface, data any) int64 {
	this._timer = nil

	this.msgSend()
	return 0
}

func (this *BruteTest) OnTimerResponseTimeout(t *common.TimeInterface, data any) int64 {
	this._timer = nil
	log.LogInfo("MsgResponse Timeout. %s action:%d OnMsg:%d index:%d timeOut:%d data:%s",
		this._tf.LogPre(),
		this._id,
		this._gen.BruteCnf().ResponeId,
		this._index-1,
		this._gen.BruteCnf().Timeout,
		this._data)

	statistic.Record(&statistic.StatisticEvent{
		Type: statistic.StatisticType_Msg_Timeout,
		ID:   int(this._gen.BruteCnf().ResponeId),
	})

	this.msgSend()
	return 0
}