github.com/xmidt-org/webpa-common@v1.11.9/xhttp/mocks_test.go (about)

     1  package xhttp
     2  
     3  import (
     4  	"net"
     5  
     6  	"github.com/stretchr/testify/mock"
     7  )
     8  
     9  type mockReader struct {
    10  	mock.Mock
    11  }
    12  
    13  func (m *mockReader) Read(b []byte) (int, error) {
    14  	arguments := m.Called(b)
    15  	return arguments.Int(0), arguments.Error(1)
    16  }
    17  
    18  type mockHTTPServer struct {
    19  	mock.Mock
    20  }
    21  
    22  func (m *mockHTTPServer) ListenAndServe() error {
    23  	return m.Called().Error(0)
    24  }
    25  
    26  func (m *mockHTTPServer) ListenAndServeTLS(certificateFile, keyFile string) error {
    27  	return m.Called(certificateFile, keyFile).Error(0)
    28  }
    29  
    30  func (m *mockHTTPServer) Serve(l net.Listener) error {
    31  	return m.Called(l).Error(0)
    32  }
    33  
    34  func (m *mockHTTPServer) ServeTLS(l net.Listener, certificateFile, keyFile string) error {
    35  	return m.Called(l, certificateFile, keyFile).Error(0)
    36  }
    37  
    38  func (m *mockHTTPServer) SetKeepAlivesEnabled(v bool) {
    39  	m.Called(v)
    40  }
    41  
    42  type mockListener struct {
    43  	mock.Mock
    44  }
    45  
    46  func (m *mockListener) Accept() (net.Conn, error) {
    47  	arguments := m.Called()
    48  	first, _ := arguments.Get(0).(net.Conn)
    49  	return first, arguments.Error(1)
    50  }
    51  
    52  func (m *mockListener) Close() error {
    53  	return m.Called().Error(0)
    54  }
    55  
    56  func (m *mockListener) Addr() net.Addr {
    57  	return m.Called().Get(0).(net.Addr)
    58  }