upstream/packer
1
0
Fork 0
mirror of https://github.com/hashicorp/packer.git synced 2026-06-11 01:30:06 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Because the new functionality makes the ``waitForRestart()` function run commands it modifies the value of `comm.StartCmd.Command` that is being checked, we need to implement the same workaround that is being used for the `waitForCommunicator()`` function. This should make the test work again and retain functionality.

Browse source
This commit is contained in:
Gonzalo Peci 2015-08-19 16:05:22 +12:00 committed by unknown
parent 08359e409a
commit 8bd3e62853
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
provisioner/windows-restart/provisioner_test.go
View file

@ -100,6 +100,10 @@ func TestProvisionerProvision_Success(t *testing.T) {
waitForCommunicator = func(p *Provisioner) error {
return nil
}
waitForRestartOld := waitForRestart
waitForRestart = func(p *Provisioner, comm packer.Communicator) error {
return nil
}
err := p.Provision(ui, comm)
if err != nil {
t.Fatal("should not have error")
@ -113,6 +117,7 @@ func TestProvisionerProvision_Success(t *testing.T) {
}
// Set this back!
waitForCommunicator = waitForCommunicatorOld
waitForRestart = waitForRestartOld
}
func TestProvisionerProvision_CustomCommand(t *testing.T) {
@ -131,6 +136,10 @@ func TestProvisionerProvision_CustomCommand(t *testing.T) {
waitForCommunicator = func(p *Provisioner) error {
return nil
}
waitForRestartOld := waitForRestart
waitForRestart = func(p *Provisioner, comm packer.Communicator) error {
return nil
}
err := p.Provision(ui, comm)
if err != nil {
t.Fatal("should not have error")
@ -142,6 +151,7 @@ func TestProvisionerProvision_CustomCommand(t *testing.T) {
}
// Set this back!
waitForCommunicator = waitForCommunicatorOld
waitForRestart = waitForRestartOld
}
func TestProvisionerProvision_RestartCommandFail(t *testing.T) {