minor fix for ci/cd

newrelic · Dec 5, 2024 · 1893619 · 1893619
1 parent 7e6a5df
commit 1893619
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/security_config/secure_config.go b/security_config/secure_config.go
@@ -7,7 +7,7 @@ type Security struct {
 	Enabled               bool   `json:"enabled"`
 	Mode                  string `json:"mode"`
 	Validator_service_url string `json:"validator_service_url"`
-	IastTestIdentifier    string `json:"iast_test_identifier"`
+	IastTestIdentifier    string `json:"iast_test_identifier" yaml:"iast_test_identifier"`
 	Agent                 struct {
 		Enabled bool `json:"enabled"`
 	} `json:"agent"`

diff --git a/security_handlers/web_socket.go b/security_handlers/web_socket.go
@@ -382,6 +382,8 @@ func getConnectionHeader() http.Header {
 
 	if env := secConfig.GlobalInfo.GetIastTestIdentifier(); env != "" {
 		connectionHeader.Add("NR-CSEC-IAST-TEST-IDENTIFIER", env)
+		connectionHeader.Add("NR-CSEC-IAST-SCAN-INSTANCE-COUNT", secUtils.IntToString(1))
+	} else {
 		connectionHeader.Add("NR-CSEC-IAST-SCAN-INSTANCE-COUNT", secUtils.IntToString(secConfig.GlobalInfo.ScanInstanceCount()))
 	}