github.com/xmidt-org/webpa-common@v1.11.9/xlistener/mocks_test.go (about) 1 package xlistener 2 3 import ( 4 "net" 5 "time" 6 7 "github.com/stretchr/testify/mock" 8 ) 9 10 type mockConn struct { 11 mock.Mock 12 } 13 14 func (m *mockConn) Read(b []byte) (int, error) { 15 arguments := m.Called(b) 16 return arguments.Int(0), arguments.Error(1) 17 } 18 19 func (m *mockConn) Write(b []byte) (int, error) { 20 arguments := m.Called(b) 21 return arguments.Int(0), arguments.Error(1) 22 } 23 24 func (m *mockConn) Close() error { 25 return m.Called().Error(0) 26 } 27 28 func (m *mockConn) LocalAddr() net.Addr { 29 return m.Called().Get(0).(net.Addr) 30 } 31 32 func (m *mockConn) RemoteAddr() net.Addr { 33 return m.Called().Get(0).(net.Addr) 34 } 35 36 func (m *mockConn) SetDeadline(t time.Time) error { 37 return m.Called(t).Error(0) 38 } 39 40 func (m *mockConn) SetReadDeadline(t time.Time) error { 41 return m.Called(t).Error(0) 42 } 43 44 func (m *mockConn) SetWriteDeadline(t time.Time) error { 45 return m.Called(t).Error(0) 46 } 47 48 type mockListener struct { 49 mock.Mock 50 } 51 52 func (m *mockListener) Accept() (net.Conn, error) { 53 arguments := m.Called() 54 first, _ := arguments.Get(0).(net.Conn) 55 return first, arguments.Error(1) 56 } 57 58 func (m *mockListener) Close() error { 59 return m.Called().Error(0) 60 } 61 62 func (m *mockListener) Addr() net.Addr { 63 return m.Called().Get(0).(net.Addr) 64 }