Using default platform when platform is ANY (fixes aerokube#256)

PYasonau · Dec 15, 2018 · e605e19 · e605e19
1 parent 8957bfa
commit e605e19
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/config.go b/config.go
@@ -45,14 +45,16 @@ type ggrBrowsers struct {
 	Browsers
 }
 
+const anyPlatform = "ANY"
+
 func (b *ggrBrowsers) find(browser, version string, platform string, excludedHosts set, excludedRegions set) (Hosts, string, set) {
 	var hosts Hosts
 	for _, b := range b.Browsers.Browsers {
 		if b.Name == browser {
 			if version == "" {
 				version = b.DefaultVersion
 			}
-			if platform == "" {
+			if platform == "" || platform == anyPlatform {
 				platform = b.DefaultPlatform
 			}
 			for _, v := range b.Versions {

diff --git a/config_test.go b/config_test.go
@@ -145,6 +145,13 @@ func TestFindWithDefaultPlatform(t *testing.T) {
 	AssertThat(t, hosts[0].Name, EqualTo{"browser-2.0-linux"})
 }
 
+func TestFindWithAnyPlatform(t *testing.T) {
+	hosts, version, _ := browsersWithMultiplePlatforms.find("browser", "2.0", "ANY", newSet(), newSet())
+	AssertThat(t, version, EqualTo{"2.0"})
+	AssertThat(t, len(hosts), EqualTo{1})
+	AssertThat(t, hosts[0].Name, EqualTo{"browser-2.0-linux"})
+}
+
 func TestFindWithPlatform(t *testing.T) {
 	hosts, version, _ := browsersWithMultiplePlatforms.find("browser", "2.0", "LINUX", newSet(), newSet())
 	AssertThat(t, version, EqualTo{"2.0"})