packer-cn/packer/rpc/builder_test.go

76 lines
1.7 KiB
Go
Raw Normal View History

2013-05-04 16:47:11 -04:00
package rpc
import (
"cgl.tideland.biz/asserts"
"github.com/mitchellh/packer/packer"
"net/rpc"
"testing"
)
type testBuilder struct {
prepareCalled bool
prepareConfig interface{}
2013-05-10 20:01:24 -04:00
runCalled bool
runHook packer.Hook
2013-05-10 20:01:24 -04:00
runUi packer.Ui
2013-05-04 16:47:11 -04:00
}
func (b *testBuilder) Prepare(config interface{}) error {
2013-05-04 16:47:11 -04:00
b.prepareCalled = true
b.prepareConfig = config
return nil
2013-05-04 16:47:11 -04:00
}
func (b *testBuilder) Run(ui packer.Ui, hook packer.Hook) {
2013-05-04 16:47:11 -04:00
b.runCalled = true
b.runHook = hook
2013-05-04 16:47:11 -04:00
b.runUi = ui
}
func TestBuilderRPC(t *testing.T) {
assert := asserts.NewTestingAsserts(t, true)
// Create the interface to test
b := new(testBuilder)
// Start the server
server := rpc.NewServer()
RegisterBuilder(server, b)
address := serveSingleConn(server)
2013-05-04 16:47:11 -04:00
// Create the client over RPC and run some methods to verify it works
client, err := rpc.Dial("tcp", address)
2013-05-04 16:47:11 -04:00
assert.Nil(err, "should be able to connect")
// Test Prepare
config := 42
bClient := Builder(client)
2013-05-04 16:47:11 -04:00
bClient.Prepare(config)
assert.True(b.prepareCalled, "prepare should be called")
assert.Equal(b.prepareConfig, 42, "prepare should be called with right arg")
2013-05-04 16:50:02 -04:00
// Test Run
hook := &testHook{}
2013-05-04 16:50:02 -04:00
ui := &testUi{}
bClient.Run(ui, hook)
2013-05-04 16:50:02 -04:00
assert.True(b.runCalled, "runs hould be called")
if b.runCalled {
b.runHook.Run("foo", nil, nil)
assert.True(hook.runCalled, "run should be called")
2013-05-04 16:50:02 -04:00
b.runUi.Say("format")
assert.True(ui.sayCalled, "say should be called")
assert.Equal(ui.sayFormat, "format", "format should be correct")
}
2013-05-04 16:47:11 -04:00
}
2013-05-09 14:32:03 -04:00
func TestBuilder_ImplementsBuilder(t *testing.T) {
2013-05-04 16:47:11 -04:00
assert := asserts.NewTestingAsserts(t, true)
var realBuilder packer.Builder
b := Builder(nil)
2013-05-04 16:47:11 -04:00
assert.Implementor(b, &realBuilder, "should be a Builder")
}