mirror of
https://github.com/hashicorp/packer.git
synced 2026-09-29 19:24:10 -04:00
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.
This commit is contained in:
@@ -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) {
|
||||
|
||||
Reference in New Issue
Block a user