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