Merge pull request #46 from hashicorp/atlas-address-env-var

Set Atlas endpoint to ATLAS_ADDRESS if present

v1/client.go

@@ -17,8 +17,12 @@ import (
 )
 
 const (
-	// atlasEndpoint is the default base URL for connecting to Atlas.
-	atlasEndpoint = "https://atlas.hashicorp.com"
+	// atlasDefaultEndpoint is the default base URL for connecting to Atlas.
+	atlasDefaultEndpoint = "https://atlas.hashicorp.com"
+
+	// atlasEndpointEnvVar is the environment variable that overrrides the
+	// default Atlas address.
+	atlasEndpointEnvVar = "ATLAS_ADDRESS"
 
 	// atlasTokenHeader is the header key used for authenticating with Atlas
 	atlasTokenHeader = "X-Atlas-Token"
@@ -60,6 +64,11 @@ type Client struct {
 
 // DefaultClient returns a client that connects to the Atlas API.
 func DefaultClient() *Client {
+	atlasEndpoint := os.Getenv(atlasEndpointEnvVar)
+	if atlasEndpoint == "" {
+		atlasEndpoint = atlasDefaultEndpoint
+	}
+
 	client, err := NewClient(atlasEndpoint)
 	if err != nil {
 		panic(err)

v1/client_test.go

@@ -2,6 +2,7 @@ package atlas
 
 import (
 	"net/url"
+	"os"
 	"reflect"
 	"strings"
 	"testing"
@@ -10,8 +11,24 @@ import (
 func TestDefaultClient_url(t *testing.T) {
 	client := DefaultClient()
 
-	if client.URL.String() != atlasEndpoint {
-		t.Fatalf("expected %q to be %q", client.URL.String(), atlasEndpoint)
+	if client.URL.String() != atlasDefaultEndpoint {
+		t.Fatalf("expected %q to be %q", client.URL.String(), atlasDefaultEndpoint)
+	}
+}
+
+func TestDefaultClient_urlFromEnvVar(t *testing.T) {
+	defer os.Setenv(atlasEndpointEnvVar, os.Getenv(atlasEndpointEnvVar))
+	otherEndpoint := "http://127.0.0.1:1234"
+
+	err := os.Setenv(atlasEndpointEnvVar, otherEndpoint)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	client := DefaultClient()
+
+	if client.URL.String() != otherEndpoint {
+		t.Fatalf("expected %q to be %q", client.URL.String(), otherEndpoint)
 	}
 }