minor race fix in tests

This commit is contained in:
James Bardin 2017-10-02 16:19:01 -04:00
parent 71e47e56a0
commit fc9ada463c
1 changed files with 5 additions and 0 deletions

View File

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