Function TestConsumer_Cancel

consumer_test.go:28–52

(t *testing.T)

}

func TestConsumer_Cancel(t *testing.T) {
	var stopped bool
	done := make(chan bool)

	c := newTestConsumer()

	go func() {
		<-done
		select {
		case <-c.stop:
			stopped = true
		case <-time.After(1 * time.Millisecond):
			return
		}
		done <- true
	}()

	done <- true
	c.Cancel()
	<-done

	if !stopped {
		t.Error("Cancel() should send stop signal")
	}
}

func TestConsumer_Cancel_willNotBlock(t *testing.T) {
	var ok bool