@@ -12,8 +12,8 @@ import (
12
12
tpt "github.com/libp2p/go-libp2p-transport"
13
13
st "github.com/libp2p/go-libp2p-transport-upgrader"
14
14
smux "github.com/libp2p/go-stream-muxer"
15
- tcp "github.com/libp2p/go-tcp-transport"
16
15
ma "github.com/multiformats/go-multiaddr"
16
+ manet "github.com/multiformats/go-multiaddr-net"
17
17
yamux "github.com/whyrusleeping/go-smux-yamux"
18
18
19
19
. "github.com/onsi/ginkgo"
@@ -85,9 +85,17 @@ var _ = Describe("Listener", func() {
85
85
createListener := func (upgrader * st.Upgrader ) tpt.Listener {
86
86
addr , err := ma .NewMultiaddr ("/ip4/0.0.0.0/tcp/0" )
87
87
ExpectWithOffset (0 , err ).ToNot (HaveOccurred ())
88
- ln , err := tcp . NewTCPTransport ( upgrader ) .Listen (addr )
88
+ ln , err := manet .Listen (addr )
89
89
ExpectWithOffset (0 , err ).ToNot (HaveOccurred ())
90
- return ln
90
+ return upgrader .UpgradeListener (nil , ln )
91
+ }
92
+
93
+ dial := func (upgrader * st.Upgrader , raddr ma.Multiaddr , p peer.ID ) (tpt.Conn , error ) {
94
+ macon , err := manet .Dial (raddr )
95
+ if err != nil {
96
+ return nil , err
97
+ }
98
+ return upgrader .UpgradeOutbound (context .Background (), nil , macon , p )
91
99
}
92
100
93
101
BeforeEach (func () {
@@ -96,7 +104,7 @@ var _ = Describe("Listener", func() {
96
104
97
105
It ("accepts a single connection" , func () {
98
106
ln := createListener (defaultUpgrader )
99
- cconn , err := tcp . NewTCPTransport (defaultUpgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (1 ))
107
+ cconn , err := dial (defaultUpgrader , ln .Multiaddr (), peer .ID (1 ))
100
108
Expect (err ).ToNot (HaveOccurred ())
101
109
sconn , err := ln .Accept ()
102
110
Expect (err ).ToNot (HaveOccurred ())
@@ -107,7 +115,7 @@ var _ = Describe("Listener", func() {
107
115
ln := createListener (defaultUpgrader )
108
116
const num = 10
109
117
for i := 0 ; i < 10 ; i ++ {
110
- cconn , err := tcp . NewTCPTransport (defaultUpgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (1 ))
118
+ cconn , err := dial (defaultUpgrader , ln .Multiaddr (), peer .ID (1 ))
111
119
Expect (err ).ToNot (HaveOccurred ())
112
120
sconn , err := ln .Accept ()
113
121
Expect (err ).ToNot (HaveOccurred ())
@@ -119,7 +127,7 @@ var _ = Describe("Listener", func() {
119
127
const timeout = 200 * time .Millisecond
120
128
tpt .AcceptTimeout = timeout
121
129
ln := createListener (defaultUpgrader )
122
- conn , err := tcp . NewTCPTransport (defaultUpgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (2 ))
130
+ conn , err := dial (defaultUpgrader , ln .Multiaddr (), peer .ID (2 ))
123
131
Expect (err ).ToNot (HaveOccurred ())
124
132
done := make (chan struct {})
125
133
go func () {
@@ -146,7 +154,7 @@ var _ = Describe("Listener", func() {
146
154
_ , _ = ln .Accept ()
147
155
close (done )
148
156
}()
149
- _ , _ = tcp . NewTCPTransport ( upgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (2 ))
157
+ _ , _ = dial ( defaultUpgrader , ln .Multiaddr (), peer .ID (2 ))
150
158
Consistently (done ).ShouldNot (BeClosed ())
151
159
// make the goroutine return
152
160
ln .Close ()
@@ -179,7 +187,7 @@ var _ = Describe("Listener", func() {
179
187
wg .Add (1 )
180
188
go func () {
181
189
defer GinkgoRecover ()
182
- _ , err := tcp . NewTCPTransport ( upgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (2 ))
190
+ _ , err := dial ( defaultUpgrader , ln .Multiaddr (), peer .ID (2 ))
183
191
Expect (err ).ToNot (HaveOccurred ())
184
192
wg .Done ()
185
193
}()
@@ -198,7 +206,7 @@ var _ = Describe("Listener", func() {
198
206
for i := 0 ; i < st .AcceptQueueLength ; i ++ {
199
207
go func () {
200
208
defer GinkgoRecover ()
201
- _ , err := tcp . NewTCPTransport (defaultUpgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (2 ))
209
+ _ , err := dial (defaultUpgrader , ln .Multiaddr (), peer .ID (2 ))
202
210
Expect (err ).ToNot (HaveOccurred ())
203
211
dialed <- struct {}{}
204
212
}()
@@ -207,7 +215,7 @@ var _ = Describe("Listener", func() {
207
215
// dial a new connection. This connection should not complete setup, since the queue is full
208
216
go func () {
209
217
defer GinkgoRecover ()
210
- _ , err := tcp . NewTCPTransport (defaultUpgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (2 ))
218
+ _ , err := dial (defaultUpgrader , ln .Multiaddr (), peer .ID (2 ))
211
219
Expect (err ).ToNot (HaveOccurred ())
212
220
dialed <- struct {}{}
213
221
}()
@@ -238,13 +246,13 @@ var _ = Describe("Listener", func() {
238
246
It ("doesn't accept new connections when it is closed" , func () {
239
247
ln := createListener (defaultUpgrader )
240
248
Expect (ln .Close ()).To (Succeed ())
241
- _ , err := tcp . NewTCPTransport (defaultUpgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (1 ))
249
+ _ , err := dial (defaultUpgrader , ln .Multiaddr (), peer .ID (1 ))
242
250
Expect (err ).To (HaveOccurred ())
243
251
})
244
252
245
253
It ("closes incoming connections that have not yet been accepted" , func () {
246
254
ln := createListener (defaultUpgrader )
247
- conn , err := tcp . NewTCPTransport (defaultUpgrader ). Dial ( context . Background () , ln .Multiaddr (), peer .ID (2 ))
255
+ conn , err := dial (defaultUpgrader , ln .Multiaddr (), peer .ID (2 ))
248
256
Expect (conn .IsClosed ()).To (BeFalse ())
249
257
Expect (err ).ToNot (HaveOccurred ())
250
258
Expect (ln .Close ()).To (Succeed ())
0 commit comments