change: add capabilities for driver

This commit is contained in:
debugtalk
2022-10-11 20:09:12 +08:00
parent 36b10cfc23
commit 7a0ce219e3
11 changed files with 34 additions and 43 deletions
+4 -4
View File
@@ -106,7 +106,7 @@ func (dev *AndroidDevice) UUID() string {
return dev.SerialNumber
}
func (dev *AndroidDevice) NewDriver() (driverExt *DriverExt, err error) {
func (dev *AndroidDevice) NewDriver(capabilities Capabilities) (driverExt *DriverExt, err error) {
var deviceOptions []AndroidDeviceOption
if dev.SerialNumber != "" {
deviceOptions = append(deviceOptions, WithSerialNumber(dev.SerialNumber))
@@ -122,11 +122,11 @@ func (dev *AndroidDevice) NewDriver() (driverExt *DriverExt, err error) {
if err != nil {
return nil, err
}
return androidDevice.InitUIAClient()
return androidDevice.initUIAClient(capabilities)
}
func (dev *AndroidDevice) InitUIAClient() (*DriverExt, error) {
driver, err := dev.NewUSBDriver(nil)
func (dev *AndroidDevice) initUIAClient(capabilities Capabilities) (*DriverExt, error) {
driver, err := dev.NewUSBDriver(capabilities)
if err != nil {
return nil, errors.Wrap(err, "failed to init UIA driver")
}