feat: 优化ext getDriver和GetDevice策略

This commit is contained in:
余泓铮
2025-02-25 17:53:11 +08:00
parent ebdd676e61
commit 1094d1d95e
5 changed files with 15 additions and 12 deletions

View File

@@ -1,9 +1,8 @@
package server_ext
import (
"strings"
"github.com/gin-gonic/gin"
"strings"
"github.com/httprunner/httprunner/v5/pkg/uixt"
"github.com/httprunner/httprunner/v5/pkg/uixt/ai"
@@ -16,9 +15,9 @@ func (p RouterBaseMethodExt) GetDriver(c *gin.Context) (driverExt uixt.IXTDriver
serial := c.Param("serial")
deviceObj, exists := c.Get("device")
var device uixt.IDevice
var driver uixt.IDriver
var driver driver_ext.IStubDriver
if !exists {
device, err = server.GetDevice(c)
device, err = p.GetDevice(c)
if err != nil {
return nil, err
}