-
Notifications
You must be signed in to change notification settings - Fork 6
/
spdystream_test.go
115 lines (104 loc) · 2.83 KB
/
spdystream_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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package peerstream_spdystream
import (
"io"
"net"
"time"
smux "github.com/libp2p/go-stream-muxer"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("SPDY", func() {
var server, client smux.Conn
Context("accepting streams", func() {
BeforeEach(func() {
// start the server
serverReady := make(chan struct{})
serverAddr := make(chan net.Addr)
go func() {
defer GinkgoRecover()
l, err := net.Listen("tcp", "localhost:0")
Expect(err).ToNot(HaveOccurred())
serverAddr <- l.Addr()
c, err := l.Accept()
Expect(err).ToNot(HaveOccurred())
server, err = Transport.NewConn(c, true)
Expect(err).ToNot(HaveOccurred())
close(serverReady)
}()
// start the client
addr := <-serverAddr
nconn, err := net.Dial("tcp", addr.String())
Expect(err).ToNot(HaveOccurred())
client, err = Transport.NewConn(nconn, false)
Expect(err).ToNot(HaveOccurred())
<-serverReady
})
AfterEach(func() {
server.Close()
client.Close()
})
It("returns an error when the connection is closed", func() {
done := make(chan struct{})
go func() {
defer GinkgoRecover()
_, err := server.AcceptStream()
Expect(err).To(MatchError(errClosed))
close(done)
}()
server.Close()
Eventually(done).Should(BeClosed())
})
It("returns an error when the connection is closed, even if there are streams in the queue", func() {
_, err := client.OpenStream()
Expect(err).ToNot(HaveOccurred())
server.Close()
_, err = server.AcceptStream()
Expect(err).To(MatchError(errClosed))
})
It("waits for new streams", func() {
done := make(chan struct{})
go func() {
defer GinkgoRecover()
server.AcceptStream()
close(done)
}()
Consistently(done).ShouldNot(BeClosed())
// kill the goroutine, so that the race detector is happy
server.Close()
Eventually(done).Should(BeClosed())
})
It("accepts a new stream", func() {
done := make(chan struct{})
go func() {
defer GinkgoRecover()
str, err := server.AcceptStream()
Expect(err).ToNot(HaveOccurred())
_, err = io.Copy(str, str)
Expect(err).ToNot(HaveOccurred())
close(done)
}()
str, err := client.OpenStream()
Expect(err).ToNot(HaveOccurred())
_, err = str.Write([]byte("foobar"))
Expect(err).ToNot(HaveOccurred())
b := make([]byte, 6)
_, err = str.Read(b)
Expect(err).ToNot(HaveOccurred())
Expect(b).To(Equal([]byte("foobar")))
str.Close()
Eventually(done).Should(BeClosed())
})
It("accepts multiple streams, that were opened before AcceptStream was called", func() {
n := 5
for i := 0; i < n; i++ {
_, err := client.OpenStream()
Expect(err).ToNot(HaveOccurred())
}
time.Sleep(50 * time.Millisecond)
for i := 0; i < n; i++ {
_, err := server.AcceptStream()
Expect(err).ToNot(HaveOccurred())
}
})
})
})