Merge pull request #16242 from hashicorp/jbardin/race

minor race fix in tests
This commit is contained in:
James Bardin 2017-10-02 16:50:02 -04:00 committed by GitHub
commit 67be68b521
1 changed files with 5 additions and 0 deletions

View File

@ -1,13 +1,18 @@
package terraform package terraform
import "sync"
// MockUIOutput is an implementation of UIOutput that can be used for tests. // MockUIOutput is an implementation of UIOutput that can be used for tests.
type MockUIOutput struct { type MockUIOutput struct {
sync.Mutex
OutputCalled bool OutputCalled bool
OutputMessage string OutputMessage string
OutputFn func(string) OutputFn func(string)
} }
func (o *MockUIOutput) Output(v string) { func (o *MockUIOutput) Output(v string) {
o.Lock()
defer o.Unlock()
o.OutputCalled = true o.OutputCalled = true
o.OutputMessage = v o.OutputMessage = v
if o.OutputFn != nil { if o.OutputFn != nil {