remove udp references from session to simplify testing

This commit is contained in:
Lucas Clemente
2016-04-26 18:00:41 +02:00
parent 8339f210cb
commit a5a06a25c2
5 changed files with 51 additions and 20 deletions

View File

@@ -17,14 +17,14 @@ type mockSession struct {
packetCount int
}
func (s *mockSession) HandlePacket(addr *net.UDPAddr, publicHeader *PublicHeader, r *bytes.Reader) {
func (s *mockSession) HandlePacket(addr interface{}, publicHeader *PublicHeader, r *bytes.Reader) {
s.packetCount++
}
func (s *mockSession) Run() {
}
func newMockSession(conn *net.UDPConn, v protocol.VersionNumber, connectionID protocol.ConnectionID, sCfg *handshake.ServerConfig, streamCallback StreamCallback) PacketHandler {
func newMockSession(conn connection, v protocol.VersionNumber, connectionID protocol.ConnectionID, sCfg *handshake.ServerConfig, streamCallback StreamCallback) PacketHandler {
return &mockSession{
connectionID: connectionID,
}