b54b82d3ac
- startup scripts don't run for Windows since it is isn't implemented yet.
- startup scripts use instance metadata instead of serial port output to flag when they are done.
- added licenses to Image data type (to check if an Image is a Windows Image).
- added GetImage and GetImageFromProject to googlecompute Drivers.
- changed some of the builder/googlecompute tests to use github.com/stretchr/testify/assert.
Tests:
- (in the Packer directory) `go test .`, `go test ./builder/googlecompute`, and `go test ./post-processor/googlecompute-export`
- manual run of `packer build packer_template.json` with the following files
--packer_template.json--
{
"builders": [
{
"type": "googlecompute",
"account_file": "creds.json",
"project_id": "google.com:packer-test",
"source_image": "debian-8-jessie-v20160629",
"zone": "us-central1-a",
"startup_script_file": "startup_script.sh",
"metadata": {
"startup-script": "#!/bin/sh\necho \"This should be overwritten.\"",
"startup-script-log-dest": "gs://packer-test.google.com.a.appspot.com/startup-script.log"
},
"image_name": "test-packer-modifications",
"ssh_username": "foo"
}
],
"post-processors": [
{
"type": "googlecompute-export",
"paths": [
"gs://packer-test.google.com.a.appspot.com/foo.tar.gz",
"gs://packer-test.google.com.a.appspot.com/bar.tar.gz"
],
"keep_input_artifact": true
}
]
}
--startup_script.sh--
\#!/bin/sh
echo "Hi, my name is Scott. I'm waiting 60 seconds!" >> /scott
sleep 60
echo "I'm done waiting!" >> /scott
118 lines
3.3 KiB
Go
118 lines
3.3 KiB
Go
package googlecompute
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/mitchellh/multistep"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestStepCreateInstance_impl(t *testing.T) {
|
|
var _ multistep.Step = new(StepCreateInstance)
|
|
}
|
|
|
|
func TestStepCreateInstance(t *testing.T) {
|
|
state := testState(t)
|
|
step := new(StepCreateInstance)
|
|
defer step.Cleanup(state)
|
|
|
|
state.Put("ssh_public_key", "key")
|
|
|
|
c := state.Get("config").(*Config)
|
|
d := state.Get("driver").(*DriverMock)
|
|
d.GetImageResult = StubImage("test-image", "test-project", []string{}, 100)
|
|
|
|
// run the step
|
|
assert.Equal(t, step.Run(state), multistep.ActionContinue, "Step should have passed and continued.")
|
|
|
|
// Verify state
|
|
nameRaw, ok := state.GetOk("instance_name")
|
|
assert.True(t, ok, "State should have an instance name.")
|
|
|
|
// cleanup
|
|
step.Cleanup(state)
|
|
|
|
// Check args passed to the driver.
|
|
assert.Equal(t, d.DeleteInstanceName, nameRaw.(string), "Incorrect instance name passed to driver.")
|
|
assert.Equal(t, d.DeleteInstanceZone, c.Zone, "Incorrect instance zone passed to driver.")
|
|
assert.Equal(t, d.DeleteDiskName, c.InstanceName, "Incorrect disk name passed to driver.")
|
|
assert.Equal(t, d.DeleteDiskZone, c.Zone, "Incorrect disk zone passed to driver.")
|
|
}
|
|
|
|
func TestStepCreateInstance_error(t *testing.T) {
|
|
state := testState(t)
|
|
step := new(StepCreateInstance)
|
|
defer step.Cleanup(state)
|
|
|
|
state.Put("ssh_public_key", "key")
|
|
|
|
d := state.Get("driver").(*DriverMock)
|
|
d.RunInstanceErr = errors.New("error")
|
|
d.GetImageResult = StubImage("test-image", "test-project", []string{}, 100)
|
|
|
|
// run the step
|
|
assert.Equal(t, step.Run(state), multistep.ActionHalt, "Step should have failed and halted.")
|
|
|
|
// Verify state
|
|
_, ok := state.GetOk("error")
|
|
assert.True(t, ok, "State should have an error.")
|
|
_, ok = state.GetOk("instance_name")
|
|
assert.False(t, ok, "State should not have an instance name.")
|
|
}
|
|
|
|
func TestStepCreateInstance_errorOnChannel(t *testing.T) {
|
|
state := testState(t)
|
|
step := new(StepCreateInstance)
|
|
defer step.Cleanup(state)
|
|
|
|
state.Put("ssh_public_key", "key")
|
|
|
|
errCh := make(chan error, 1)
|
|
errCh <- errors.New("error")
|
|
|
|
d := state.Get("driver").(*DriverMock)
|
|
d.RunInstanceErrCh = errCh
|
|
d.GetImageResult = StubImage("test-image", "test-project", []string{}, 100)
|
|
|
|
// run the step
|
|
assert.Equal(t, step.Run(state), multistep.ActionHalt, "Step should have failed and halted.")
|
|
|
|
// Verify state
|
|
_, ok := state.GetOk("error")
|
|
assert.True(t, ok, "State should have an error.")
|
|
_, ok = state.GetOk("instance_name")
|
|
assert.False(t, ok, "State should not have an instance name.")
|
|
}
|
|
|
|
func TestStepCreateInstance_errorTimeout(t *testing.T) {
|
|
state := testState(t)
|
|
step := new(StepCreateInstance)
|
|
defer step.Cleanup(state)
|
|
|
|
state.Put("ssh_public_key", "key")
|
|
|
|
errCh := make(chan error, 1)
|
|
go func() {
|
|
<-time.After(10 * time.Millisecond)
|
|
errCh <- nil
|
|
}()
|
|
|
|
config := state.Get("config").(*Config)
|
|
config.stateTimeout = 1 * time.Microsecond
|
|
|
|
d := state.Get("driver").(*DriverMock)
|
|
d.RunInstanceErrCh = errCh
|
|
d.GetImageResult = StubImage("test-image", "test-project", []string{}, 100)
|
|
|
|
// run the step
|
|
assert.Equal(t, step.Run(state), multistep.ActionHalt, "Step should have failed and halted.")
|
|
|
|
// Verify state
|
|
_, ok := state.GetOk("error")
|
|
assert.True(t, ok, "State should have an error.")
|
|
_, ok = state.GetOk("instance_name")
|
|
assert.False(t, ok, "State should not have an instance name.")
|
|
}
|