-
-
Notifications
You must be signed in to change notification settings - Fork 1.3k
/
datagram_test.go
76 lines (68 loc) · 1.98 KB
/
datagram_test.go
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package http3
import (
"context"
"errors"
"time"
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
)
var _ = Describe("Datagrams", func() {
It("receives a datagram", func() {
dg := newDatagrammer(nil)
dg.enqueue([]byte("foobar"))
data, err := dg.Receive(context.Background())
Expect(err).ToNot(HaveOccurred())
Expect(data).To(Equal([]byte("foobar")))
})
It("queues up to 32 datagrams", func() {
dg := newDatagrammer(nil)
for i := 0; i < streamDatagramQueueLen+1; i++ {
dg.enqueue([]byte{uint8(i)})
}
for i := 0; i < streamDatagramQueueLen; i++ {
data, err := dg.Receive(context.Background())
Expect(err).ToNot(HaveOccurred())
Expect(data[0]).To(BeEquivalentTo(i))
}
ctx, cancel := context.WithCancel(context.Background())
cancel()
_, err := dg.Receive(ctx)
Expect(err).To(MatchError(context.Canceled))
})
It("blocks until a new datagram is received", func() {
dg := newDatagrammer(nil)
done := make(chan struct{})
go func() {
defer GinkgoRecover()
defer close(done)
data, err := dg.Receive(context.Background())
Expect(err).ToNot(HaveOccurred())
Expect(data).To(Equal([]byte("foobar")))
}()
Consistently(done, 50*time.Millisecond).ShouldNot(BeClosed())
dg.enqueue([]byte("foobar"))
Eventually(done).Should(BeClosed())
})
It("drops datagrams when the stream's receive side is closed", func() {
dg := newDatagrammer(nil)
dg.enqueue([]byte("foo"))
testErr := errors.New("test error")
dg.SetReceiveError(testErr)
dg.enqueue([]byte("bar"))
data, err := dg.Receive(context.Background())
Expect(err).ToNot(HaveOccurred())
Expect(data).To(Equal([]byte("foo")))
_, err = dg.Receive(context.Background())
Expect(err).To(MatchError(testErr))
})
It("sends datagrams", func() {
var sent []byte
testErr := errors.New("test error")
dg := newDatagrammer(func(b []byte) error {
sent = b
return testErr
})
Expect(dg.Send([]byte("foobar"))).To(MatchError(testErr))
Expect(sent).To(Equal([]byte("foobar")))
})
})