GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-25 17:44:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add global AntiRisk configuration support

Browse Source 
- Add AntiRisk field to TConfig struct for global anti-risk switch
- Add SetAntiRisk method to configure global anti-risk setting
- Implement automatic AntiRisk application in mobile UI steps
- Global AntiRisk setting applies to all actions unless explicitly disabled
- Maintains backward compatibility with existing action-level AntiRisk settings
This commit is contained in:
lilong.129
2025-05-29 00:11:34 +08:00
parent 08a8b06578
commit c5fb391ef5
3 changed files with 48 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
config.go
View File

@@ -43,6 +43,7 @@ type TConfig struct {
Path string `json:"path,omitempty" yaml:"path,omitempty"` // testcase file path
PluginSetting *PluginConfig `json:"plugin,omitempty" yaml:"plugin,omitempty"` // plugin config
MCPConfigPath string `json:"mcp_config_path,omitempty" yaml:"mcp_config_path,omitempty"`
AntiRisk bool `json:"anti_risk,omitempty" yaml:"anti_risk,omitempty"` // global anti-risk switch
IgnorePopup bool `json:"ignore_popup,omitempty" yaml:"ignore_popup,omitempty"`
LLMService option.LLMServiceType `json:"llm_service,omitempty" yaml:"llm_service,omitempty"`
CVService option.CVServiceType `json:"cv_service,omitempty" yaml:"cv_service,omitempty"`
@@ -76,6 +77,12 @@ func (c *TConfig) SetVerifySSL(verify bool) *TConfig {
return c
}
// SetAntiRisk sets global anti-risk switch for current testcase.
func (c *TConfig) SetAntiRisk(antiRisk bool) *TConfig {
c.AntiRisk = antiRisk
return c
}
// WithParameters sets parameters for current testcase.
func (c *TConfig) WithParameters(parameters map[string]interface{}) *TConfig {
c.Parameters = parameters

2
internal/version/VERSION
View File

@@ -1 +1 @@
v5.0.0-beta-2505282311
v5.0.0-beta-2505290011

58
step_ui.go
View File

@@ -709,7 +709,7 @@ func runStepMobileUI(s *SessionRunner, step IStep) (stepResult *StepResult, err
}
uiDriver, err := uixt.GetOrCreateXTDriver(config)
if err != nil {
return
return nil, err
}
identifier := mobileStep.Identifier
@@ -741,25 +741,31 @@ func runStepMobileUI(s *SessionRunner, step IStep) (stepResult *StepResult, err
attachments["error"] = err.Error()
// save foreground app
startTime := time.Now()
actionResult := &ActionResult{
MobileAction: uixt.MobileAction{
Method: option.ACTION_GetForegroundApp,
Params: "[ForDebug] check foreground app",
},
StartTime: startTime.Unix(),
if uiDriver != nil {
startTime := time.Now()
actionResult := &ActionResult{
MobileAction: uixt.MobileAction{
Method: option.ACTION_GetForegroundApp,
Params: "[ForDebug] check foreground app",
},
StartTime: startTime.Unix(),
}
if app, err1 := uiDriver.ForegroundInfo(); err1 == nil {
attachments["foreground_app"] = app.AppBaseInfo
} else {
log.Warn().Err(err1).Msg("save foreground app failed, ignore")
}
actionResult.Elapsed = time.Since(startTime).Milliseconds()
stepResult.Actions = append(stepResult.Actions, actionResult)
}
if app, err1 := uiDriver.ForegroundInfo(); err1 == nil {
attachments["foreground_app"] = app.AppBaseInfo
} else {
log.Warn().Err(err1).Msg("save foreground app failed, ignore")
}
actionResult.Elapsed = time.Since(startTime).Milliseconds()
stepResult.Actions = append(stepResult.Actions, actionResult)
}
// automatic handling of pop-up windows on each step finished
if !ignorePopup && !s.caseRunner.Config.Get().IgnorePopup {
var config *TConfig
if s.caseRunner != nil && s.caseRunner.Config != nil {
config = s.caseRunner.Config.Get()
}
if !ignorePopup && (config == nil || !config.IgnorePopup) && uiDriver != nil {
startTime := time.Now()
actionResult := &ActionResult{
MobileAction: uixt.MobileAction{
@@ -776,8 +782,10 @@ func runStepMobileUI(s *SessionRunner, step IStep) (stepResult *StepResult, err
}
// save attachments
for key, value := range uiDriver.GetData(true) {
attachments[key] = value
if uiDriver != nil {
for key, value := range uiDriver.GetData(true) {
attachments[key] = value
}
}
stepResult.Attachments = attachments
stepResult.Elapsed = time.Since(start).Milliseconds()
@@ -806,6 +814,20 @@ func runStepMobileUI(s *SessionRunner, step IStep) (stepResult *StepResult, err
return stepResult, err
}
// Apply global AntiRisk configuration if enabled in testcase config
if s.caseRunner != nil && s.caseRunner.Config != nil {
config := s.caseRunner.Config.Get()
if config != nil && config.AntiRisk {
if action.Options == nil {
action.Options = &option.ActionOptions{}
}
// Only set AntiRisk to true if it's not already explicitly set to false
if !action.Options.AntiRisk {
action.Options.AntiRisk = true
}
}
}
// stat uixt action
if action.Method == option.ACTION_LOG {
log.Info().Interface("action", action.Params).Msg("stat uixt action")