mock_rate_limit_updater_test.go 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. // Code generated by mockery; DO NOT EDIT.
  2. // github.com/vektra/mockery
  3. // template: testify
  4. package oscar
  5. import (
  6. "context"
  7. "github.com/mk6i/open-oscar-server/state"
  8. mock "github.com/stretchr/testify/mock"
  9. )
  10. // newMockRateLimitUpdater creates a new instance of mockRateLimitUpdater. 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 newMockRateLimitUpdater(t interface {
  13. mock.TestingT
  14. Cleanup(func())
  15. }) *mockRateLimitUpdater {
  16. mock := &mockRateLimitUpdater{}
  17. mock.Mock.Test(t)
  18. t.Cleanup(func() { mock.AssertExpectations(t) })
  19. return mock
  20. }
  21. // mockRateLimitUpdater is an autogenerated mock type for the RateLimitUpdater type
  22. type mockRateLimitUpdater struct {
  23. mock.Mock
  24. }
  25. type mockRateLimitUpdater_Expecter struct {
  26. mock *mock.Mock
  27. }
  28. func (_m *mockRateLimitUpdater) EXPECT() *mockRateLimitUpdater_Expecter {
  29. return &mockRateLimitUpdater_Expecter{mock: &_m.Mock}
  30. }
  31. // MonitorRateLimits provides a mock function for the type mockRateLimitUpdater
  32. func (_mock *mockRateLimitUpdater) MonitorRateLimits(ctx context.Context, session *state.Session) {
  33. _mock.Called(ctx, session)
  34. return
  35. }
  36. // mockRateLimitUpdater_MonitorRateLimits_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'MonitorRateLimits'
  37. type mockRateLimitUpdater_MonitorRateLimits_Call struct {
  38. *mock.Call
  39. }
  40. // MonitorRateLimits is a helper method to define mock.On call
  41. // - ctx context.Context
  42. // - session *state.Session
  43. func (_e *mockRateLimitUpdater_Expecter) MonitorRateLimits(ctx interface{}, session interface{}) *mockRateLimitUpdater_MonitorRateLimits_Call {
  44. return &mockRateLimitUpdater_MonitorRateLimits_Call{Call: _e.mock.On("MonitorRateLimits", ctx, session)}
  45. }
  46. func (_c *mockRateLimitUpdater_MonitorRateLimits_Call) Run(run func(ctx context.Context, session *state.Session)) *mockRateLimitUpdater_MonitorRateLimits_Call {
  47. _c.Call.Run(func(args mock.Arguments) {
  48. var arg0 context.Context
  49. if args[0] != nil {
  50. arg0 = args[0].(context.Context)
  51. }
  52. var arg1 *state.Session
  53. if args[1] != nil {
  54. arg1 = args[1].(*state.Session)
  55. }
  56. run(
  57. arg0,
  58. arg1,
  59. )
  60. })
  61. return _c
  62. }
  63. func (_c *mockRateLimitUpdater_MonitorRateLimits_Call) Return() *mockRateLimitUpdater_MonitorRateLimits_Call {
  64. _c.Call.Return()
  65. return _c
  66. }
  67. func (_c *mockRateLimitUpdater_MonitorRateLimits_Call) RunAndReturn(run func(ctx context.Context, session *state.Session)) *mockRateLimitUpdater_MonitorRateLimits_Call {
  68. _c.Run(run)
  69. return _c
  70. }