device_test.go

package giDevice

import (
	"fmt"
	"log"
	"os"
	"os/signal"
	"testing"
	"time"
)

var dev Device

func setupDevice(t *testing.T) {
	setupUsbmux(t)
	devices, err := um.Devices()
	if err != nil {
		t.Fatal(err)
	}

	if len(devices) == 0 {
		t.Fatal("No Device")
	}

	dev = devices[0]
}
func Test_device_ReadPairRecord(t *testing.T) {
	setupDevice(t)

	pairRecord, err := dev.ReadPairRecord()
	if err != nil {
		t.Fatal(err)
	}

	t.Log(pairRecord.HostID, pairRecord.SystemBUID, pairRecord.WiFiMACAddress)
}

func Test_device_NewConnect(t *testing.T) {
	setupDevice(t)

	if _, err := dev.NewConnect(LockdownPort); err != nil {
		t.Fatal(err)
	}
}

func Test_device_DeletePairRecord(t *testing.T) {
	setupDevice(t)

	if err := dev.DeletePairRecord(); err != nil {
		t.Fatal(err)
	}

}

func Test_device_SavePairRecord(t *testing.T) {
	setupLockdownSrv(t)

	pairRecord, err := lockdownSrv.Pair()
	if err != nil {
		t.Fatal(err)
	}

	err = dev.SavePairRecord(pairRecord)
	if err != nil {
		t.Fatal(err)
	}
}

func Test_device_XCTest(t *testing.T) {
	setupLockdownSrv(t)

	bundleID = "com.DataMesh.CheckList"
	out, cancel, err := dev.XCTest(bundleID)
	// out, cancel, err := dev.XCTest(bundleID, WithXCTestEnv(map[string]interface{}{"USE_PORT": 8222, "MJPEG_SERVER_PORT": 8333}))
	if err != nil {
		t.Fatal(err)
	}

	done := make(chan os.Signal, 1)
	signal.Notify(done, os.Interrupt)

	go func() {
		for s := range out {
			fmt.Print(s)
		}
		done <- os.Interrupt
	}()

	for {
		select {
		case <-done:
			cancel()
			fmt.Println()
			t.Log("DONE")
			return
		}
	}
}

func Test_device_AppInstall(t *testing.T) {
	setupLockdownSrv(t)

	ipaPath := "/private/tmp/derivedDataPath/Build/Products/Release-iphoneos/WebDriverAgentRunner-Runner.ipa"
	err := dev.AppInstall(ipaPath)
	if err != nil {
		t.Fatal(err)
	}
}

func Test_device_AppUninstall(t *testing.T) {
	setupLockdownSrv(t)

	bundleID = "com.leixipaopao.WebDriverAgentRunner.xctrunner"
	err := dev.AppUninstall(bundleID)
	if err != nil {
		t.Fatal(err)
	}
}

func Test_device_Syslog(t *testing.T) {
	setupLockdownSrv(t)

	dev.SyslogStop()

	lines, err := dev.Syslog()
	if err != nil {
		t.Fatal(err)
	}

	done := make(chan os.Signal, 1)

	go func() {
		for line := range lines {
			fmt.Println(line)
		}
		done <- os.Interrupt
		t.Log("DONE!!!")
	}()

	signal.Notify(done, os.Interrupt, os.Kill)

	// <-done
	time.Sleep(3 * time.Second)
	dev.SyslogStop()
	time.Sleep(200 * time.Millisecond)
}

func Test_device_Reboot(t *testing.T) {
	setupDevice(t)
	dev.Reboot()
}

func Test_device_Shutdown(t *testing.T) {
	setupDevice(t)
	dev.Shutdown()
}

func Test_device_PowerInfo(t *testing.T) {
	setupDevice(t)
	data, err := dev.PowerSource()
	if err != nil {
		panic(err)
	}
	log.Println(data)
}

func Test_device_InstallationProxyBrowse(t *testing.T) {
	setupDevice(t)

	list, err := dev.InstallationProxyBrowse(
		WithApplicationType(ApplicationTypeUser),
		WithReturnAttributes("CFBundleDisplayName", "CFBundleIdentifier", "SequenceNumber", "SequenceNumber"),
	)
	// list, err := dev.InstallationProxyBrowse()
	if err != nil {
		t.Fatal(err)
	}

	t.Log(len(list))

	for _, l := range list {
		t.Logf("%#v", l)
	}
}

func Test_device_share(t *testing.T) {
	setupDevice(t)
	SetDebug(true, true)
	err := dev.Share(9123)
	if err != nil {
		log.Panic(err)
	}
}

func Test_device_connect(t *testing.T) {
	SetDebug(true, true)
	remoteDev, _ := NewRemoteConnect("127.0.0.1", 9123, 30)
	data, err := remoteDev.GetValue("", "ProductVersion")
	if err != nil {
		log.Panic(err)
	}
	log.Println(data)
}

func TestMockPerfdUser1(t *testing.T) {
	//SetDebug(true, true)
	remoteDev, _ := NewRemoteConnect("127.0.0.1", 9123, 30)
	data, err := remoteDev.PerfStart(
		WithPerfSystemCPU(true),
		WithPerfSystemMem(true),
		WithPerfSystemDisk(true),
		WithPerfSystemNetwork(true),
		WithPerfNetwork(true),
		WithPerfFPS(true),
		WithPerfGPU(true),
		//WithPerfProcessAttributes("cpuUsage", "memAnon"),
		//WithPerfBundleID("com.apple.mobilesafari"),
	)
	if err != nil {
		t.Fatal(err)
	}

	timer := time.NewTimer(time.Duration(time.Second * 20))
	for {
		select {
		case <-timer.C:
			dev.PerfStop()
			return
		case d := <-data:
			fmt.Println(string(d))
		}
	}
}

func TestMockPerfdUser2(t *testing.T) {
	//SetDebug(true, true)
	remoteDev, _ := NewRemoteConnect("127.0.0.1", 9123, 30)
	data, err := remoteDev.PerfStart(
		WithPerfSystemCPU(true),
		WithPerfSystemMem(true),
		WithPerfSystemDisk(true),
		WithPerfSystemNetwork(true),
		WithPerfNetwork(true),
		WithPerfFPS(true),
		WithPerfGPU(true),
		//WithPerfProcessAttributes("cpuUsage", "memAnon"),
		//WithPerfBundleID("com.apple.mobilesafari"),
	)
	if err != nil {
		t.Fatal(err)
	}

	timer := time.NewTimer(time.Duration(time.Second * 10))
	for {
		select {
		case <-timer.C:
			dev.PerfStop()
			return
		case d := <-data:
			fmt.Println(string(d))
		}
	}
}