Function TestThrottleContextTimeout

ch04/throttle_test.go:151–176

TestThrottleContextTimeout tests whether a Throttle will return an error when its context is canceled.

(t *testing.T)

// TestThrottleContextTimeout tests whether a Throttle will return an error
// when its context is canceled.
func TestThrottleContextTimeout(t *testing.T) {
	callsCounter := 0
	effector := callsCountFunction(&callsCounter)

	ctx, cancel := context.WithTimeout(context.Background(), 250*time.Millisecond)
	defer cancel()

	throttle := Throttle(effector, 1, 1, time.Second)

	s, e := throttle(ctx)
	if e != nil {
		t.Error("unexpected error:", e)
	} else {
		t.Log("output:", s)
	}

	// Wait for timeout
	time.Sleep(300 * time.Millisecond)

	_, e = throttle(ctx)
	if e != nil {
		t.Log("got expected error:", e)
	} else {
		t.Error("didn't get expected error")
	}
}