Allow for arbitrary tags when specifying "agent"

adikari · Mar 21, 2022 · 6421eda · 6421eda
1 parent 40e837e
commit 6421eda
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/plugin.go b/plugin.go
@@ -48,9 +48,7 @@ type Step struct {
 }
 
 // Agent is Buildkite agent definition
-type Agent struct {
-	Queue string `yaml:"queue,omitempty"`
-}
+type Agent map[string]string
 
 // Build is buildkite build definition
 type Build struct {

diff --git a/plugin.yml b/plugin.yml
@@ -46,6 +46,8 @@ configuration:
               properties:
                 queue:
                   type: string
+              additionalProperties:
+                type: string
             artifacts:
               type: array
             env:

diff --git a/plugin_test.go b/plugin_test.go
@@ -97,7 +97,8 @@ func TestPluginShouldUnmarshallCorrectly(t *testing.T) {
 						},
 						"async": true,
 						"agents": {
-							"queue": "queue-1"
+							"queue": "queue-1",
+							"database": "postgres"
 						},
 						"artifacts": [ "artifiact-1" ]
 					}
@@ -170,7 +171,7 @@ func TestPluginShouldUnmarshallCorrectly(t *testing.T) {
 						},
 					},
 					Async:     true,
-					Agents:    Agent{Queue: "queue-1"},
+					Agents:    map[string]string{"queue": "queue-1", "database": "postgres"},
 					Artifacts: []string{"artifiact-1"},
 				},
 			},