75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
|
package rpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"cgl.tideland.biz/asserts"
|
||
|
|
"github.com/mitchellh/packer/packer"
|
||
|
|
"net/rpc"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
type testProvisioner struct {
|
||
|
|
prepareCalled bool
|
||
|
|
prepareConfig interface{}
|
||
|
|
prepareUi packer.Ui
|
||
|
|
provCalled bool
|
||
|
|
provComm packer.Communicator
|
||
|
|
provUi packer.Ui
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *testProvisioner) Prepare(config interface{}, ui packer.Ui) {
|
||
|
|
p.prepareCalled = true
|
||
|
|
p.prepareConfig = config
|
||
|
|
p.prepareUi = ui
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *testProvisioner) Provision(ui packer.Ui, comm packer.Communicator) {
|
||
|
|
p.provCalled = true
|
||
|
|
p.provComm = comm
|
||
|
|
p.provUi = ui
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestProvisionerRPC(t *testing.T) {
|
||
|
|
assert := asserts.NewTestingAsserts(t, true)
|
||
|
|
|
||
|
|
// Create the interface to test
|
||
|
|
p := new(testProvisioner)
|
||
|
|
|
||
|
|
// Start the server
|
||
|
|
server := rpc.NewServer()
|
||
|
|
RegisterProvisioner(server, p)
|
||
|
|
address := serveSingleConn(server)
|
||
|
|
|
||
|
|
// Create the client over RPC and run some methods to verify it works
|
||
|
|
client, err := rpc.Dial("tcp", address)
|
||
|
|
assert.Nil(err, "should be able to connect")
|
||
|
|
|
||
|
|
// Test Prepare
|
||
|
|
config := 42
|
||
|
|
ui := &testUi{}
|
||
|
|
pClient := Provisioner(client)
|
||
|
|
pClient.Prepare(config, ui)
|
||
|
|
assert.True(p.prepareCalled, "prepare should be called")
|
||
|
|
assert.Equal(p.prepareConfig, 42, "prepare should be called with right arg")
|
||
|
|
|
||
|
|
p.prepareUi.Say("foo")
|
||
|
|
assert.True(ui.sayCalled, "say should be called")
|
||
|
|
|
||
|
|
// Test Provision
|
||
|
|
ui = &testUi{}
|
||
|
|
comm := &testCommunicator{}
|
||
|
|
pClient.Provision(ui, comm)
|
||
|
|
assert.True(p.provCalled, "provision should be called")
|
||
|
|
|
||
|
|
p.provUi.Say("foo")
|
||
|
|
assert.True(ui.sayCalled, "say should be called")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestProvisioner_Implements(t *testing.T) {
|
||
|
|
assert := asserts.NewTestingAsserts(t, true)
|
||
|
|
|
||
|
|
var r packer.Provisioner
|
||
|
|
p := Provisioner(nil)
|
||
|
|
|
||
|
|
assert.Implementor(p, &r, "should be a provisioner")
|
||
|
|
}
|