gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

fix flaky proxy tests

Browse Source 
fixes #275
This commit is contained in:
Marten Seemann
2016-08-16 13:39:26 +07:00
parent 5c80eee490
commit 2d98933d69
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
integrationtests/proxy/udp_proxy_test.go
View File

@@ -4,6 +4,7 @@ import (
"bytes" "bytes"
"net" "net"
"strconv" "strconv"
"sync/atomic"
"time" "time"
"github.com/lucas-clemente/quic-go" "github.com/lucas-clemente/quic-go"
@@ -130,7 +131,7 @@ var _ = Describe("UDP Proxy", func() {
Eventually(func() map[string]*connection { return proxy.clientDict }).Should(HaveLen(1)) Eventually(func() map[string]*connection { return proxy.clientDict }).Should(HaveLen(1))
var conn *connection var conn *connection
for _, conn = range proxy.clientDict { for _, conn = range proxy.clientDict {
Expect(conn.incomingPacketCounter).To(Equal(uint64(1))) Expect(atomic.LoadUint64(&conn.incomingPacketCounter)).To(Equal(uint64(1)))
} }
_, err = clientConn.Write(makePacket(2, []byte("decafbad"))) _, err = clientConn.Write(makePacket(2, []byte("decafbad")))
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
@@ -147,7 +148,7 @@ var _ = Describe("UDP Proxy", func() {
var key string var key string
var conn *connection var conn *connection
for key, conn = range proxy.clientDict { for key, conn = range proxy.clientDict {
Eventually(conn.outgoingPacketCounter).Should(Equal(uint64(1))) Eventually(func() uint64 { return atomic.LoadUint64(&conn.outgoingPacketCounter) }).Should(Equal(uint64(1)))
} }
_, err = clientConn.Write(makePacket(2, []byte("decafbad"))) _, err = clientConn.Write(makePacket(2, []byte("decafbad")))
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())