129 lines
4.0 KiB
Go
129 lines
4.0 KiB
Go
package netxmocks
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"net"
|
|
|
|
"github.com/lucas-clemente/quic-go"
|
|
)
|
|
|
|
// QUICListener is a mockable netxlite.QUICListener.
|
|
type QUICListener struct {
|
|
MockListen func(addr *net.UDPAddr) (net.PacketConn, error)
|
|
}
|
|
|
|
// Listen calls MockListen.
|
|
func (ql *QUICListener) Listen(addr *net.UDPAddr) (net.PacketConn, error) {
|
|
return ql.MockListen(addr)
|
|
}
|
|
|
|
// QUICContextDialer is a mockable netxlite.QUICContextDialer.
|
|
type QUICContextDialer struct {
|
|
MockDialContext func(ctx context.Context, network, address string,
|
|
tlsConfig *tls.Config, quicConfig *quic.Config) (quic.EarlySession, error)
|
|
}
|
|
|
|
// DialContext calls MockDialContext.
|
|
func (qcd *QUICContextDialer) DialContext(ctx context.Context, network, address string,
|
|
tlsConfig *tls.Config, quicConfig *quic.Config) (quic.EarlySession, error) {
|
|
return qcd.MockDialContext(ctx, network, address, tlsConfig, quicConfig)
|
|
}
|
|
|
|
// QUICEarlySession is a mockable quic.EarlySession.
|
|
type QUICEarlySession struct {
|
|
MockAcceptStream func(context.Context) (quic.Stream, error)
|
|
MockAcceptUniStream func(context.Context) (quic.ReceiveStream, error)
|
|
MockOpenStream func() (quic.Stream, error)
|
|
MockOpenStreamSync func(ctx context.Context) (quic.Stream, error)
|
|
MockOpenUniStream func() (quic.SendStream, error)
|
|
MockOpenUniStreamSync func(ctx context.Context) (quic.SendStream, error)
|
|
MockLocalAddr func() net.Addr
|
|
MockRemoteAddr func() net.Addr
|
|
MockCloseWithError func(code quic.ApplicationErrorCode, reason string) error
|
|
MockContext func() context.Context
|
|
MockConnectionState func() quic.ConnectionState
|
|
MockHandshakeComplete func() context.Context
|
|
MockNextSession func() quic.Session
|
|
MockSendMessage func(b []byte) error
|
|
MockReceiveMessage func() ([]byte, error)
|
|
}
|
|
|
|
var _ quic.EarlySession = &QUICEarlySession{}
|
|
|
|
// AcceptStream calls MockAcceptStream.
|
|
func (s *QUICEarlySession) AcceptStream(ctx context.Context) (quic.Stream, error) {
|
|
return s.MockAcceptStream(ctx)
|
|
}
|
|
|
|
// AcceptUniStream calls MockAcceptUniStream.
|
|
func (s *QUICEarlySession) AcceptUniStream(ctx context.Context) (quic.ReceiveStream, error) {
|
|
return s.MockAcceptUniStream(ctx)
|
|
}
|
|
|
|
// OpenStream calls MockOpenStream.
|
|
func (s *QUICEarlySession) OpenStream() (quic.Stream, error) {
|
|
return s.MockOpenStream()
|
|
}
|
|
|
|
// OpenStreamSync calls MockOpenStreamSync.
|
|
func (s *QUICEarlySession) OpenStreamSync(ctx context.Context) (quic.Stream, error) {
|
|
return s.MockOpenStreamSync(ctx)
|
|
}
|
|
|
|
// OpenUniStream calls MockOpenUniStream.
|
|
func (s *QUICEarlySession) OpenUniStream() (quic.SendStream, error) {
|
|
return s.MockOpenUniStream()
|
|
}
|
|
|
|
// OpenUniStreamSync calls MockOpenUniStreamSync.
|
|
func (s *QUICEarlySession) OpenUniStreamSync(ctx context.Context) (quic.SendStream, error) {
|
|
return s.MockOpenUniStreamSync(ctx)
|
|
}
|
|
|
|
// LocalAddr class MockLocalAddr.
|
|
func (c *QUICEarlySession) LocalAddr() net.Addr {
|
|
return c.MockLocalAddr()
|
|
}
|
|
|
|
// RemoteAddr calls MockRemoteAddr.
|
|
func (c *QUICEarlySession) RemoteAddr() net.Addr {
|
|
return c.MockRemoteAddr()
|
|
}
|
|
|
|
// CloseWithError calls MockCloseWithError.
|
|
func (c *QUICEarlySession) CloseWithError(
|
|
code quic.ApplicationErrorCode, reason string) error {
|
|
return c.MockCloseWithError(code, reason)
|
|
}
|
|
|
|
// Context calls MockContext.
|
|
func (s *QUICEarlySession) Context() context.Context {
|
|
return s.MockContext()
|
|
}
|
|
|
|
// ConnectionState calls MockConnectionState.
|
|
func (s *QUICEarlySession) ConnectionState() quic.ConnectionState {
|
|
return s.MockConnectionState()
|
|
}
|
|
|
|
// HandshakeComplete calls MockHandshakeComplete.
|
|
func (s *QUICEarlySession) HandshakeComplete() context.Context {
|
|
return s.MockHandshakeComplete()
|
|
}
|
|
|
|
// NextSession calls MockNextSession.
|
|
func (s *QUICEarlySession) NextSession() quic.Session {
|
|
return s.MockNextSession()
|
|
}
|
|
|
|
// SendMessage calls MockSendMessage.
|
|
func (s *QUICEarlySession) SendMessage(b []byte) error {
|
|
return s.MockSendMessage(b)
|
|
}
|
|
|
|
// ReceiveMessage calls MockReceiveMessage.
|
|
func (s *QUICEarlySession) ReceiveMessage() ([]byte, error) {
|
|
return s.MockReceiveMessage()
|
|
}
|