99 lines
1.8 KiB
Go
99 lines
1.8 KiB
Go
package rpc
|
|
|
|
import (
|
|
"github.com/mitchellh/packer/packer"
|
|
"net/rpc"
|
|
"reflect"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestHookRPC(t *testing.T) {
|
|
// Create the UI to test
|
|
h := new(packer.MockHook)
|
|
|
|
// Serve
|
|
server := rpc.NewServer()
|
|
RegisterHook(server, h)
|
|
address := serveSingleConn(server)
|
|
|
|
// Create the client over RPC and run some methods to verify it works
|
|
client, err := rpc.Dial("tcp", address)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
hClient := Hook(client)
|
|
|
|
// Test Run
|
|
ui := &testUi{}
|
|
hClient.Run("foo", ui, nil, 42)
|
|
if !h.RunCalled {
|
|
t.Fatal("should be called")
|
|
}
|
|
|
|
// Test Cancel
|
|
hClient.Cancel()
|
|
if !h.CancelCalled {
|
|
t.Fatal("should be called")
|
|
}
|
|
}
|
|
|
|
func TestHook_Implements(t *testing.T) {
|
|
var _ packer.Hook = new(hook)
|
|
}
|
|
|
|
func TestHook_cancelWhileRun(t *testing.T) {
|
|
var finishLock sync.Mutex
|
|
finishOrder := make([]string, 0, 2)
|
|
|
|
h := &packer.MockHook{
|
|
RunFunc: func() error {
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
finishLock.Lock()
|
|
finishOrder = append(finishOrder, "run")
|
|
finishLock.Unlock()
|
|
return nil
|
|
},
|
|
}
|
|
|
|
// Serve
|
|
server := rpc.NewServer()
|
|
RegisterHook(server, h)
|
|
address := serveSingleConn(server)
|
|
|
|
// Create the client over RPC and run some methods to verify it works
|
|
client, err := rpc.Dial("tcp", address)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
hClient := Hook(client)
|
|
|
|
// Start the run
|
|
finished := make(chan struct{})
|
|
go func() {
|
|
hClient.Run("foo", nil, nil, nil)
|
|
close(finished)
|
|
}()
|
|
|
|
// Cancel it pretty quickly.
|
|
time.Sleep(10 * time.Millisecond)
|
|
hClient.Cancel()
|
|
|
|
finishLock.Lock()
|
|
finishOrder = append(finishOrder, "cancel")
|
|
finishLock.Unlock()
|
|
|
|
// Verify things are good
|
|
<-finished
|
|
|
|
// Check the results
|
|
expected := []string{"cancel", "run"}
|
|
if !reflect.DeepEqual(finishOrder, expected) {
|
|
t.Fatalf("bad: %#v", finishOrder)
|
|
}
|
|
}
|