diff options
Diffstat (limited to 'container/dispatcher_test.go')
| -rw-r--r-- | container/dispatcher_test.go | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/container/dispatcher_test.go b/container/dispatcher_test.go index ee31f91e..bbcb26ad 100644 --- a/container/dispatcher_test.go +++ b/container/dispatcher_test.go @@ -493,6 +493,21 @@ func (k *kstub) start(c *exec.Cmd) error { return err } +func (k *kstub) wait(c *exec.Cmd) error { + k.Helper() + expect := k.Expects("wait") + err := expect.Error( + stub.CheckArg(k.Stub, "c.Path", c.Path, 0), + stub.CheckArgReflect(k.Stub, "c.Args", c.Args, 1), + stub.CheckArgReflect(k.Stub, "c.Env", c.Env, 2), + stub.CheckArg(k.Stub, "c.Dir", c.Dir, 3)) + + if mgc, ok := expect.Ret.(uintptr); ok && mgc == stub.PanicExit { + panic(stub.PanicExit) + } + return err +} + func (k *kstub) signal(c *exec.Cmd, sig os.Signal) error { k.Helper() expect := k.Expects("signal") @@ -759,7 +774,8 @@ func (k *kstub) checkMsg(msg message.Msg) { } func (k *kstub) GetLogger() *log.Logger { panic("unreachable") } -func (k *kstub) IsVerbose() bool { panic("unreachable") } + +func (k *kstub) IsVerbose() bool { k.Helper(); return k.Expects("isVerbose").Ret.(bool) } func (k *kstub) SwapVerbose(verbose bool) bool { k.Helper() |