diff --git a/pkg/uixt/android_device.go b/pkg/uixt/android_device.go index ae1f3535..5d60dca9 100644 --- a/pkg/uixt/android_device.go +++ b/pkg/uixt/android_device.go @@ -111,7 +111,6 @@ func (dev *AndroidDevice) Setup() error { if err != nil { return errors.Wrap(code.DeviceShellExecError, err.Error()) } - return nil } @@ -339,6 +338,15 @@ func (dev *AndroidDevice) GetPackageInfo(packageName string) (types.AppInfo, err return appInfo, nil } +func (dev *AndroidDevice) ScreenShot() (*bytes.Buffer, error) { + raw, err := dev.Device.ScreenCap() + if err != nil { + return nil, errors.Wrapf(code.DeviceScreenShotError, + "adb screencap failed %v", err) + } + return bytes.NewBuffer(raw), nil +} + func (dev *AndroidDevice) GetAppInfo(packageName string) (app types.AppInfo, err error) { packageInfo, err := dev.RunShellCommand( "CLASSPATH=/data/local/tmp/evalite", "app_process", "/",