mock_stream_getter_test.go 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  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. ret := m.ctrl.Call(m, "GetOrOpenReceiveStream", arg0)
  32. ret0, _ := ret[0].(receiveStreamI)
  33. ret1, _ := ret[1].(error)
  34. return ret0, ret1
  35. }
  36. // GetOrOpenReceiveStream indicates an expected call of GetOrOpenReceiveStream
  37. func (mr *MockStreamGetterMockRecorder) GetOrOpenReceiveStream(arg0 interface{}) *gomock.Call {
  38. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrOpenReceiveStream", reflect.TypeOf((*MockStreamGetter)(nil).GetOrOpenReceiveStream), arg0)
  39. }
  40. // GetOrOpenSendStream mocks base method
  41. func (m *MockStreamGetter) GetOrOpenSendStream(arg0 protocol.StreamID) (sendStreamI, error) {
  42. ret := m.ctrl.Call(m, "GetOrOpenSendStream", arg0)
  43. ret0, _ := ret[0].(sendStreamI)
  44. ret1, _ := ret[1].(error)
  45. return ret0, ret1
  46. }
  47. // GetOrOpenSendStream indicates an expected call of GetOrOpenSendStream
  48. func (mr *MockStreamGetterMockRecorder) GetOrOpenSendStream(arg0 interface{}) *gomock.Call {
  49. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrOpenSendStream", reflect.TypeOf((*MockStreamGetter)(nil).GetOrOpenSendStream), arg0)
  50. }