Merge pull request #8 from FireTail-io/api-url-env-var

Get FIRETAIL_API_URL from env var

Author: rileyfiretail

logs-handler/main.go

@@ -6,11 +6,21 @@ import (
 	"github.com/aws/aws-lambda-go/lambda"
 )
 
+const DefaultFiretailApiUrl string = "https://api.logging.eu-west-1.prod.firetail.app/logs/aws/appsync"
+
 var firetailApiUrl string
 var firetailApiToken string
 
+func loadEnvVars() {
+	var firetailApiUrlSet bool
+	firetailApiUrl, firetailApiUrlSet = os.LookupEnv("FIRETAIL_API_URL")
+	if !firetailApiUrlSet {
+		firetailApiUrl = DefaultFiretailApiUrl
+	}
+	firetailApiToken = os.Getenv("FIRETAIL_API_TOKEN")	
+}
+
 func main() {
-	firetailApiUrl = "https://api.logging.eu-west-1.prod.firetail.app/logs/aws/appsync"
-	firetailApiToken = os.Getenv("FIRETAIL_API_TOKEN")
+	loadEnvVars()
 	lambda.Start(Handler)
 }

logs-handler/main_test.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestLoadEnvVars(t *testing.T) {
+	const MockFiretailApiUrl = "MOCK_FIRETAIL_API_URL"
+	const MockFiretailApiToken = "MOCK_FIRETAIL_API_TOKEN"
+
+	t.Setenv("FIRETAIL_API_URL", MockFiretailApiUrl)
+	t.Setenv("FIRETAIL_API_TOKEN", MockFiretailApiToken)
+
+	loadEnvVars()
+
+	assert.Equal(t, firetailApiUrl, MockFiretailApiUrl)
+	assert.Equal(t, firetailApiToken, MockFiretailApiToken)
+}
+
+func TestLoadEnvVarsApiUrlUnset(t *testing.T) {
+	const MockFiretailApiToken = "MOCK_FIRETAIL_API_TOKEN"
+
+	t.Setenv("FIRETAIL_API_TOKEN", MockFiretailApiToken)
+
+	loadEnvVars()
+
+	assert.Equal(t, firetailApiUrl, DefaultFiretailApiUrl)
+	assert.Equal(t, firetailApiToken, MockFiretailApiToken)
+}