From 69664ed168db4c0e8f72c40a050773a0b3294824 Mon Sep 17 00:00:00 2001
From: Kyle Carberry <kyle@coder.com>
Date: Wed, 3 Aug 2022 12:19:13 -0500
Subject: [PATCH] fix: Use "virtual_machine_id" for instance identity with
 Azure (#3355)

This was using the wrong property, causing automatic auth to break.
---
 provisioner/terraform/resources.go      | 4 ++--
 provisioner/terraform/resources_test.go | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/provisioner/terraform/resources.go b/provisioner/terraform/resources.go
index 23b2278af9..95cf93ad94 100644
--- a/provisioner/terraform/resources.go
+++ b/provisioner/terraform/resources.go
@@ -310,8 +310,8 @@ func applyAutomaticInstanceID(resource *tfjson.StateResource, agents []*proto.Ag
 	key, isValid := map[string]string{
 		"google_compute_instance":         "instance_id",
 		"aws_instance":                    "id",
-		"azurerm_linux_virtual_machine":   "id",
-		"azurerm_windows_virtual_machine": "id",
+		"azurerm_linux_virtual_machine":   "virtual_machine_id",
+		"azurerm_windows_virtual_machine": "virtual_machine_id",
 	}[resource.Type]
 	if !isValid {
 		return
diff --git a/provisioner/terraform/resources_test.go b/provisioner/terraform/resources_test.go
index 7999d04b98..5e6e849c9e 100644
--- a/provisioner/terraform/resources_test.go
+++ b/provisioner/terraform/resources_test.go
@@ -222,11 +222,11 @@ func TestInstanceIDAssociation(t *testing.T) {
 	}, {
 		Auth:          "azure-instance-identity",
 		ResourceType:  "azurerm_linux_virtual_machine",
-		InstanceIDKey: "id",
+		InstanceIDKey: "virtual_machine_id",
 	}, {
 		Auth:          "azure-instance-identity",
 		ResourceType:  "azurerm_windows_virtual_machine",
-		InstanceIDKey: "id",
+		InstanceIDKey: "virtual_machine_id",
 	}} {
 		tc := tc
 		t.Run(tc.ResourceType, func(t *testing.T) {