iSharkFly-Docs
/
packer-cn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

provisioner/windows-restart: remove race in test

This commit is contained in:
Mark Peek 2015-10-12 13:20:18 -07:00
parent f401e765df
commit 1aad5cf6d5
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
provisioner/windows-restart/provisioner_test.go
View File

@ -193,12 +193,14 @@ func TestProvision_waitForRestartTimeout(t *testing.T) {
p.Prepare(config) p.Prepare(config)
waitForCommunicatorOld := waitForCommunicator waitForCommunicatorOld := waitForCommunicator
waitDone := make(chan bool) waitDone := make(chan bool)
waitContinue := make(chan bool)
// Block until cancel comes through // Block until cancel comes through
waitForCommunicator = func(p *Provisioner) error { waitForCommunicator = func(p *Provisioner) error {
for { for {
select { select {
case <-waitDone: case <-waitDone:
waitContinue <- true
} }
} }
} }
@ -207,7 +209,7 @@ func TestProvision_waitForRestartTimeout(t *testing.T) {
err = p.Provision(ui, comm) err = p.Provision(ui, comm)
waitDone <- true waitDone <- true
}() }()
<-waitDone <-waitContinue
if err == nil { if err == nil {
t.Fatal("should not have error") t.Fatal("should not have error")