From 0279d11c8a2bd5ee667635d6c882e69377bf80d6 Mon Sep 17 00:00:00 2001
From: James Bardin <j.bardin@gmail.com>
Date: Tue, 7 Mar 2017 10:05:07 -0500
Subject: [PATCH] Add TestReset to terraformMockResourceProvider

Have MockResourceProvider implement TestProvider to check that TestReset
is called by the test harness.
---
 terraform/resource_provider_mock.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/terraform/resource_provider_mock.go b/terraform/resource_provider_mock.go
index f5315339f..19a449597 100644
--- a/terraform/resource_provider_mock.go
+++ b/terraform/resource_provider_mock.go
@@ -56,6 +56,8 @@ type MockResourceProvider struct {
 	ReadDataDiffFn                 func(*InstanceInfo, *ResourceConfig) (*InstanceDiff, error)
 	ReadDataDiffReturn             *InstanceDiff
 	ReadDataDiffReturnError        error
+	TestResetCalled                bool
+	TestResetError                 error
 	StopCalled                     bool
 	StopFn                         func() error
 	StopReturnError                error
@@ -144,6 +146,14 @@ func (p *MockResourceProvider) Configure(c *ResourceConfig) error {
 	return p.ConfigureReturnError
 }
 
+func (p *MockResourceProvider) TestReset() error {
+	p.Lock()
+	defer p.Unlock()
+
+	p.TestResetCalled = true
+	return p.TestResetError
+}
+
 func (p *MockResourceProvider) Stop() error {
 	p.Lock()
 	defer p.Unlock()