From 3cee60ee3a9e3f7024e1a7ee8785fda10b330dc9 Mon Sep 17 00:00:00 2001
From: Martin Atkins <mart@degeneration.co.uk>
Date: Tue, 9 Oct 2018 16:55:42 -0700
Subject: [PATCH] command: Fix various "untaint" command tests

Resource instance state now always includes the provider address fo each
resource, so our expected state output strings needed to be adjusted.
---
 command/untaint_test.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/command/untaint_test.go b/command/untaint_test.go
index fc3d32eb6..681cb2673 100644
--- a/command/untaint_test.go
+++ b/command/untaint_test.go
@@ -135,12 +135,14 @@ func TestUntaint_backup(t *testing.T) {
 	testStateOutput(t, path+".backup", strings.TrimSpace(`
 test_instance.foo: (tainted)
   ID = bar
+  provider = provider.test
 	`))
 
 	// State is untainted
 	testStateOutput(t, path, strings.TrimSpace(`
 test_instance.foo:
   ID = bar
+  provider = provider.test
 	`))
 }
 
@@ -190,6 +192,7 @@ func TestUntaint_backupDisable(t *testing.T) {
 	testStateOutput(t, path, strings.TrimSpace(`
 test_instance.foo:
   ID = bar
+  provider = provider.test
 	`))
 }
 
@@ -251,6 +254,7 @@ func TestUntaint_defaultState(t *testing.T) {
 	testStateOutput(t, path, strings.TrimSpace(`
 test_instance.foo:
   ID = bar
+  provider = provider.test
 	`))
 }
 
@@ -363,10 +367,12 @@ func TestUntaint_stateOut(t *testing.T) {
 	testStateOutput(t, path, strings.TrimSpace(`
 test_instance.foo: (tainted)
   ID = bar
+  provider = provider.test
 	`))
 	testStateOutput(t, "foo", strings.TrimSpace(`
 test_instance.foo:
   ID = bar
+  provider = provider.test
 	`))
 }