SE_NODE_RELAY_STEREOTYPE_EXTRA to append custom capabilities to default Relay stereotype

Signed-off-by: Viet Nguyen Duc <[email protected]>

Author: VietND96

ENV_VARIABLES.md

@@ -142,3 +142,5 @@
 | SE_SESSIONS_MAP_EXTERNAL_JDBC_USER |  |  |  |
 | SE_SESSIONS_MAP_EXTERNAL_PORT |  |  |  |
 | SE_SESSIONS_MAP_EXTERNAL_SCHEME |  |  |  |
+| SE_NODE_RELAY_STEREOTYPE |  |  |  |
+| SE_NODE_RELAY_STEREOTYPE_EXTRA |  |  |  |

Makefile

@@ -72,8 +72,8 @@ set_containerd_image_store:
 	docker info -f '{{ .DriverStatus }}'
 
 format_shell_scripts:
-	sudo apt-get update -qq ; \
-  sudo apt-get install -yq shfmt ; \
+	sudo apt-get update -qq || true ; \
+  sudo apt-get install -yq shfmt || true ; \
   shfmt -l -w -d $${PWD}/*.sh $${PWD}/**/*.sh $$PWD/**.sh $$PWD/**/generate_** $$PWD/**/wrap_* ; \
   git diff --stat --exit-code ; \
   EXIT_CODE=$$? ; \

NodeBase/generate_relay_config

@@ -17,8 +17,19 @@ if [[ -n "${SE_NODE_RELAY_URL}" ]]; then
 	if [[ -n "${SE_NODE_RELAY_PROTOCOL_VERSION}" ]]; then
 		echo "protocol-version = \"${SE_NODE_RELAY_PROTOCOL_VERSION}\"" >>"$FILENAME"
 	fi
-	echo "configs = [
-  \"${SE_NODE_RELAY_MAX_SESSIONS}\", \"{\\\"browserName\\\": \\\"${SE_NODE_RELAY_BROWSER_NAME}\\\", \\\"platformName\\\": \\\"${SE_NODE_RELAY_PLATFORM_NAME}\\\", \\\"appium:platformVersion\\\": \\\"${SE_NODE_RELAY_PLATFORM_VERSION}\\\"}\"
-  ]
-  " >>"$FILENAME"
+	if [[ -z "${SE_NODE_RELAY_STEREOTYPE}" ]]; then
+		SE_NODE_RELAY_STEREOTYPE="{\"browserName\": \"${SE_NODE_RELAY_BROWSER_NAME}\", \"platformName\": \"${SE_NODE_RELAY_PLATFORM_NAME}\", \"appium:platformVersion\": \"${SE_NODE_RELAY_PLATFORM_VERSION}\"}"
+	else
+		SE_NODE_RELAY_STEREOTYPE="${SE_NODE_RELAY_STEREOTYPE}"
+	fi
+	if [[ -n "${SE_NODE_RELAY_STEREOTYPE_EXTRA}" ]]; then
+		echo "Merging SE_NODE_RELAY_STEREOTYPE_EXTRA=${SE_NODE_RELAY_STEREOTYPE_EXTRA} to main relay stereotype"
+		SE_NODE_RELAY_STEREOTYPE="$(python3 /opt/bin/json_merge.py "${SE_NODE_RELAY_STEREOTYPE}" "${SE_NODE_RELAY_STEREOTYPE_EXTRA}")"
+		if [[ $? -ne 0 ]]; then
+			echo "Failed to merge SE_NODE_RELAY_STEREOTYPE_EXTRA. Please check the format of the JSON string. Keep using main relay stereotype."
+		else
+			echo "Merged relay stereotype: ${SE_NODE_RELAY_STEREOTYPE}"
+		fi
+	fi
+	echo "configs = ['${SE_NODE_RELAY_MAX_SESSIONS}', '${SE_NODE_RELAY_STEREOTYPE}']" >>"$FILENAME"
 fi

README.md

@@ -1180,7 +1180,7 @@ The following is an en example of configuration relay commands.
 If you want to relay commands only, `selenium/node-base` is suitable and lightweight for this purpose.
 In case you want to configure node with both browsers and relay commands, respective node images can be used.
 
-To use environment variables for generate relay configs, set `SE_NODE_RELAY_URL` and other variables as below
+To use environment variables for generate relay configs, set `SE_NODE_RELAY_URL` and other variables as below. Those will be used to generate the default relay config in TOML format looks like below.
 
 ```toml
 [relay]
@@ -1190,6 +1190,10 @@ protocol-version = "${SE_NODE_RELAY_PROTOCOL_VERSION}"
 configs = [ '${SE_NODE_RELAY_MAX_SESSIONS}', '{"browserName": "${SE_NODE_RELAY_BROWSER_NAME}", "platformName": "${SE_NODE_RELAY_PLATFORM_NAME}", "appium:platformVersion": "${SE_NODE_RELAY_PLATFORM_VERSION}"}' ]
 ```
 
+Instead of input value for each environment variable to construct the default relay stereotype, you can use the `SE_NODE_RELAY_STEREOTYPE` environment variable to overwrite the default relay stereotype with your custom stereotype.
+
+In another case, if you want to retain the default relay stereotype and append additional capabilities, you can use the `SE_NODE_RELAY_STEREOTYPE_EXTRA` environment variable to set your capabilities. Those will be merged to the default relay stereotype.
+
 To run a sample test with the relayed node, you can clone the project and try below command:
 
 ```bash

scripts/generate_list_env_vars/description.yaml

@@ -431,3 +431,9 @@
 - name: SE_SESSIONS_MAP_EXTERNAL_SCHEME
   description: ''
   cli: ''
+- name: SE_NODE_RELAY_STEREOTYPE
+  description: ''
+  cli: ''
+- name: SE_NODE_RELAY_STEREOTYPE_EXTRA
+  description: ''
+  cli: ''

scripts/generate_list_env_vars/value.yaml

@@ -124,6 +124,10 @@
   default: ''
 - name: SE_NODE_RELAY_STATUS_ENDPOINT
   default: ''
+- name: SE_NODE_RELAY_STEREOTYPE
+  default: ''
+- name: SE_NODE_RELAY_STEREOTYPE_EXTRA
+  default: ''
 - name: SE_NODE_RELAY_URL
   default: ''
 - name: SE_NODE_SESSION_TIMEOUT

tests/docker-compose-v3-test-node-relay.yml

@@ -68,6 +68,7 @@ services:
       - SE_NODE_RELAY_PLATFORM_NAME=Android
       - SE_NODE_RELAY_PLATFORM_VERSION=${ANDROID_PLATFORM_API}
       - SE_NODE_RELAY_BROWSER_NAME=chrome
+      - SE_NODE_RELAY_STEREOTYPE_EXTRA={"myApp:version":"beta","myApp:publish":"public"}
       - SE_NODE_RELAY_WEB_VNC=ws://emulator:6080/websockify
 
   emulator:

tests/relay_config.toml

@@ -4,14 +4,14 @@ subscribe = "tcp://selenium-hub:4443"
 
 [node]
 session-timeout = "300"
-override-max-sessions = false
+override-max-sessions = true
 detect-drivers = false
 drain-after-session-count = 0
-max-sessions = 1
+max-sessions = 3
 
 [relay]
 url = "http://standalone:4444/wd/hub"
 status-endpoint = "/status"
 configs = [
-    '1', '{"browserName":"${BROWSER_NAME}","platformName":"linux"}'
+    '3', '{"browserName":"${BROWSER_NAME}","platformName":"linux"}'
 ]