fix locking and race in test mocks

This commit is contained in:
James Bardin 2023-07-06 11:46:17 -04:00
parent b9d8d99b04
commit 76332db98d
2 changed files with 6 additions and 2 deletions

View File

@ -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
}

View File

@ -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
}
}