Function TestStripDeadline

context2/context_test.go:106–187

(outerT *testing.T)

}

func TestStripDeadline(outerT *testing.T) {
	outerT.Run("DeadlineDoesntPropagate", func(t *testing.T) {
		timeout := 1 * time.Second
		ctx, cancel := context.WithTimeout(context.Background(), timeout)
		defer cancel()

		procrastinator := StripDeadline(ctx)

		deadline, ok := procrastinator.Deadline()
		if ok {
			t.Fatal("Procrastinator had a deadline:", deadline)
		}
	})

	outerT.Run("DeadlineCausesCancel", func(t *testing.T) {
		timeout := 1 * time.Second
		ctx, cancel := context.WithTimeout(context.Background(), timeout)
		defer cancel()

		procrastinator := StripDeadline(ctx)

		<-procrastinator.Done()

		if procrastinator.Err() == nil {
			t.Fatal("Expected error")
		}
	})

	outerT.Run("CancelPropDown", func(t *testing.T) {
		parent, parentCancel := context.WithCancel(context.Background())
		procrastinator := StripDeadline(parent)
		child, childCancel := context.WithCancel(procrastinator)
		defer childCancel()

		parentCancel()

		<-child.Done()
	})

	outerT.Run("CancelPropUp", func(t *testing.T) {
		parent, parentCancel := context.WithCancel(context.Background())
		procrastinator := StripDeadline(parent)
		_, childCancel := context.WithCancel(procrastinator)

		childCancel()

		select {
		case <-procrastinator.Done():
			t.Fatal("Procrastinator got cancelled.")
		case <-parent.Done():
			t.Fatal("Parent got cancelled.")
		default:
		}

		parentCancel()

		<-procrastinator.Done()
	})