// TestRunnerWait tests whether the Wait function stops after Stop is done and
// doesn't cause a deadlock or race condition.
func TestRunnerWait(t *testing.T) {
funcs := []runner.Runnable{goFunc(0, false), goFunc(1, false), goFunc(2, false)}
run := runner.New(funcs)
err := run.Start()
if err != nil {
t.Error("Failed to start: ", err)
}
startChan := make(chan struct{})
doneWaitChan := make(chan struct{})
doneStopChan := make(chan struct{})
go func() {
<-startChan
run.Wait()
doneWaitChan <- struct{}{}
}()
// This will cause Wait to be called.
startChan <- struct{}{}
time.Sleep(time.Second)
go func() {
<-startChan
run.Stop()
doneStopChan <- struct{}{}
}()
// This will cause Stop to be called.
close(startChan)
select {
case <-doneWaitChan:
t.Error("Wait ended before stop.")
case <-doneStopChan:
}
<-doneWaitChan
}
// TestRunnerErr tests error conditions and ensures that the runner dies.
func TestRunnerErr(t *testing.T) {
funcs := []runner.Runnable{goFunc(0, true), goFunc(1, true), goFunc(2, false)}
run := runner.New(funcs)
err := run.Start()
if err != nil {
t.Error("Failed to start: ", err)
}
time.Sleep(time.Second)
run.Stop()
err = run.Start()
if err == nil {
t.Error("Should fail to start.")
}
time.Sleep(time.Second)
run.Stop()
}
