Set required URL and TOKEN env vars for agent config

cmd/otelcol/main.go

@@ -62,6 +62,7 @@ func main() {
 
 	if !contains(os.Args[1:], "-h") && !contains(os.Args[1:], "--help") {
 		checkRuntimeParams()
+		setDefaultEnvVars()
 	}
 
 	// Allow dumping configuration locally by default
@@ -286,6 +287,32 @@ func setMemoryLimit(memTotalSizeMiB int) int {
 	return memLimit
 }
 
+// Sets environment variables expected by agent_config.yaml if missing
+func setDefaultEnvVars() {
+	realm, realmOk := os.LookupEnv("SPLUNK_REALM")
+	if realmOk {
+		testArgs := [][]string{
+			{"SPLUNK_API_URL", "https://api." + realm + ".signalfx.com"},
+			{"SPLUNK_INGEST_URL", "https://ingest." + realm + ".signalfx.com"},
+			{"SPLUNK_TRACE_URL", "https://ingest." + realm + ".signalfx.com/v2/trace"},
+			{"SPLUNK_HEC_URL", "https://ingest." + realm + ".signalfx.com/v1/log"},
+		}
+		for _, v := range testArgs {
+			_, ok := os.LookupEnv(v[0])
+			if !ok {
+				os.Setenv(v[0], v[1])
+			}
+		}
+	}
+	token, tokenOk := os.LookupEnv("SPLUNK_ACCESS_TOKEN")
+	if tokenOk {
+		_, ok := os.LookupEnv("SPLUNK_HEC_TOKEN")
+		if !ok {
+			os.Setenv("SPLUNK_HEC_TOKEN", token)
+		}
+	}
+}
+
 // Returns a ParserProvider that reads configuration YAML from an environment variable when applicable.
 func newBaseParserProvider() parserprovider.ParserProvider {
 	_, configPathFlag := getKeyValue(os.Args[1:], "--config")

cmd/otelcol/main_test.go

@@ -112,6 +112,57 @@ func TestCheckRuntimeParams_LimitAndBallastEnvs(t *testing.T) {
 	os.Clearenv()
 }
 
+func TestSetDefaultEnvVars(t *testing.T) {
+	testArgs := []string{"SPLUNK_API_URL", "SPLUNK_INGEST_URL", "SPLUNK_TRACE_URL", "SPLUNK_HEC_URL", "SPLUNK_HEC_TOKEN"}
+	for _, v := range testArgs {
+		setDefaultEnvVars()
+		_, ok := os.LookupEnv(v)
+		if ok {
+			t.Errorf("Expected %v unset given SPLUNK_ACCESS_TOKEN or SPLUNK_TOKEN is unset", v)
+		}
+	}
+
+	realm := "us1"
+	token := "1234"
+	valTest := "test"
+	valEmpty := ""
+
+	os.Setenv("SPLUNK_REALM", realm)
+	os.Setenv("SPLUNK_ACCESS_TOKEN", token)
+	setDefaultEnvVars()
+	testArgs2 := [][]string{
+		{"SPLUNK_API_URL", "https://api." + realm + ".signalfx.com"},
+		{"SPLUNK_INGEST_URL", "https://ingest." + realm + ".signalfx.com"},
+		{"SPLUNK_TRACE_URL", "https://ingest." + realm + ".signalfx.com/v2/trace"},
+		{"SPLUNK_HEC_URL", "https://ingest." + realm + ".signalfx.com/v1/log"},
+		{"SPLUNK_HEC_TOKEN", token},
+	}
+	for _, v := range testArgs2 {
+		val, _ := os.LookupEnv(v[0])
+		if val != v[1] {
+			t.Errorf("Expected %v got %v for %v", v[1], val, v[0])
+		}
+	}
+
+	for _, v := range testArgs {
+		os.Setenv(v, valTest)
+		setDefaultEnvVars()
+		val, _ := os.LookupEnv(v)
+		if val != valTest {
+			t.Errorf("Expected %v got %v for %v", valTest, val, v)
+		}
+	}
+
+	for _, v := range testArgs {
+		os.Setenv(v, valEmpty)
+		setDefaultEnvVars()
+		val, _ := os.LookupEnv(v)
+		if val != valEmpty {
+			t.Errorf("Expected %v got %v for %v", valEmpty, val, v)
+		}
+	}
+}
+
 func TestCheckRuntimeParams_MemTotalLimitAndBallastEnvs(t *testing.T) {
 	oldArgs := os.Args
 	assert.NoError(t, os.Setenv(configEnvVarName, path.Join("../../", defaultLocalSAPMConfig)))