packer-cn/packer/rpc/ui_test.go

105 lines
1.9 KiB
Go
Raw Normal View History

2013-05-03 18:49:15 -04:00
package rpc
import (
"reflect"
2013-05-03 18:49:15 -04:00
"testing"
)
type testUi struct {
askCalled bool
askQuery string
2013-06-03 14:30:38 -04:00
errorCalled bool
errorMessage string
2013-08-11 21:16:00 -04:00
machineCalled bool
machineType string
machineArgs []string
2013-06-03 14:30:38 -04:00
messageCalled bool
messageMessage string
sayCalled bool
sayMessage string
2013-05-03 18:49:15 -04:00
}
func (u *testUi) Ask(query string) (string, error) {
u.askCalled = true
u.askQuery = query
return "foo", nil
}
func (u *testUi) Error(message string) {
2013-05-08 18:12:48 -04:00
u.errorCalled = true
u.errorMessage = message
2013-05-08 18:12:48 -04:00
}
2013-08-11 21:16:00 -04:00
func (u *testUi) Machine(t string, args ...string) {
u.machineCalled = true
u.machineType = t
u.machineArgs = args
}
2013-06-03 14:30:38 -04:00
func (u *testUi) Message(message string) {
u.messageCalled = true
u.messageMessage = message
}
func (u *testUi) Say(message string) {
2013-05-03 18:49:15 -04:00
u.sayCalled = true
u.sayMessage = message
2013-05-03 18:49:15 -04:00
}
func TestUiRPC(t *testing.T) {
// Create the UI to test
ui := new(testUi)
// Start the RPC server
2013-12-09 19:22:11 -05:00
client, server := testClientServer(t)
defer client.Close()
defer server.Close()
server.RegisterUi(ui)
2013-05-03 18:49:15 -04:00
2013-12-09 19:22:11 -05:00
uiClient := client.Ui()
2013-05-03 18:49:15 -04:00
// Basic error and say tests
result, err := uiClient.Ask("query")
2013-10-16 23:04:57 -04:00
if err != nil {
t.Fatalf("err: %s", err)
}
if !ui.askCalled {
t.Fatal("should be called")
}
if ui.askQuery != "query" {
t.Fatalf("bad: %s", ui.askQuery)
}
if result != "foo" {
t.Fatalf("bad: %#v", result)
}
uiClient.Error("message")
2013-10-16 23:04:57 -04:00
if ui.errorMessage != "message" {
t.Fatalf("bad: %#v", ui.errorMessage)
}
2013-05-08 18:12:48 -04:00
2013-06-03 14:30:38 -04:00
uiClient.Message("message")
2013-10-16 23:04:57 -04:00
if ui.messageMessage != "message" {
t.Fatalf("bad: %#v", ui.errorMessage)
}
2013-06-03 14:30:38 -04:00
uiClient.Say("message")
2013-10-16 23:04:57 -04:00
if ui.sayMessage != "message" {
t.Fatalf("bad: %#v", ui.errorMessage)
}
uiClient.Machine("foo", "bar", "baz")
if !ui.machineCalled {
t.Fatal("machine should be called")
}
if ui.machineType != "foo" {
t.Fatalf("bad type: %#v", ui.machineType)
}
expected := []string{"bar", "baz"}
if !reflect.DeepEqual(ui.machineArgs, expected) {
t.Fatalf("bad: %#v", ui.machineArgs)
}
2013-05-03 18:49:15 -04:00
}