1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
package httpctx
import (
"context"
"testing"
"time"
)
func TestWithTimeoutUnlessSet_AlreadyHasDeadline(t *testing.T) {
t.Parallel()
parent, cancel := context.WithTimeout(context.Background(), time.Hour)
defer cancel()
ctx, childCancel := WithTimeoutUnlessSet(parent, time.Nanosecond)
defer childCancel()
if ctx != parent {
t.Fatal("expected same context when parent already has deadline")
}
deadline, ok := ctx.Deadline()
if !ok {
t.Fatal("expected deadline on context")
}
if time.Until(deadline) < 30*time.Minute {
t.Fatalf("expected parent ~1h deadline preserved, got %v", time.Until(deadline))
}
}
func TestWithTimeoutUnlessSet_NoDeadline(t *testing.T) {
t.Parallel()
ctx, cancel := WithTimeoutUnlessSet(context.Background(), 50*time.Millisecond)
defer cancel()
deadline, ok := ctx.Deadline()
if !ok {
t.Fatal("expected deadline")
}
if time.Until(deadline) > time.Second {
t.Fatalf("deadline too far: %v", deadline)
}
}
func TestWithTimeoutUnlessSet_NilUsesBackground(t *testing.T) {
t.Parallel()
ctx, cancel := WithTimeoutUnlessSet(nil, 50*time.Millisecond)
defer cancel()
if err := ctx.Err(); err != nil {
t.Fatalf("context should not be done: %v", err)
}
}
|
