mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-15 12:27:59 +08:00
39 lines
802 B
Go
39 lines
802 B
Go
package ios
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var rebootCmd = &cobra.Command{
|
|
Use: "reboot",
|
|
Short: "reboot or shutdown ios device",
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {},
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
device, err := getDevice(udid)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if isShutdown {
|
|
err = device.Shutdown()
|
|
} else {
|
|
err = device.Reboot()
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Printf("reboot %s success\n", device.Properties().UDID)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var isShutdown bool
|
|
|
|
func init() {
|
|
rebootCmd.Flags().StringVarP(&udid, "udid", "u", "", "specify device by udid")
|
|
rebootCmd.Flags().BoolVarP(&isShutdown, "shutdown", "s", false, "shutdown ios device")
|
|
iosRootCmd.AddCommand(rebootCmd)
|
|
}
|