From eb201724c1e1558d86ea86c9e6ae08f864993720 Mon Sep 17 00:00:00 2001
From: Christopher Boumenot <chrboum@microsoft.com>
Date: Tue, 20 Mar 2018 16:20:11 -0700
Subject: [PATCH] azure: unit tests for naming temp resources

---
 builder/azure/arm/config_test.go | 38 +++++++++++++++++++++++++++++++-
 1 file changed, 37 insertions(+), 1 deletion(-)

diff --git a/builder/azure/arm/config_test.go b/builder/azure/arm/config_test.go
index 71f3c352d..abb0f5a7d 100644
--- a/builder/azure/arm/config_test.go
+++ b/builder/azure/arm/config_test.go
@@ -1058,7 +1058,6 @@ func TestConfigShouldRejectManagedDiskNames(t *testing.T) {
 }
 
 func TestConfigAdditionalDiskDefaultIsNil(t *testing.T) {
-
 	c, _, _ := newConfig(getArmBuilderConfiguration(), getPackerConfiguration())
 	if c.AdditionalDiskSize != nil {
 		t.Errorf("Expected Config to not have a set of additional disks, but got a non nil value")
@@ -1228,6 +1227,43 @@ func TestPlanInfoTooManyTagsErrors(t *testing.T) {
 	}
 }
 
+// The Azure builder creates temporary resources, but the user has some control over
+// these values. This test asserts those values are controllable by the user.
+func TestConfigShouldAllowTempNameOverrides(t *testing.T) {
+	config := map[string]interface{}{
+		"image_offer":                       "ignore",
+		"image_publisher":                   "ignore",
+		"image_sku":                         "ignore",
+		"location":                          "ignore",
+		"subscription_id":                   "ignore",
+		"communicator":                      "none",
+		"os_type":                           "linux",
+		"managed_image_name":                "ignore",
+		"managed_image_resource_group_name": "ignore",
+		"temp_resource_group_name":          "myTempResourceGroupName",
+		"temp_compute_name":                 "myTempComputeName",
+	}
+
+	c, _, err := newConfig(config, getPackerConfiguration())
+	if err != nil {
+		t.Errorf("newConfig failed with %q", err)
+	}
+
+	if c.TempResourceGroupName != "myTempResourceGroupName" {
+		t.Errorf("expected TempResourceGroupName to be %q, but got %q", "myTempResourceGroupName", c.TempResourceGroupName)
+	}
+	if c.tmpResourceGroupName != "myTempResourceGroupName" {
+		t.Errorf("expected tmpResourceGroupName to be %q, but got %q", "myTempResourceGroupName", c.tmpResourceGroupName)
+	}
+
+	if c.TempComputeName != "myTempComputeName" {
+		t.Errorf("expected TempComputeName to be %q, but got %q", "myTempComputeName", c.TempComputeName)
+	}
+	if c.tmpComputeName != "myTempComputeName" {
+		t.Errorf("expected tmpComputeName to be %q, but got %q", "myTempComputeName", c.tmpResourceGroupName)
+	}
+}
+
 func getArmBuilderConfiguration() map[string]string {
 	m := make(map[string]string)
 	for _, v := range requiredConfigValues {