50 lines
943 B
Go
50 lines
943 B
Go
package packer
|
|
|
|
import "testing"
|
|
|
|
type TestProvisioner struct {
|
|
prepCalled bool
|
|
prepConfigs []interface{}
|
|
provCalled bool
|
|
}
|
|
|
|
func (t *TestProvisioner) Prepare(configs ...interface{}) error {
|
|
t.prepCalled = true
|
|
t.prepConfigs = configs
|
|
return nil
|
|
}
|
|
|
|
func (t *TestProvisioner) Provision(Ui, Communicator) {
|
|
t.provCalled = true
|
|
}
|
|
|
|
func TestProvisionHook_Impl(t *testing.T) {
|
|
var raw interface{}
|
|
raw = &ProvisionHook{}
|
|
if _, ok := raw.(Hook); !ok {
|
|
t.Fatalf("must be a Hook")
|
|
}
|
|
}
|
|
|
|
func TestProvisionHook(t *testing.T) {
|
|
pA := &TestProvisioner{}
|
|
pB := &TestProvisioner{}
|
|
|
|
ui := testUi()
|
|
var comm Communicator = nil
|
|
var data interface{} = nil
|
|
|
|
hook := &ProvisionHook{[]Provisioner{pA, pB}}
|
|
hook.Run("foo", ui, comm, data)
|
|
|
|
if !pA.provCalled {
|
|
t.Error("provision should be called on pA")
|
|
}
|
|
|
|
if !pB.provCalled {
|
|
t.Error("provision should be called on pB")
|
|
}
|
|
}
|
|
|
|
// TODO(mitchellh): Test that they're run in the proper order
|