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  }