diff --git a/cmd/robot/robot.go b/cmd/robot/robot.go index 41fdeccd2..76d2c8b7f 100644 --- a/cmd/robot/robot.go +++ b/cmd/robot/robot.go @@ -172,7 +172,7 @@ func (r *Robot) handleReq() { delete(r.builderMap, b.id) zlog.Errorf("send to client err:%v", err) } - b.requested = true + r.handleRsp(b.id) } @@ -233,10 +233,9 @@ func (r *Robot) handleRsp(id string) { r.handleNotify(uuid, msg) if v, ok := r.builderMap[uuid]; ok { - // if v.enabled && - // (msg.MainType == v.mainType && - // msg.SubType == v.subType) && - // v.requested { + if v.enabled && + (msg.MainType == v.mainType && + msg.SubType == v.subType) { v.hs = time.Since(v.start) if !comm.ProtoUnmarshal(msg, v.rsp) { @@ -266,7 +265,7 @@ func (r *Robot) handleRsp(id string) { } r.caseSuccess++ } - // } + } } func (r *Robot) findTestCase(msg *pb.UserMessage) (string, bool) {