mock_stream_getter_test.go 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. // Code generated by MockGen. DO NOT EDIT.
  2. // Source: github.com/lucas-clemente/quic-go (interfaces: StreamGetter)
  3. // Package quic is a generated GoMock package.
  4. package quic
  5. import (
  6. reflect "reflect"
  7. gomock "github.com/golang/mock/gomock"
  8. protocol "github.com/lucas-clemente/quic-go/internal/protocol"
  9. )
  10. // MockStreamGetter is a mock of StreamGetter interface
  11. type MockStreamGetter struct {
  12. ctrl *gomock.Controller
  13. recorder *MockStreamGetterMockRecorder
  14. }
  15. // MockStreamGetterMockRecorder is the mock recorder for MockStreamGetter
  16. type MockStreamGetterMockRecorder struct {
  17. mock *MockStreamGetter
  18. }
  19. // NewMockStreamGetter creates a new mock instance
  20. func NewMockStreamGetter(ctrl *gomock.Controller) *MockStreamGetter {
  21. mock := &MockStreamGetter{ctrl: ctrl}
  22. mock.recorder = &MockStreamGetterMockRecorder{mock}
  23. return mock
  24. }
  25. // EXPECT returns an object that allows the caller to indicate expected use
  26. func (m *MockStreamGetter) EXPECT() *MockStreamGetterMockRecorder {
  27. return m.recorder
  28. }
  29. // GetOrOpenReceiveStream mocks base method
  30. func (m *MockStreamGetter) GetOrOpenReceiveStream(arg0 protocol.StreamID) (receiveStreamI, error) {
  31. m.ctrl.T.Helper()
  32. ret := m.ctrl.Call(m, "GetOrOpenReceiveStream", arg0)
  33. ret0, _ := ret[0].(receiveStreamI)
  34. ret1, _ := ret[1].(error)
  35. return ret0, ret1
  36. }
  37. // GetOrOpenReceiveStream indicates an expected call of GetOrOpenReceiveStream
  38. func (mr *MockStreamGetterMockRecorder) GetOrOpenReceiveStream(arg0 interface{}) *gomock.Call {
  39. mr.mock.ctrl.T.Helper()
  40. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrOpenReceiveStream", reflect.TypeOf((*MockStreamGetter)(nil).GetOrOpenReceiveStream), arg0)
  41. }
  42. // GetOrOpenSendStream mocks base method
  43. func (m *MockStreamGetter) GetOrOpenSendStream(arg0 protocol.StreamID) (sendStreamI, error) {
  44. m.ctrl.T.Helper()
  45. ret := m.ctrl.Call(m, "GetOrOpenSendStream", arg0)
  46. ret0, _ := ret[0].(sendStreamI)
  47. ret1, _ := ret[1].(error)
  48. return ret0, ret1
  49. }
  50. // GetOrOpenSendStream indicates an expected call of GetOrOpenSendStream
  51. func (mr *MockStreamGetterMockRecorder) GetOrOpenSendStream(arg0 interface{}) *gomock.Call {
  52. mr.mock.ctrl.T.Helper()
  53. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrOpenSendStream", reflect.TypeOf((*MockStreamGetter)(nil).GetOrOpenSendStream), arg0)
  54. }