refactor: move uixt pkg

uixt/android_test.go

@@ -0,0 +1,236 @@
+//go:build localtest
+
+package uixt
+
+import (
+	"os"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/httprunner/httprunner/v5/uixt/ai"
+	"github.com/httprunner/httprunner/v5/uixt/option"
+	"github.com/httprunner/httprunner/v5/uixt/types"
+)
+
+func setupADBDriverExt(t *testing.T) *XTDriver {
+	device, err := NewAndroidDevice()
+	require.Nil(t, err)
+	device.Options.UIA2 = false
+	device.Options.LogOn = false
+	driver, err := device.NewDriver()
+	require.Nil(t, err)
+	return NewXTDriver(driver,
+		ai.WithCVService(ai.CVServiceTypeVEDEM))
+}
+
+func setupUIA2DriverExt(t *testing.T) *XTDriver {
+	device, err := NewAndroidDevice()
+	require.Nil(t, err)
+	device.Options.UIA2 = true // use uiautomator2 driver
+	device.Options.LogOn = false
+	driver, err := device.NewDriver()
+	require.Nil(t, err)
+	return NewXTDriver(driver,
+		ai.WithCVService(ai.CVServiceTypeVEDEM))
+}
+
+func TestDevice_Android_GetPackageInfo(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	appInfo, err := driver.GetDevice().GetPackageInfo("com.android.settings")
+	require.Nil(t, err)
+	t.Log(appInfo)
+	assert.Equal(t, "com.android.settings", appInfo.Name)
+	assert.NotEmpty(t, appInfo.AppPath)
+	assert.NotEmpty(t, appInfo.AppMD5)
+}
+
+func TestDevice_Android_GetCurrentWindow(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	driver.AppLaunch("com.android.settings")
+	windowInfo, err := driver.GetDevice().(*AndroidDevice).GetCurrentWindow()
+	require.Nil(t, err)
+	assert.Equal(t, "com.android.settings", windowInfo.PackageName)
+}
+
+func TestDriver_ADB_Session_TODO(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.InitSession(nil)
+	require.Nil(t, err)
+	err = driver.DeleteSession()
+	assert.Nil(t, err)
+}
+
+func TestDriver_ADB_Status_TODO(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	status, err := driver.Status()
+	require.Nil(t, err)
+	t.Log(status)
+}
+
+func TestDriver_ADB_ScreenShot(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	screenshot, err := driver.ScreenShot()
+	assert.Nil(t, err)
+	path, err := saveScreenShot(screenshot, "1234")
+	require.Nil(t, err)
+	defer os.Remove(path)
+	t.Logf("save screenshot to %s", path)
+}
+
+func TestDriver_ADB_Rotation_TODO(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	rotation, err := driver.Rotation()
+	require.Nil(t, err)
+	t.Logf("x = %d\ty = %d\tz = %d", rotation.X, rotation.Y, rotation.Z)
+}
+
+func TestDriver_ADB_DeviceSize(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	deviceSize, err := driver.WindowSize()
+	require.Nil(t, err)
+	assert.Greater(t, deviceSize.Width, 200)
+	assert.Greater(t, deviceSize.Height, 200)
+}
+
+func TestDriver_ADB_Source(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	source, err := driver.Source()
+	require.Nil(t, err)
+	assert.Contains(t, source, "<?xml version")
+	assert.Contains(t, source, "android.widget.TextView")
+	t.Log(source)
+}
+
+func TestDriver_ADB_BatteryInfo_TODO(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	batteryInfo, err := driver.BatteryInfo()
+	require.Nil(t, err)
+	t.Log(batteryInfo)
+}
+
+func TestDriver_ADB_DeviceInfo_TODO(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	devInfo, err := driver.DeviceInfo()
+	require.Nil(t, err)
+	t.Logf("api version: %s", devInfo.APIVersion)
+	t.Logf("platform version: %s", devInfo.PlatformVersion)
+	t.Logf("bluetooth state: %s", devInfo.Bluetooth.State)
+}
+
+func TestDriver_ADB_TapXY(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.TapXY(0.4, 0.5)
+	assert.Nil(t, err)
+}
+
+func TestDriver_ADB_TapAbsXY(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.TapAbsXY(100, 300)
+	assert.Nil(t, err)
+}
+
+func TestDriver_ADB_Swipe(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.Swipe(0.5, 0.7, 0.5, 0.5,
+		option.WithPressDuration(0.5))
+	assert.Nil(t, err)
+}
+
+func TestDriver_ADB_Drag(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.Drag(0.5, 0.7, 0.5, 0.5)
+	assert.Nil(t, err)
+}
+
+func TestDriver_ADB_Input(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.Input("Hi 你好\n",
+		option.WithIdentifier("test"))
+	assert.Nil(t, err)
+	time.Sleep(time.Second * 1)
+	err = driver.Input("123\n")
+	assert.Nil(t, err)
+}
+
+func TestDriver_ADB_PressBack(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.Back()
+	assert.Nil(t, err)
+}
+
+func TestDriver_ADB_SetRotation_TODO(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.SetRotation(types.Rotation{Z: 270})
+	assert.Nil(t, err)
+}
+
+func TestDriver_ADB_Orientation(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	orientation, err := driver.Orientation()
+	assert.Nil(t, err)
+	assert.Equal(t, types.OrientationPortrait, orientation)
+}
+
+func TestDriver_ADB_AppLaunchTerminate(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.AppLaunch("com.android.settings")
+	assert.Nil(t, err)
+	time.Sleep(1 * time.Second)
+	ok, err := driver.AppTerminate("com.android.settings")
+	assert.Nil(t, err)
+	assert.True(t, ok)
+}
+
+func TestDriver_ADB_ForegroundInfo(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.AppLaunch("com.android.settings")
+	assert.Nil(t, err)
+	app, err := driver.ForegroundInfo()
+	assert.Nil(t, err)
+	assert.Equal(t, "com.android.settings", app.PackageName)
+}
+
+func TestDriver_ADB_ScreenRecord(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	path, err := driver.ScreenRecord(5 * time.Second)
+	assert.Nil(t, err)
+	defer os.Remove(path)
+	t.Log(path)
+}
+
+func TestDriver_ADB_Backspace(t *testing.T) {
+	driver := setupADBDriverExt(t)
+	err := driver.Backspace(1)
+	assert.Nil(t, err)
+}
+
+func TestDriver_UIA2_TapXY(t *testing.T) {
+	driver := setupUIA2DriverExt(t)
+	driver.StartCaptureLog("tap_xy")
+	err := driver.TapXY(0.5, 0.5,
+		option.WithIdentifier("test"),
+		option.WithPressDuration(4))
+	assert.Nil(t, err)
+	result, _ := driver.StopCaptureLog()
+	t.Log(result)
+}
+
+func TestDriver_UIA2_Swipe(t *testing.T) {
+	driver := setupUIA2DriverExt(t)
+	err := driver.Swipe(0.5, 0.7, 0.5, 0.5,
+		option.WithPressDuration(0.5))
+	assert.Nil(t, err)
+}
+
+func TestDriver_UIA2_Input(t *testing.T) {
+	driver := setupUIA2DriverExt(t)
+	err := driver.Input("Hi 你好\n",
+		option.WithIdentifier("test"))
+	assert.Nil(t, err)
+	time.Sleep(time.Second * 1)
+	err = driver.Input("123\n")
+	assert.Nil(t, err)
+}