diff --git a/internal/terraform/provider_mock.go b/internal/terraform/provider_mock.go index 10c8902184..05f94699d9 100644 --- a/internal/terraform/provider_mock.go +++ b/internal/terraform/provider_mock.go @@ -511,6 +511,9 @@ func (p *MockProvider) ReadDataSource(r providers.ReadDataSourceRequest) (resp p } func (p *MockProvider) Close() error { + p.Lock() + defer p.Unlock() + p.CloseCalled = true return p.CloseError } diff --git a/internal/terraform/terraform_test.go b/internal/terraform/terraform_test.go index b80cb3411c..2ba8829a4a 100644 --- a/internal/terraform/terraform_test.go +++ b/internal/terraform/terraform_test.go @@ -193,9 +193,10 @@ func testProviderFuncFixed(rp providers.Interface) providers.Factory { } func testProvisionerFuncFixed(rp *MockProvisioner) provisioners.Factory { + // make sure this provisioner has has not been closed + rp.CloseCalled = false + return func() (provisioners.Interface, error) { - // make sure this provisioner has has not been closed - rp.CloseCalled = false return rp, nil } }