feat: add environment variable of build cluster name (#451)

launch.go

@@ -635,6 +635,7 @@ func launch(api screwdriver.API, buildID int, rootDir, emitterPath, metaSpace, s
 		"SD_STORE_URL":            fmt.Sprintf("%s/%s/", storeURL, "v1"),
 		"SD_UI_URL":               fmt.Sprintf("%s/", uiURL),
 		"SD_UI_BUILD_URL":         fmt.Sprintf("%s/pipelines/%s/builds/%s", uiURL, strconv.Itoa(job.PipelineID), strconv.Itoa(buildID)),
+		"SD_BUILD_CLUSTER_NAME":   build.BuildClusterName,
 		"SD_TOKEN":                buildToken,
 		"SD_CACHE_STRATEGY":       cacheStrategy,
 		"SD_PIPELINE_CACHE_DIR":   pipelineCacheDir,

launch_test.go

@@ -40,6 +40,7 @@ const (
 	TestStoreURL         = "https://store.screwdriver.cd"
 	TestUIURL            = "https://screwdriver.cd"
 	TestBuildToken       = "foobar"
+	TestBuildClusterName = "test-build-cluster-name"
 
 	TestScmURI = "github.com:123456:master"
 	TestSHA    = "abc123"
@@ -863,6 +864,7 @@ func TestSetEnv(t *testing.T) {
 		"SD_UI_URL":              "https://screwdriver.cd/",
 		"SD_UI_BUILD_URL":        "https://screwdriver.cd/pipelines/3456/builds/1234",
 		"SD_TOKEN":               "foobar",
+		"SD_BUILD_CLUSTER_NAME":  "",
 		"SD_SONAR_AUTH_URL":      "https://api.screwdriver.cd/v4/coverage/token",
 		"SD_SONAR_HOST":          "https://sonar.screwdriver.cd",
 		"SD_PIPELINE_CACHE_DIR":  "",
@@ -958,6 +960,23 @@ func TestSetEnv(t *testing.T) {
 		}
 	}
 
+	// in case of the build cluster exists
+	api.buildFromID = func(buildID int) (screwdriver.Build, error) {
+		return screwdriver.Build(FakeBuild{ID: TestBuildID, EventID: TestEventID, JobID: TestJobID, BuildClusterName: TestBuildClusterName, SHA: TestSHA, ParentBuildID: float64(1234)}), nil
+	}
+	tests["SD_BUILD_CLUSTER_NAME"] = TestBuildClusterName
+	TestEnvVars = map[string]interface{}{}
+	foundEnv = map[string]string{}
+	err, _, _ = launch(screwdriver.API(api), TestBuildID, TestWorkspace, TestEmitter, TestMetaSpace, TestStoreURL, TestUIURL, TestShellBin, TestBuildTimeout, TestBuildToken, "", "", "", "", false, false, false, 0, 10000)
+	if err != nil {
+		t.Fatalf("Unexpected error from launch: %v", err)
+	}
+	for k, v := range tests {
+		if foundEnv[k] != v {
+			t.Fatalf("foundEnv[%s] = %s, want %s", k, foundEnv[k], v)
+		}
+	}
+
 	// in case of the pipeline is private
 	TestPrivateScmRepo := screwdriver.ScmRepo(FakeScmRepo{
 		Name:    "screwdriver-cd/launcher",

screwdriver/screwdriver.go

@@ -184,6 +184,7 @@ type Build struct {
 	Stats         struct {
 		QueueEntertime string `json:"queueEnterTime"`
 	} `json:"stats"`
+	BuildClusterName string `json:"buildClusterName"`
 }
 
 // Coverage is a Coverage object returned when getInfo is called

screwdriver/screwdriver_test.go

@@ -145,6 +145,7 @@ func TestBuildFromID(t *testing.T) {
 				}{
 					QueueEntertime: "2020-05-01T20:15:31.508Z",
 				},
+				BuildClusterName: "test-build-cluster-name",
 			},
 			statusCode: 200,
 			err:        nil,