mirror of
https://github.com/httprunner/httprunner.git
synced 2026-07-10 23:12:41 +08:00
feat: 支持获取剪贴板
This commit is contained in:
@@ -4,6 +4,7 @@ import (
|
|||||||
"bufio"
|
"bufio"
|
||||||
"bytes"
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
|
"encoding/base64"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"encoding/xml"
|
"encoding/xml"
|
||||||
"fmt"
|
"fmt"
|
||||||
@@ -1166,6 +1167,59 @@ func (ad *ADBDriver) ClearFiles(paths ...string) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ad *ADBDriver) GetPasteboard() (content string, err error) {
|
||||||
|
/**
|
||||||
|
adb shell am broadcast -n io.appium.settings/.receivers.ClipboardReceiver -a io.appium.settings.clipboard.get
|
||||||
|
Broadcasting: Intent { act=io.appium.settings.clipboard.get flg=0x400000 cmp=io.appium.settings/.receivers.ClipboardReceiver }
|
||||||
|
Broadcast completed: result=-1, data="SEhHRw=="
|
||||||
|
**/
|
||||||
|
|
||||||
|
// Check and switch input method if needed, similar to SendUnicodeKeys
|
||||||
|
currentIme, err := ad.GetIme()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
// If current IME is not the required one, switch temporarily and restore later
|
||||||
|
if currentIme != option.UnicodeImePackageName {
|
||||||
|
defer func() {
|
||||||
|
_ = ad.SetIme(currentIme)
|
||||||
|
}()
|
||||||
|
|
||||||
|
err = ad.SetIme(option.UnicodeImePackageName)
|
||||||
|
if err != nil {
|
||||||
|
log.Warn().Err(err).Msgf("set Unicode Ime failed for clipboard operation")
|
||||||
|
// Continue anyway, might still work with current IME
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
res, err := ad.Device.RunShellCommand("am", "broadcast", "-n", option.AppiumSettingsPackageName+"/.receivers.ClipboardReceiver", "-a", option.AppiumSettingsPackageName+".clipboard.get")
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the response to extract the base64 encoded data
|
||||||
|
dataPrefix := "data=\""
|
||||||
|
dataIndex := strings.Index(res, dataPrefix)
|
||||||
|
if dataIndex == -1 {
|
||||||
|
return "", fmt.Errorf("clipboard data not found in response: %s", res)
|
||||||
|
}
|
||||||
|
|
||||||
|
dataStart := dataIndex + len(dataPrefix)
|
||||||
|
dataEnd := strings.Index(res[dataStart:], "\"")
|
||||||
|
if dataEnd == -1 {
|
||||||
|
return "", fmt.Errorf("malformed clipboard data in response: %s", res)
|
||||||
|
}
|
||||||
|
|
||||||
|
base64Data := res[dataStart : dataStart+dataEnd]
|
||||||
|
decodedData, err := base64.StdEncoding.DecodeString(base64Data)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("failed to decode clipboard content: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return string(decodedData), nil
|
||||||
|
}
|
||||||
|
|
||||||
type ExportPoint struct {
|
type ExportPoint struct {
|
||||||
Start int `json:"start" yaml:"start"`
|
Start int `json:"start" yaml:"start"`
|
||||||
End int `json:"end" yaml:"end"`
|
End int `json:"end" yaml:"end"`
|
||||||
|
|||||||
@@ -321,3 +321,10 @@ func TestDriver_UIA2_Input(t *testing.T) {
|
|||||||
err = driver.Input("123\n")
|
err = driver.Input("123\n")
|
||||||
assert.Nil(t, err)
|
assert.Nil(t, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDriver_ADB_GetPasteboard(t *testing.T) {
|
||||||
|
driver := setupADBDriverExt(t)
|
||||||
|
pasteboard, err := driver.IDriver.(*ADBDriver).GetPasteboard()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
t.Log(pasteboard)
|
||||||
|
}
|
||||||
|
|||||||
@@ -630,9 +630,13 @@ func (wd *WDADriver) SetPasteboard(contentType types.PasteboardType, content str
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (wd *WDADriver) GetPasteboard(contentType types.PasteboardType) (raw *bytes.Buffer, err error) {
|
func (wd *WDADriver) GetPasteboard() (raw *bytes.Buffer, err error) {
|
||||||
// [[FBRoute POST:@"/wda/getPasteboard"] respondWithTarget:self action:@selector(handleGetPasteboard:)]
|
// [[FBRoute POST:@"/wda/getPasteboard"] respondWithTarget:self action:@selector(handleGetPasteboard:)]
|
||||||
data := map[string]interface{}{"contentType": contentType}
|
err = wd.AppLaunch("com.gtf.wda.runner.xctrunner")
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "GetPasteboard failed. WDA app not launched")
|
||||||
|
}
|
||||||
|
data := map[string]interface{}{}
|
||||||
var rawResp DriverRawResponse
|
var rawResp DriverRawResponse
|
||||||
if rawResp, err = wd.Session.POST(data, "/wda/getPasteboard"); err != nil {
|
if rawResp, err = wd.Session.POST(data, "/wda/getPasteboard"); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|||||||
@@ -356,3 +356,10 @@ func TestDriver_WDA_PushImage(t *testing.T) {
|
|||||||
err = driver.ClearImages()
|
err = driver.ClearImages()
|
||||||
assert.Nil(t, err)
|
assert.Nil(t, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDriver_WDA_GetPasteboard(t *testing.T) {
|
||||||
|
driver := setupWDADriverExt(t)
|
||||||
|
pasteboard, err := driver.IDriver.(*WDADriver).GetPasteboard()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
t.Log(pasteboard)
|
||||||
|
}
|
||||||
|
|||||||
@@ -49,8 +49,9 @@ const (
|
|||||||
defaultUIA2ServerPackageName = "io.appium.uiautomator2.server"
|
defaultUIA2ServerPackageName = "io.appium.uiautomator2.server"
|
||||||
defaultUIA2ServerTestPackageName = "io.appium.uiautomator2.server.test"
|
defaultUIA2ServerTestPackageName = "io.appium.uiautomator2.server.test"
|
||||||
|
|
||||||
AdbKeyBoardPackageName = "com.android.adbkeyboard/.AdbIME"
|
AdbKeyBoardPackageName = "com.android.adbkeyboard/.AdbIME"
|
||||||
UnicodeImePackageName = "io.appium.settings/.UnicodeIME"
|
UnicodeImePackageName = AppiumSettingsPackageName + "/.UnicodeIME"
|
||||||
|
AppiumSettingsPackageName = "io.appium.settings"
|
||||||
)
|
)
|
||||||
|
|
||||||
func NewAndroidDeviceOptions(opts ...AndroidDeviceOption) *AndroidDeviceOptions {
|
func NewAndroidDeviceOptions(opts ...AndroidDeviceOption) *AndroidDeviceOptions {
|
||||||
|
|||||||
Reference in New Issue
Block a user