mock_dir_search_service_test.go 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
  1. // Code generated by mockery; DO NOT EDIT.
  2. // github.com/vektra/mockery
  3. // template: testify
  4. package toc
  5. import (
  6. "context"
  7. "github.com/mk6i/open-oscar-server/wire"
  8. mock "github.com/stretchr/testify/mock"
  9. )
  10. // newMockDirSearchService creates a new instance of mockDirSearchService. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  11. // The first argument is typically a *testing.T value.
  12. func newMockDirSearchService(t interface {
  13. mock.TestingT
  14. Cleanup(func())
  15. }) *mockDirSearchService {
  16. if helper, ok := t.(interface{ Helper() }); ok {
  17. helper.Helper()
  18. }
  19. mock := &mockDirSearchService{}
  20. mock.Mock.Test(t)
  21. t.Cleanup(func() {
  22. if helper, ok := t.(interface{ Helper() }); ok {
  23. helper.Helper()
  24. }
  25. mock.AssertExpectations(t)
  26. })
  27. return mock
  28. }
  29. // mockDirSearchService is an autogenerated mock type for the DirSearchService type
  30. type mockDirSearchService struct {
  31. mock.Mock
  32. }
  33. type mockDirSearchService_Expecter struct {
  34. mock *mock.Mock
  35. }
  36. func (_m *mockDirSearchService) EXPECT() *mockDirSearchService_Expecter {
  37. return &mockDirSearchService_Expecter{mock: &_m.Mock}
  38. }
  39. // InfoQuery provides a mock function for the type mockDirSearchService
  40. func (_mock *mockDirSearchService) InfoQuery(ctx context.Context, inFrame wire.SNACFrame, inBody wire.SNAC_0x0F_0x02_InfoQuery) (wire.SNACMessage, error) {
  41. ret := _mock.Called(ctx, inFrame, inBody)
  42. if len(ret) == 0 {
  43. panic("no return value specified for InfoQuery")
  44. }
  45. var r0 wire.SNACMessage
  46. var r1 error
  47. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNACFrame, wire.SNAC_0x0F_0x02_InfoQuery) (wire.SNACMessage, error)); ok {
  48. return returnFunc(ctx, inFrame, inBody)
  49. }
  50. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNACFrame, wire.SNAC_0x0F_0x02_InfoQuery) wire.SNACMessage); ok {
  51. r0 = returnFunc(ctx, inFrame, inBody)
  52. } else {
  53. r0 = ret.Get(0).(wire.SNACMessage)
  54. }
  55. if returnFunc, ok := ret.Get(1).(func(context.Context, wire.SNACFrame, wire.SNAC_0x0F_0x02_InfoQuery) error); ok {
  56. r1 = returnFunc(ctx, inFrame, inBody)
  57. } else {
  58. r1 = ret.Error(1)
  59. }
  60. return r0, r1
  61. }
  62. // mockDirSearchService_InfoQuery_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'InfoQuery'
  63. type mockDirSearchService_InfoQuery_Call struct {
  64. *mock.Call
  65. }
  66. // InfoQuery is a helper method to define mock.On call
  67. // - ctx context.Context
  68. // - inFrame wire.SNACFrame
  69. // - inBody wire.SNAC_0x0F_0x02_InfoQuery
  70. func (_e *mockDirSearchService_Expecter) InfoQuery(ctx any, inFrame any, inBody any) *mockDirSearchService_InfoQuery_Call {
  71. return &mockDirSearchService_InfoQuery_Call{Call: _e.mock.On("InfoQuery", ctx, inFrame, inBody)}
  72. }
  73. func (_c *mockDirSearchService_InfoQuery_Call) Run(run func(ctx context.Context, inFrame wire.SNACFrame, inBody wire.SNAC_0x0F_0x02_InfoQuery)) *mockDirSearchService_InfoQuery_Call {
  74. _c.Call.Run(func(args mock.Arguments) {
  75. var arg0 context.Context
  76. if args[0] != nil {
  77. arg0 = args[0].(context.Context)
  78. }
  79. var arg1 wire.SNACFrame
  80. if args[1] != nil {
  81. arg1 = args[1].(wire.SNACFrame)
  82. }
  83. var arg2 wire.SNAC_0x0F_0x02_InfoQuery
  84. if args[2] != nil {
  85. arg2 = args[2].(wire.SNAC_0x0F_0x02_InfoQuery)
  86. }
  87. run(
  88. arg0,
  89. arg1,
  90. arg2,
  91. )
  92. })
  93. return _c
  94. }
  95. func (_c *mockDirSearchService_InfoQuery_Call) Return(sNACMessage wire.SNACMessage, err error) *mockDirSearchService_InfoQuery_Call {
  96. _c.Call.Return(sNACMessage, err)
  97. return _c
  98. }
  99. func (_c *mockDirSearchService_InfoQuery_Call) RunAndReturn(run func(ctx context.Context, inFrame wire.SNACFrame, inBody wire.SNAC_0x0F_0x02_InfoQuery) (wire.SNACMessage, error)) *mockDirSearchService_InfoQuery_Call {
  100. _c.Call.Return(run)
  101. return _c
  102. }