provisioner/windows-restart: remove race in test

11 years ago · 1aad5cf6d5
parent f401e765df
commit 1aad5cf6d5
1 changed files with 3 additions and 1 deletions
--- a/provisioner/windows-restart/provisioner_test.go
+++ b/provisioner/windows-restart/provisioner_test.go
@ -193,12 +193,14 @@ func TestProvision_waitForRestartTimeout(t *testing.T) {
 	p.Prepare(config)
 	waitForCommunicatorOld := waitForCommunicator
 	waitDone := make(chan bool)
+	waitContinue := make(chan bool)

 	// Block until cancel comes through
 	waitForCommunicator = func(p *Provisioner) error {
 		for {
 			select {
 			case <-waitDone:
+				waitContinue <- true
 			}
 		}
 	}
@ -207,7 +209,7 @@ func TestProvision_waitForRestartTimeout(t *testing.T) {
 		err = p.Provision(ui, comm)
 		waitDone <- true
 	}()
-	<-waitDone
+	<-waitContinue

 	if err == nil {
 		t.Fatal("should not have error")