github.com/kaleido-io/firefly@v0.0.0-20210622132723-8b4b6aacb971/mocks/wsmocks/ws_client.go (about)

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  package wsmocks
     4  
     5  import (
     6  	context "context"
     7  
     8  	mock "github.com/stretchr/testify/mock"
     9  )
    10  
    11  // WSClient is an autogenerated mock type for the WSClient type
    12  type WSClient struct {
    13  	mock.Mock
    14  }
    15  
    16  // Close provides a mock function with given fields:
    17  func (_m *WSClient) Close() {
    18  	_m.Called()
    19  }
    20  
    21  // Connect provides a mock function with given fields:
    22  func (_m *WSClient) Connect() error {
    23  	ret := _m.Called()
    24  
    25  	var r0 error
    26  	if rf, ok := ret.Get(0).(func() error); ok {
    27  		r0 = rf()
    28  	} else {
    29  		r0 = ret.Error(0)
    30  	}
    31  
    32  	return r0
    33  }
    34  
    35  // Receive provides a mock function with given fields:
    36  func (_m *WSClient) Receive() <-chan []byte {
    37  	ret := _m.Called()
    38  
    39  	var r0 <-chan []byte
    40  	if rf, ok := ret.Get(0).(func() <-chan []byte); ok {
    41  		r0 = rf()
    42  	} else {
    43  		if ret.Get(0) != nil {
    44  			r0 = ret.Get(0).(<-chan []byte)
    45  		}
    46  	}
    47  
    48  	return r0
    49  }
    50  
    51  // Send provides a mock function with given fields: ctx, message
    52  func (_m *WSClient) Send(ctx context.Context, message []byte) error {
    53  	ret := _m.Called(ctx, message)
    54  
    55  	var r0 error
    56  	if rf, ok := ret.Get(0).(func(context.Context, []byte) error); ok {
    57  		r0 = rf(ctx, message)
    58  	} else {
    59  		r0 = ret.Error(0)
    60  	}
    61  
    62  	return r0
    63  }
    64  
    65  // SetURL provides a mock function with given fields: url
    66  func (_m *WSClient) SetURL(url string) {
    67  	_m.Called(url)
    68  }
    69  
    70  // URL provides a mock function with given fields:
    71  func (_m *WSClient) URL() string {
    72  	ret := _m.Called()
    73  
    74  	var r0 string
    75  	if rf, ok := ret.Get(0).(func() string); ok {
    76  		r0 = rf()
    77  	} else {
    78  		r0 = ret.Get(0).(string)
    79  	}
    80  
    81  	return r0
    82  }