gitee.com/KyleChenSource/lib-robot@v1.0.2/robottest/testcaseimp/brutetest_task.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 (
	ctx_register_key = "action_brutetesttask"
)

type BruteTestTaskFactory struct {
}

type BruteTestTaskCnf struct {
	testcase.TestcaseActionConfig_
}

func (this *BruteTestTaskFactory) Cnf() testcase.TestcaseActionCnf {
	return &BruteTestTaskCnf{}
}

func (this *BruteTestTaskFactory) New(id testcase.TestActionID, tf testcase.ITestFlow, cnf testcase.TestcaseActionCnf) (testcase.ITestcaseAction, error) {
	b, ok := tf.TestCtx().Get(ctx_register_key)
	if ok {
		brute := b.(*BruteTestTask)
		brute._id = id
		brute._tf = tf

		return b.(*BruteTestTask), nil
	}

	n := &BruteTestTask{
		_id:  id,
		_tf:  tf,
		_cnf: cnf.(*BruteTestTaskCnf),
	}
	tf.TestCtx().Set(ctx_register_key, n)
	return n, nil
}

func init() {
	testcase.TESTCASEACTION_MGR.Register("BruteTestTask", &BruteTestTaskFactory{})
}

type BruteTestTask struct {
	_id  testcase.TestActionID
	_tf  testcase.ITestFlow
	_cnf *BruteTestTaskCnf

	_task *brutetest.BrutetestTask

	_brutemsgcnf []*brutetest.BruteMsgCnf
	_gen         *brutetest.BruteMsgGenerator
	_index       int // 当前测试流的索引id
	_logname     string
	_cnt         int // 当前测试流的索引id
	_data        protos.JsonString
	_timeSend    int64 // 发送时间
	_timer       *common.TimeInterface
}

func (this *BruteTestTask) taskGet() bool {
	task, ok := <-brutetest.BrutetestTaskChan
	if !ok {
		// 通道关闭，结束
		return false
	}

	this._brutemsgcnf, _ = brutetest.BrutetestCaselistMap[task.Cid]
	this._index = len(this._brutemsgcnf) - 1
	this._task = task

	this.genGet()

	log.LogInfo("%s %s BruteTestTask gen:%d cid:%d skip:%d limit:%d",
		this._tf.LogPre(),
		this._logname,
		len(this._brutemsgcnf),
		task.Cid,
		task.Skip,
		task.Limit)

	return true
}

func (this *BruteTestTask) genGet() {
	if this._index == 0 {
		this._gen = this._brutemsgcnf[this._index].Generator(this._task.Skip, this._task.Limit)
		this._logname = fmt.Sprintf("%s_%s_fin_%d(%d)", this._cnf.Name, this._brutemsgcnf[this._index].MsgName, this._task.Skip, this._task.Limit)
	} else {
		this._gen = this._brutemsgcnf[this._index].Generator(0, 0)
		this._logname = fmt.Sprintf("%s_%s_mid_%d(%d)", this._cnf.Name, this._brutemsgcnf[this._index].MsgName, len(this._brutemsgcnf)-this._index-1, len(this._brutemsgcnf))
	}

	log.LogInfo("%s %s BruteTestTask gen:%d msgsendCnt:%d",
		this._tf.LogPre(),
		this._logname,
		len(this._brutemsgcnf),
		this._cnt)
}

// 启动
// 目前仅仅支持定时器和消息两种驱动模式
func (this *BruteTestTask) Start() error {
	if this._task == nil {
		ok := this.taskGet()
		if !ok {
			// 通道关闭，结束
			this._tf.OnActionEnd(this, nil, this._cnf.FailContinue)
			return nil
		}
	} else if len(this._brutemsgcnf) > 1 {
		this._index = len(this._brutemsgcnf) - 1
		this.genGet()
	}

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

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

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

func (this *BruteTestTask) Name() string {
	return this._logname
}

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

func (this *BruteTestTask) OnMsg(protoId protos.ProtoID, header protos.JsonString, data protos.JsonString) {
	tCost := time.Now().UnixMilli() - this._timeSend
	log.LogInfo("MsgResponse OK. %s action:%s OnMsg:%d index:%d header:%s data:%s timeCost:%d",
		this._tf.LogPre(),
		this._logname,
		protoId,
		this._cnt-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 *BruteTestTask) msgSend() {
	ok, data := this._gen.Next()
	if !ok {
		this._index--
		if this._index < 0 {
			// 结束了
			ok = this.taskGet()
			if !ok {
				this._tf.OnActionEnd(this, nil, this._cnf.FailContinue)
				return
			}
		} else {
			this.genGet()
		}

		if this._gen == nil {
			this._tf.OnActionEnd(this, fmt.Errorf("BruteTestTask len:%d cid:%d skip:%d limit:%d index:%d gen nil",
				len(this._brutemsgcnf),
				this._task.Cid,
				this._task.Skip,
				this._task.Limit,
				this._index), this._cnf.FailContinue)
			return
		}

		ok, data = this._gen.Next()
		if !ok {
			this._tf.OnActionEnd(this, fmt.Errorf("BruteTestTask len:%d cid:%d skip:%d limit:%d index:%d gen first nil",
				len(this._brutemsgcnf),
				this._task.Cid,
				this._task.Skip,
				this._task.Limit,
				this._index), 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:%s msgSend:%d(%s) index:%d left:%d data:%s",
		this._tf.LogPre(),
		this._logname,
		this._gen.BruteCnf().MsgId,
		this._gen.BruteCnf().MsgName,
		this._cnt,
		this._task.Limit,
		data,
	)
	this._cnt++

	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 *BruteTestTask) OnTimerNext(t *common.TimeInterface, data any) int64 {
	this._timer = nil

	this.msgSend()
	return 0
}

func (this *BruteTestTask) OnTimerResponseTimeout(t *common.TimeInterface, data any) int64 {
	this._timer = nil
	log.LogInfo("MsgResponse Timeout. %s action:%s OnMsg:%d index:%d timeOut:%d data:%s",
		this._tf.LogPre(),
		this._logname,
		this._gen.BruteCnf().ResponeId,
		this._cnt-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
}