packer-cn/packer/rpc/provisioner_test.go

48 lines
921 B
Go
Raw Normal View History

2013-05-22 18:35:52 -04:00
package rpc
import (
"github.com/mitchellh/packer/packer"
2013-10-16 23:04:57 -04:00
"reflect"
2013-05-22 18:35:52 -04:00
"testing"
)
func TestProvisionerRPC(t *testing.T) {
// Create the interface to test
p := new(packer.MockProvisioner)
2013-05-22 18:35:52 -04:00
// Start the server
2013-12-10 14:56:15 -05:00
client, server := testClientServer(t)
defer client.Close()
defer server.Close()
server.RegisterProvisioner(p)
pClient := client.Provisioner()
2013-05-22 18:35:52 -04:00
// Test Prepare
config := 42
pClient.Prepare(config)
2013-10-16 23:04:57 -04:00
if !p.PrepCalled {
t.Fatal("should be called")
}
if !reflect.DeepEqual(p.PrepConfigs, []interface{}{42}) {
t.Fatalf("bad: %#v", p.PrepConfigs)
}
2013-05-22 18:35:52 -04:00
// Test Provision
ui := &testUi{}
comm := &packer.MockCommunicator{}
2013-05-22 18:35:52 -04:00
pClient.Provision(ui, comm)
2013-10-16 23:04:57 -04:00
if !p.ProvCalled {
t.Fatal("should be called")
}
2013-05-22 18:35:52 -04:00
// Test Cancel
pClient.Cancel()
if !p.CancelCalled {
t.Fatal("cancel should be called")
}
2013-05-22 18:35:52 -04:00
}
func TestProvisioner_Implements(t *testing.T) {
2013-12-10 16:26:07 -05:00
var _ packer.Provisioner = new(provisioner)
2013-05-22 18:35:52 -04:00
}