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 mock := &MockHTTPProber{}
20 mock.Mock.Test(t)
21
22 t.Cleanup(func() { mock.AssertExpectations(t) })
23
24 return mock
25}
26
27// MockHTTPProber is an autogenerated mock type for the HTTPProber type
28type MockHTTPProber struct {
29 mock.Mock
30}
31
32type MockHTTPProber_Expecter struct {
33 mock *mock.Mock
34}
35
36func (_m *MockHTTPProber) EXPECT() *MockHTTPProber_Expecter {
37 return &MockHTTPProber_Expecter{mock: &_m.Mock}
38}
39
40// Probe provides a mock function for the type MockHTTPProber
41func (_mock *MockHTTPProber) Probe(ctx context.Context, url string) (int, int64, error) {
42 ret := _mock.Called(ctx, url)
43
44 if len(ret) == 0 {
45 panic("no return value specified for Probe")
46 }
47
48 var r0 int
49 var r1 int64
50 var r2 error
51 if returnFunc, ok := ret.Get(0).(func(context.Context, string) (int, int64, error)); ok {
52 return returnFunc(ctx, url)
53 }
54 if returnFunc, ok := ret.Get(0).(func(context.Context, string) int); ok {
55 r0 = returnFunc(ctx, url)
56 } else {
57 r0 = ret.Get(0).(int)
58 }
59 if returnFunc, ok := ret.Get(1).(func(context.Context, string) int64); ok {
60 r1 = returnFunc(ctx, url)
61 } else {
62 r1 = ret.Get(1).(int64)
63 }
64 if returnFunc, ok := ret.Get(2).(func(context.Context, string) error); ok {
65 r2 = returnFunc(ctx, url)
66 } else {
67 r2 = ret.Error(2)
68 }
69 return r0, r1, r2
70}
71
72// MockHTTPProber_Probe_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Probe'
73type MockHTTPProber_Probe_Call struct {
74 *mock.Call
75}
76
77// Probe is a helper method to define mock.On call
78// - ctx context.Context
79// - url string
80func (_e *MockHTTPProber_Expecter) Probe(ctx interface{}, url interface{}) *MockHTTPProber_Probe_Call {
81 return &MockHTTPProber_Probe_Call{Call: _e.mock.On("Probe", ctx, url)}
82}
83
84func (_c *MockHTTPProber_Probe_Call) Run(run func(ctx context.Context, url string)) *MockHTTPProber_Probe_Call {
85 _c.Call.Run(func(args mock.Arguments) {
86 var arg0 context.Context
87 if args[0] != nil {
88 arg0 = args[0].(context.Context)
89 }
90 var arg1 string
91 if args[1] != nil {
92 arg1 = args[1].(string)
93 }
94 run(
95 arg0,
96 arg1,
97 )
98 })
99 return _c
100}
101
102func (_c *MockHTTPProber_Probe_Call) Return(n int, n1 int64, err error) *MockHTTPProber_Probe_Call {
103 _c.Call.Return(n, n1, err)
104 return _c
105}
106
107func (_c *MockHTTPProber_Probe_Call) RunAndReturn(run func(ctx context.Context, url string) (int, int64, error)) *MockHTTPProber_Probe_Call {
108 _c.Call.Return(run)
109 return _c
110}