89 lines
1.5 KiB
Go
89 lines
1.5 KiB
Go
package packer
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// A helper Hook implementation for testing cancels.
|
|
type CancelHook struct {
|
|
sync.Mutex
|
|
cancelCh chan struct{}
|
|
doneCh chan struct{}
|
|
|
|
Cancelled bool
|
|
}
|
|
|
|
func (h *CancelHook) Run(string, Ui, Communicator, interface{}) error {
|
|
h.Lock()
|
|
h.cancelCh = make(chan struct{})
|
|
h.doneCh = make(chan struct{})
|
|
h.Unlock()
|
|
|
|
defer close(h.doneCh)
|
|
|
|
select {
|
|
case <-h.cancelCh:
|
|
h.Cancelled = true
|
|
case <-time.After(1 * time.Second):
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (h *CancelHook) Cancel() {
|
|
h.Lock()
|
|
close(h.cancelCh)
|
|
h.Unlock()
|
|
|
|
<-h.doneCh
|
|
}
|
|
|
|
func TestDispatchHook_Implements(t *testing.T) {
|
|
var _ Hook = new(DispatchHook)
|
|
}
|
|
|
|
func TestDispatchHook_Run_NoHooks(t *testing.T) {
|
|
// Just make sure nothing blows up
|
|
dh := &DispatchHook{}
|
|
dh.Run("foo", nil, nil, nil)
|
|
}
|
|
|
|
func TestDispatchHook_Run(t *testing.T) {
|
|
hook := &MockHook{}
|
|
|
|
mapping := make(map[string][]Hook)
|
|
mapping["foo"] = []Hook{hook}
|
|
dh := &DispatchHook{Mapping: mapping}
|
|
dh.Run("foo", nil, nil, 42)
|
|
|
|
if !hook.RunCalled {
|
|
t.Fatal("should be called")
|
|
}
|
|
if hook.RunName != "foo" {
|
|
t.Fatalf("bad: %s", hook.RunName)
|
|
}
|
|
if hook.RunData != 42 {
|
|
t.Fatalf("bad: %#v", hook.RunData)
|
|
}
|
|
}
|
|
|
|
func TestDispatchHook_cancel(t *testing.T) {
|
|
hook := new(CancelHook)
|
|
|
|
dh := &DispatchHook{
|
|
Mapping: map[string][]Hook{
|
|
"foo": {hook},
|
|
},
|
|
}
|
|
|
|
go dh.Run("foo", nil, nil, 42)
|
|
time.Sleep(100 * time.Millisecond)
|
|
dh.Cancel()
|
|
|
|
if !hook.Cancelled {
|
|
t.Fatal("hook should've cancelled")
|
|
}
|
|
}
|