packer-cn/packer/rpc/command_test.go

package rpc

import (
	"cgl.tideland.biz/asserts"
	"github.com/mitchellh/packer/packer"
	"net/rpc"
	"testing"
)

type TestCommand struct {
	runArgs []string
	runCalled bool
	runEnv packer.Environment
}

func (tc *TestCommand) Run(env packer.Environment, args []string) int {
	tc.runCalled = true
	tc.runArgs = args
	tc.runEnv = env
	return 0
}

func (tc *TestCommand) Synopsis() string {
	return "foo"
}

func TestRPCCommand(t *testing.T) {
	assert := asserts.NewTestingAsserts(t, true)

	// Create the command
	command := new(TestCommand)

	// Start the server
	server := rpc.NewServer()
	RegisterCommand(server, command)
	address := serveSingleConn(server)

	// Create the command client over RPC and run some methods to verify
	// we get the proper behavior.
	client, err := rpc.Dial("tcp", address)
	assert.Nil(err, "should be no error")

	clientComm := Command(client)

	// Test run
	runArgs := []string{"foo", "bar"}
	testEnv := &testEnvironment{}
	exitCode := clientComm.Run(testEnv, runArgs)
	assert.Equal(command.runArgs, runArgs, "Correct args should be sent")
	assert.Equal(exitCode, 0, "Exit code should be correct")

	assert.NotNil(command.runEnv, "should have an env")
	if command.runEnv != nil {
		command.runEnv.Ui()
		assert.True(testEnv.uiCalled, "UI should be called on env")
	}

	// Test Synopsis
	synopsis := clientComm.Synopsis()
	assert.Equal(synopsis, "foo", "Synopsis should be correct")
}