From 3a0db1020bcf95a93d908637074417bbadaea3c3 Mon Sep 17 00:00:00 2001
From: Ivan Krutov <vania-pooh@aerokube.com>
Date: Thu, 13 May 2021 17:38:11 +0500
Subject: [PATCH] Explicitly setting browserVersion in top-level capabilites
 only

---
 proxy.go      | 17 ++++++++++-------
 proxy_test.go |  6 ++++++
 2 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/proxy.go b/proxy.go
index 12c18db..f21f4a0 100644
--- a/proxy.go
+++ b/proxy.go
@@ -87,27 +87,27 @@ type Routes map[string]*Host
 
 type caps map[string]interface{}
 
-func (c caps) capabilities(fn func(m map[string]interface{}, w3c bool)) {
+func (c caps) capabilities(fn func(m map[string]interface{}, w3c bool, extension bool)) {
 	if desiredCapabilities, ok := c[keys.desiredCapabilities]; ok {
 		if m, ok := desiredCapabilities.(map[string]interface{}); ok {
-			fn(m, false)
+			fn(m, false, false)
 		}
 	}
 	if w3cCapabilities, ok := c[keys.w3cCapabilities]; ok {
 		if m, ok := w3cCapabilities.(map[string]interface{}); ok {
 			if alwaysMatch, ok := m[keys.alwaysMatch]; ok {
 				if m, ok := alwaysMatch.(map[string]interface{}); ok {
-					fn(m, true)
+					fn(m, true, false)
 					for k, v := range m { // Extension capabilities have ":" in key
 						if ec, ok := v.(map[string]interface{}); ok && strings.Contains(k, ":") {
-							fn(ec, true)
+							fn(ec, true, true)
 						}
 					}
 				}
 			}
 		}
 	} else {
-		fn(make(map[string]interface{}), false)
+		fn(make(map[string]interface{}), false, false)
 	}
 }
 
@@ -117,7 +117,7 @@ func (c caps) capability(k string) string {
 
 func (c caps) capabilityJsonWireW3C(jsonWire, W3C string) string {
 	result := ""
-	c.capabilities(func(m map[string]interface{}, w3c bool) {
+	c.capabilities(func(m map[string]interface{}, w3c bool, _ bool) {
 		k := jsonWire
 		if w3c {
 			k = W3C
@@ -162,7 +162,10 @@ func (c caps) labels() string {
 }
 
 func (c caps) setVersion(version string) {
-	c.capabilities(func(m map[string]interface{}, w3c bool) {
+	c.capabilities(func(m map[string]interface{}, w3c bool, extension bool) {
+		if extension {
+			return
+		}
 		if w3c {
 			m["browserVersion"] = version
 		} else {
diff --git a/proxy_test.go b/proxy_test.go
index 53c8e0a..a3be70d 100644
--- a/proxy_test.go
+++ b/proxy_test.go
@@ -886,6 +886,12 @@ func TestStartSessionWithDefaultVersionW3C(t *testing.T) {
 		AssertThat(t, err, Is{nil})
 		AssertThat(t, sess["capabilities"]["alwaysMatch"]["browserVersion"], EqualTo{"2.0"})
 
+		so, ok := sess["capabilities"]["alwaysMatch"]["selenoid:options"]
+		AssertThat(t, ok, Is{true})
+		selenoidOptions, ok := so.(map[string]interface{})
+		AssertThat(t, ok, Is{true})
+		_, ok = selenoidOptions["browserVersion"]
+		AssertThat(t, ok, Is{false})
 	}))
 	selenium := httptest.NewServer(mux)
 	defer selenium.Close()