5cb7355814
/cc @andytson - Wanted to CC you in here so you could see some changes I made. First, I changed commit to opt-in, so you set "commit": true to get commit behavior. This simplifies the logic a bit. Then, I removed the skipping for StepExport and StepCommit and put that into the Builder itself. This simplifies those steps (limits abstraction leakage). Otherwise, everything looks great!
100 lines
2.1 KiB
Go
100 lines
2.1 KiB
Go
package docker
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"github.com/mitchellh/multistep"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func testStepExportState(t *testing.T) multistep.StateBag {
|
|
state := testState(t)
|
|
state.Put("container_id", "foo")
|
|
return state
|
|
}
|
|
|
|
func TestStepExport_impl(t *testing.T) {
|
|
var _ multistep.Step = new(StepExport)
|
|
}
|
|
|
|
func TestStepExport(t *testing.T) {
|
|
state := testStepExportState(t)
|
|
step := new(StepExport)
|
|
defer step.Cleanup(state)
|
|
|
|
// Create a tempfile for our output path
|
|
tf, err := ioutil.TempFile("", "packer")
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
tf.Close()
|
|
defer os.Remove(tf.Name())
|
|
|
|
config := state.Get("config").(*Config)
|
|
config.ExportPath = tf.Name()
|
|
driver := state.Get("driver").(*MockDriver)
|
|
driver.ExportReader = bytes.NewReader([]byte("data!"))
|
|
|
|
// run the step
|
|
if action := step.Run(state); action != multistep.ActionContinue {
|
|
t.Fatalf("bad action: %#v", action)
|
|
}
|
|
|
|
// verify we did the right thing
|
|
if !driver.ExportCalled {
|
|
t.Fatal("should've exported")
|
|
}
|
|
if driver.ExportID != "foo" {
|
|
t.Fatalf("bad: %#v", driver.ExportID)
|
|
}
|
|
|
|
// verify the data exported to the file
|
|
contents, err := ioutil.ReadFile(tf.Name())
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
if string(contents) != "data!" {
|
|
t.Fatalf("bad: %#v", string(contents))
|
|
}
|
|
}
|
|
|
|
func TestStepExport_error(t *testing.T) {
|
|
state := testStepExportState(t)
|
|
step := new(StepExport)
|
|
defer step.Cleanup(state)
|
|
|
|
// Create a tempfile for our output path
|
|
tf, err := ioutil.TempFile("", "packer")
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
tf.Close()
|
|
|
|
if err := os.Remove(tf.Name()); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
config := state.Get("config").(*Config)
|
|
config.ExportPath = tf.Name()
|
|
driver := state.Get("driver").(*MockDriver)
|
|
driver.ExportError = errors.New("foo")
|
|
|
|
// run the step
|
|
if action := step.Run(state); action != multistep.ActionHalt {
|
|
t.Fatalf("bad action: %#v", action)
|
|
}
|
|
|
|
// verify we have an error
|
|
if _, ok := state.GetOk("error"); !ok {
|
|
t.Fatal("should have error")
|
|
}
|
|
|
|
// verify we didn't make that file
|
|
if _, err := os.Stat(tf.Name()); err == nil {
|
|
t.Fatal("export path shouldn't exist")
|
|
}
|
|
}
|