Add SetEnvironment method to azure package (#517)

Allows modification of the global environments map.

CHANGELOG.md

@@ -1,5 +1,11 @@
 # CHANGELOG
 
+## v14.1.0
+
+### New Features
+
+- Added `azure.SetEnvironment()` that will update the global environments map with the specified values.
+
 ## v14.0.1
 
 ### Bug Fixes

autorest/azure/environments.go

@@ -242,3 +242,8 @@ func EnvironmentFromFile(location string) (unmarshaled Environment, err error) {
 
 	return
 }
+
+// SetEnvironment updates the environment map with the specified values.
+func SetEnvironment(name string, env Environment) {
+	environments[strings.ToUpper(name)] = env
+}

autorest/azure/environments_test.go

@@ -405,3 +405,19 @@ func TestRoundTripSerialization(t *testing.T) {
 		t.Errorf("Expected ResourceIdentifiers.OperationalInsights to be %q, but got %q", env.ResourceIdentifiers.OperationalInsights, testSubject.ResourceIdentifiers.OperationalInsights)
 	}
 }
+
+func TestSetEnvironment(t *testing.T) {
+	const testEnvName = "testenvironment"
+	if _, err := EnvironmentFromName(testEnvName); err == nil {
+		t.Fatal("expected non-nil error")
+	}
+	testEnv := Environment{Name: testEnvName}
+	SetEnvironment(testEnvName, testEnv)
+	result, err := EnvironmentFromName(testEnvName)
+	if err != nil {
+		t.Fatalf("failed to get custom environment: %v", err)
+	}
+	if testEnv != result {
+		t.Fatalf("expected %v, got %v", testEnv, result)
+	}
+}

autorest/version.go

@@ -19,7 +19,7 @@ import (
 	"runtime"
 )
 
-const number = "v14.0.1"
+const number = "v14.1.0"
 
 var (
 	userAgent = fmt.Sprintf("Go/%s (%s-%s) go-autorest/%s",