change: wda ports are used preferentially from environment variables

hrp/internal/env/env.go

@@ -4,6 +4,8 @@ import "os"
 
 var (
 	WDA_USB_DRIVER       = os.Getenv("WDA_USB_DRIVER")
+	WDA_LOCAL_PORT       = os.Getenv("WDA_LOCAL_PORT")
+	WDA_LOCAL_MJPEG_PORT = os.Getenv("WDA_LOCAL_MJPEG_PORT")
 	VEDEM_OCR_URL        = os.Getenv("VEDEM_OCR_URL")
 	VEDEM_OCR_AK         = os.Getenv("VEDEM_OCR_AK")
 	VEDEM_OCR_SK         = os.Getenv("VEDEM_OCR_SK")

hrp/pkg/uixt/ios_device.go

@@ -13,6 +13,7 @@ import (
 	"net/url"
 	"os"
 	"regexp"
+	"strconv"
 	"strings"
 	"time"
 
@@ -384,16 +385,25 @@ func (dev *IOSDevice) perfOpitons() (perfOptions []giDevice.PerfOption) {
 
 // NewHTTPDriver creates new remote HTTP client, this will also start a new session.
 func (dev *IOSDevice) NewHTTPDriver(capabilities Capabilities) (driver WebDriver, err error) {
-	localPort, err := getFreePort()
+	var localPort int
+	localPort, err = strconv.Atoi(env.WDA_LOCAL_PORT)
+	if err != nil {
+		localPort, err = getFreePort()
 		if err != nil {
 			return nil, errors.Wrap(code.IOSDeviceHTTPDriverError,
 				fmt.Sprintf("get free port failed: %v", err))
 		}
+
 		if err = dev.forward(localPort, dev.Port); err != nil {
 			return nil, errors.Wrap(code.IOSDeviceHTTPDriverError,
 				fmt.Sprintf("forward tcp port failed: %v", err))
 		}
-	localMjpegPort, err := getFreePort()
+	}
+
+	var localMjpegPort int
+	localMjpegPort, err = strconv.Atoi(env.WDA_LOCAL_MJPEG_PORT)
+	if err != nil {
+		localMjpegPort, err = getFreePort()
 		if err != nil {
 			return nil, errors.Wrap(code.IOSDeviceHTTPDriverError,
 				fmt.Sprintf("get free port failed: %v", err))
@@ -402,6 +412,7 @@ func (dev *IOSDevice) NewHTTPDriver(capabilities Capabilities) (driver WebDriver
 			return nil, errors.Wrap(code.IOSDeviceHTTPDriverError,
 				fmt.Sprintf("forward tcp port failed: %v", err))
 		}
+	}
 
 	log.Info().Interface("capabilities", capabilities).
 		Int("localPort", localPort).Int("localMjpegPort", localMjpegPort).