mock_bart_test.go 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. // Code generated by mockery v2.52.1. DO NOT EDIT.
  2. package handler
  3. import (
  4. context "context"
  5. state "github.com/mk6i/retro-aim-server/state"
  6. mock "github.com/stretchr/testify/mock"
  7. wire "github.com/mk6i/retro-aim-server/wire"
  8. )
  9. // mockBARTService is an autogenerated mock type for the BARTService type
  10. type mockBARTService struct {
  11. mock.Mock
  12. }
  13. type mockBARTService_Expecter struct {
  14. mock *mock.Mock
  15. }
  16. func (_m *mockBARTService) EXPECT() *mockBARTService_Expecter {
  17. return &mockBARTService_Expecter{mock: &_m.Mock}
  18. }
  19. // RetrieveItem provides a mock function with given fields: ctx, sess, inFrame, inBody
  20. func (_m *mockBARTService) RetrieveItem(ctx context.Context, sess *state.Session, inFrame wire.SNACFrame, inBody wire.SNAC_0x10_0x04_BARTDownloadQuery) (wire.SNACMessage, error) {
  21. ret := _m.Called(ctx, sess, inFrame, inBody)
  22. if len(ret) == 0 {
  23. panic("no return value specified for RetrieveItem")
  24. }
  25. var r0 wire.SNACMessage
  26. var r1 error
  27. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x10_0x04_BARTDownloadQuery) (wire.SNACMessage, error)); ok {
  28. return rf(ctx, sess, inFrame, inBody)
  29. }
  30. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x10_0x04_BARTDownloadQuery) wire.SNACMessage); ok {
  31. r0 = rf(ctx, sess, inFrame, inBody)
  32. } else {
  33. r0 = ret.Get(0).(wire.SNACMessage)
  34. }
  35. if rf, ok := ret.Get(1).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x10_0x04_BARTDownloadQuery) error); ok {
  36. r1 = rf(ctx, sess, inFrame, inBody)
  37. } else {
  38. r1 = ret.Error(1)
  39. }
  40. return r0, r1
  41. }
  42. // mockBARTService_RetrieveItem_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RetrieveItem'
  43. type mockBARTService_RetrieveItem_Call struct {
  44. *mock.Call
  45. }
  46. // RetrieveItem is a helper method to define mock.On call
  47. // - ctx context.Context
  48. // - sess *state.Session
  49. // - inFrame wire.SNACFrame
  50. // - inBody wire.SNAC_0x10_0x04_BARTDownloadQuery
  51. func (_e *mockBARTService_Expecter) RetrieveItem(ctx interface{}, sess interface{}, inFrame interface{}, inBody interface{}) *mockBARTService_RetrieveItem_Call {
  52. return &mockBARTService_RetrieveItem_Call{Call: _e.mock.On("RetrieveItem", ctx, sess, inFrame, inBody)}
  53. }
  54. func (_c *mockBARTService_RetrieveItem_Call) Run(run func(ctx context.Context, sess *state.Session, inFrame wire.SNACFrame, inBody wire.SNAC_0x10_0x04_BARTDownloadQuery)) *mockBARTService_RetrieveItem_Call {
  55. _c.Call.Run(func(args mock.Arguments) {
  56. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame), args[3].(wire.SNAC_0x10_0x04_BARTDownloadQuery))
  57. })
  58. return _c
  59. }
  60. func (_c *mockBARTService_RetrieveItem_Call) Return(_a0 wire.SNACMessage, _a1 error) *mockBARTService_RetrieveItem_Call {
  61. _c.Call.Return(_a0, _a1)
  62. return _c
  63. }
  64. func (_c *mockBARTService_RetrieveItem_Call) RunAndReturn(run func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x10_0x04_BARTDownloadQuery) (wire.SNACMessage, error)) *mockBARTService_RetrieveItem_Call {
  65. _c.Call.Return(run)
  66. return _c
  67. }
  68. // UpsertItem provides a mock function with given fields: ctx, sess, inFrame, inBody
  69. func (_m *mockBARTService) UpsertItem(ctx context.Context, sess *state.Session, inFrame wire.SNACFrame, inBody wire.SNAC_0x10_0x02_BARTUploadQuery) (wire.SNACMessage, error) {
  70. ret := _m.Called(ctx, sess, inFrame, inBody)
  71. if len(ret) == 0 {
  72. panic("no return value specified for UpsertItem")
  73. }
  74. var r0 wire.SNACMessage
  75. var r1 error
  76. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x10_0x02_BARTUploadQuery) (wire.SNACMessage, error)); ok {
  77. return rf(ctx, sess, inFrame, inBody)
  78. }
  79. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x10_0x02_BARTUploadQuery) wire.SNACMessage); ok {
  80. r0 = rf(ctx, sess, inFrame, inBody)
  81. } else {
  82. r0 = ret.Get(0).(wire.SNACMessage)
  83. }
  84. if rf, ok := ret.Get(1).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x10_0x02_BARTUploadQuery) error); ok {
  85. r1 = rf(ctx, sess, inFrame, inBody)
  86. } else {
  87. r1 = ret.Error(1)
  88. }
  89. return r0, r1
  90. }
  91. // mockBARTService_UpsertItem_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UpsertItem'
  92. type mockBARTService_UpsertItem_Call struct {
  93. *mock.Call
  94. }
  95. // UpsertItem is a helper method to define mock.On call
  96. // - ctx context.Context
  97. // - sess *state.Session
  98. // - inFrame wire.SNACFrame
  99. // - inBody wire.SNAC_0x10_0x02_BARTUploadQuery
  100. func (_e *mockBARTService_Expecter) UpsertItem(ctx interface{}, sess interface{}, inFrame interface{}, inBody interface{}) *mockBARTService_UpsertItem_Call {
  101. return &mockBARTService_UpsertItem_Call{Call: _e.mock.On("UpsertItem", ctx, sess, inFrame, inBody)}
  102. }
  103. func (_c *mockBARTService_UpsertItem_Call) Run(run func(ctx context.Context, sess *state.Session, inFrame wire.SNACFrame, inBody wire.SNAC_0x10_0x02_BARTUploadQuery)) *mockBARTService_UpsertItem_Call {
  104. _c.Call.Run(func(args mock.Arguments) {
  105. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame), args[3].(wire.SNAC_0x10_0x02_BARTUploadQuery))
  106. })
  107. return _c
  108. }
  109. func (_c *mockBARTService_UpsertItem_Call) Return(_a0 wire.SNACMessage, _a1 error) *mockBARTService_UpsertItem_Call {
  110. _c.Call.Return(_a0, _a1)
  111. return _c
  112. }
  113. func (_c *mockBARTService_UpsertItem_Call) RunAndReturn(run func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x10_0x02_BARTUploadQuery) (wire.SNACMessage, error)) *mockBARTService_UpsertItem_Call {
  114. _c.Call.Return(run)
  115. return _c
  116. }
  117. // newMockBARTService creates a new instance of mockBARTService. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  118. // The first argument is typically a *testing.T value.
  119. func newMockBARTService(t interface {
  120. mock.TestingT
  121. Cleanup(func())
  122. }) *mockBARTService {
  123. mock := &mockBARTService{}
  124. mock.Mock.Test(t)
  125. t.Cleanup(func() { mock.AssertExpectations(t) })
  126. return mock
  127. }