Add queue to RunnerRegister (#607)

Author: jasonopslevel

job.go

@@ -124,14 +124,19 @@ type RunnerScale struct {
 	RecommendedReplicaCount int `json:"recommendedReplicaCount"`
 }
 
-func (client *Client) RunnerRegister() (*Runner, error) {
+func (client *Client) RunnerRegister(queue ...string) (*Runner, error) {
 	var m struct {
 		Payload struct {
 			Runner Runner
 			Errors []Error
-		} `graphql:"runnerRegister"`
+		} `graphql:"runnerRegister(queue: $queue)"`
 	}
 	v := PayloadVariables{}
+	if len(queue) > 0 {
+		v["queue"] = &queue[0]
+	} else {
+		v["queue"] = (*string)(nil)
+	}
 	err := client.Mutate(&m, v, WithName("RunnerRegister"))
 	return &m.Payload.Runner, HandleErrors(err, m.Payload.Errors)
 }

job_test.go

@@ -10,8 +10,8 @@ import (
 func TestRunnerRegister(t *testing.T) {
 	// Arrange
 	testRequest := autopilot.NewTestRequest(
-		`mutation RunnerRegister{runnerRegister{runner{id,status},errors{message,path}}}`,
-		`{}`,
+		`mutation RunnerRegister($queue:String){runnerRegister(queue: $queue){runner{id,status},errors{message,path}}}`,
+		`{"queue": null}`,
 		`{"data": {"runnerRegister": { "runner": { "id": "1234", "status": "registered" }, "errors": [] }}}`,
 	)
 
@@ -23,6 +23,22 @@ func TestRunnerRegister(t *testing.T) {
 	autopilot.Equals(t, ol.ID("1234"), result.Id)
 }
 
+func TestRunnerRegisterWithQueue(t *testing.T) {
+	// Arrange
+	testRequest := autopilot.NewTestRequest(
+		`mutation RunnerRegister($queue:String){runnerRegister(queue: $queue){runner{id,status},errors{message,path}}}`,
+		`{"queue": "my-queue"}`,
+		`{"data": {"runnerRegister": { "runner": { "id": "1234", "status": "registered" }, "errors": [] }}}`,
+	)
+
+	client := BestTestClient(t, "job/register_with_queue", testRequest)
+	// Act
+	result, err := client.RunnerRegister("my-queue")
+	// Assert
+	autopilot.Ok(t, err)
+	autopilot.Equals(t, ol.ID("1234"), result.Id)
+}
+
 func TestRunnerGetScale(t *testing.T) {
 	// Arrange
 	testRequest := autopilot.NewTestRequest(