mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-06 13:47:35 +03:00
Code Issues Projects Releases Packages Wiki Activity

add tests for session.LocalAddr and session.RemoteAddr

Browse source
This commit is contained in:
Marten Seemann 2017-03-10 00:13:20 +07:00
parent 34b688b3b9
commit ab90b3f4d0
No known key found for this signature in database
GPG key ID: 3603F40B121FCDEA
1 changed files with 19 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

23
session_test.go
View file

@ -25,6 +25,7 @@ import (
type mockConnection struct {
remoteAddr net.Addr
localAddr net.Addr
written [][]byte
}
@ -39,9 +40,9 @@ func (m *mockConnection) Read([]byte) (int, net.Addr, error) { panic("not implem
func (m *mockConnection) SetCurrentRemoteAddr(addr net.Addr) {
m.remoteAddr = addr
}
func (*mockConnection) LocalAddr() net.Addr { panic("not implemented") }
func (*mockConnection) RemoteAddr() net.Addr { return &net.UDPAddr{} }
func (*mockConnection) Close() error { panic("not implemented") }
func (m *mockConnection) LocalAddr() net.Addr { return m.localAddr }
func (m *mockConnection) RemoteAddr() net.Addr { return m.remoteAddr }
func (*mockConnection) Close() error { panic("not implemented") }
type mockUnpacker struct {
unpackErr error
@ -125,7 +126,9 @@ var _ = Describe("Session", func() {
)
BeforeEach(func() {
mconn = &mockConnection{}
mconn = &mockConnection{
remoteAddr: &net.UDPAddr{},
}
closeCallbackCalled = false
certChain := crypto.NewCertChain(testdata.GetTLSConfig())
@ -1427,4 +1430,16 @@ var _ = Describe("Session", func() {
Expect(frames[0].ByteOffset).To(Equal(protocol.ReceiveConnectionFlowControlWindow * 2))
})
})
It("returns the local address", func() {
addr := &net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1337}
mconn.localAddr = addr
Expect(sess.LocalAddr()).To(Equal(addr))
})
It("returns the remote address", func() {
addr := &net.UDPAddr{IP: net.IPv4(1, 2, 7, 1), Port: 7331}
mconn.remoteAddr = addr
Expect(sess.RemoteAddr()).To(Equal(addr))
})
})