automated headless tests

Author: adomurad

host/main.go

@@ -12,7 +12,30 @@ import (
 
 //export main
 func main() {
-	exitCode := roc.Main()
+	options := roc.Options{
+		SetupOnly:               false,
+		PrintBrowserVersionOnly: false,
+	}
+
+	// os.Args is a slice of strings
+	// os.Args[0] is the name of the program
+	// os.Args[1:] contains the command-line arguments
+	args := os.Args[1:]
+
+	for _, arg := range args {
+		// fmt.Printf("arg: %s", arg)
+		switch arg {
+		case "--setup":
+			options.SetupOnly = true
+		case "--print-browser-version-only":
+			options.PrintBrowserVersionOnly = true
+		case "--headless":
+			// TODO - this will be provided by the Roc app
+			options.Headless = true
+		}
+	}
+
+	exitCode := roc.Main(options)
 
 	os.Exit(exitCode)
 }

host/roc/app.go

@@ -6,6 +6,7 @@ import "C"
 import (
 	"fmt"
 	"host/driversetup"
+	"host/setup"
 	"host/utils"
 	"host/webdriver"
 	"os"
@@ -14,15 +15,36 @@ import (
 	"unsafe"
 )
 
-func Main() int {
-	// fmt.Println(utils.FG_BLUE + "=============================" + utils.RESET)
-	// fmt.Println(utils.FG_BLUE + "============SETUP============" + utils.RESET)
-	err := setup()
+type Options struct {
+	SetupOnly               bool
+	PrintBrowserVersionOnly bool
+	Headless                bool
+}
+
+// TODO change when passing more than 1 value from Roc app is possible
+var headless = false
+
+func Main(options Options) int {
+	if options.PrintBrowserVersionOnly {
+		fmt.Printf("%s", setup.BrowserVersion)
+		return 0
+	}
+
+	if options.Headless {
+		headless = true
+	}
+
+	err := driversetup.DownloadChromeAndDriver()
 	if err != nil {
 		fmt.Println(utils.FG_RED+"Setup failed with: "+utils.RESET, err)
 		return 1
 	}
 
+	if options.SetupOnly {
+		fmt.Println("Browser and driver ready.")
+		return 0
+	}
+
 	cmd, err := driversetup.RunChromedriver()
 	if err != nil {
 		// todo
@@ -37,10 +59,6 @@ func Main() int {
 		return 1
 	}
 
-	// fmt.Println(utils.FG_BLUE + "Driver and Browser are ready." + utils.RESET)
-	// fmt.Println(utils.FG_BLUE + "=============================" + utils.RESET)
-	// fmt.Print("\n\n")
-
 	size := C.roc__mainForHost_1_exposed_size()
 	capturePtr := roc_alloc(size, 0)
 	defer roc_dealloc(capturePtr, 0)
@@ -88,18 +106,13 @@ func roc_fx_wait(timeout int64) C.struct_ResultVoidStr {
 	return createRocResultStr(RocOk, "")
 }
 
-func setup() error {
-	err := driversetup.DownloadChromeAndDriver()
-	if err != nil {
-		return err
-	}
-
-	return nil
-}
-
 //export roc_fx_startSession
 func roc_fx_startSession() C.struct_ResultVoidStr {
-	sessionId, err := webdriver.CreateSession()
+	serverOptions := webdriver.SessionOptions{
+		Headless: headless,
+	}
+
+	sessionId, err := webdriver.CreateSession(serverOptions)
 
 	if err != nil {
 		fmt.Println("error value ")

host/setup/setup.go

@@ -15,11 +15,17 @@ type BrowserPaths struct {
 	DriverDirPath  string
 }
 
+var (
+	// TODO - this will be passed from the roc program
+	BrowserVersion = fmt.Sprintf("Chrome-%s", ChromeVersion)
+	ChromeVersion  = "117.0.5846.0"
+)
+
 func GetChromePaths() (*BrowserPaths, error) {
 	os := fmt.Sprintf("%s-%s", runtime.GOOS, runtime.GOARCH)
 
 	browserFilesDir := "browser_files"
-	chromeVersion := "117.0.5846.0"
+	chromeVersion := ChromeVersion
 	var osName string
 
 	switch os {

host/webdriver/webdriver.go

@@ -21,25 +21,36 @@ type CreateSession_Response struct {
 	Value CreateSession_ResponseValue `json:"value"`
 }
 
-func CreateSession() (string, error) {
+type SessionOptions struct {
+	// TODO - props like window-size, timeouts, etc passed down from the  Roc app
+	Headless bool
+}
+
+func CreateSession(options SessionOptions) (string, error) {
 	url := fmt.Sprintf("%s/session", baseUrl)
 	paths, err := setup.GetChromePaths()
 	if err != nil {
 		return "", err
 	}
 
+	headlessSwtich := ""
+	if options.Headless {
+		headlessSwtich = ", \"--headless\""
+	}
+
+	// TODO parametrize this when passing more data from Roc app is possible
 	jsonData := []byte(fmt.Sprintf(`{
 		"capabilities": {
 			"firstMatch": [
 				{
 					"goog:chromeOptions": {
 						"binary": "%s",
-            "args": ["--window-size=1920,1080"]
+            "args": ["--window-size=1920,1080"%s]
 					}
 				}
 			]
 		}
-	}`, paths.BrowserPath))
+	}`, paths.BrowserPath, headlessSwtich))
 
 	var response CreateSession_Response
 	err = makeHttpRequest("POST", url, bytes.NewBuffer(jsonData), &response)