// Code generated by mockery; DO NOT EDIT. // github.com/vektra/mockery // template: testify package mocks import ( "context" mock "github.com/stretchr/testify/mock" ) // 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. // The first argument is typically a *testing.T value. func NewMockHTTPProber(t interface { mock.TestingT Cleanup(func()) }) *MockHTTPProber { mock := &MockHTTPProber{} mock.Mock.Test(t) t.Cleanup(func() { mock.AssertExpectations(t) }) return mock } // MockHTTPProber is an autogenerated mock type for the HTTPProber type type MockHTTPProber struct { mock.Mock } type MockHTTPProber_Expecter struct { mock *mock.Mock } func (_m *MockHTTPProber) EXPECT() *MockHTTPProber_Expecter { return &MockHTTPProber_Expecter{mock: &_m.Mock} } // Probe provides a mock function for the type MockHTTPProber func (_mock *MockHTTPProber) Probe(ctx context.Context, url string) (int, int64, error) { ret := _mock.Called(ctx, url) if len(ret) == 0 { panic("no return value specified for Probe") } var r0 int var r1 int64 var r2 error if returnFunc, ok := ret.Get(0).(func(context.Context, string) (int, int64, error)); ok { return returnFunc(ctx, url) } if returnFunc, ok := ret.Get(0).(func(context.Context, string) int); ok { r0 = returnFunc(ctx, url) } else { r0 = ret.Get(0).(int) } if returnFunc, ok := ret.Get(1).(func(context.Context, string) int64); ok { r1 = returnFunc(ctx, url) } else { r1 = ret.Get(1).(int64) } if returnFunc, ok := ret.Get(2).(func(context.Context, string) error); ok { r2 = returnFunc(ctx, url) } else { r2 = ret.Error(2) } return r0, r1, r2 } // MockHTTPProber_Probe_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Probe' type MockHTTPProber_Probe_Call struct { *mock.Call } // Probe is a helper method to define mock.On call // - ctx context.Context // - url string func (_e *MockHTTPProber_Expecter) Probe(ctx any, url any) *MockHTTPProber_Probe_Call { return &MockHTTPProber_Probe_Call{Call: _e.mock.On("Probe", ctx, url)} } func (_c *MockHTTPProber_Probe_Call) Run(run func(ctx context.Context, url string)) *MockHTTPProber_Probe_Call { _c.Call.Run(func(args mock.Arguments) { var arg0 context.Context if args[0] != nil { arg0 = args[0].(context.Context) } var arg1 string if args[1] != nil { arg1 = args[1].(string) } run( arg0, arg1, ) }) return _c } func (_c *MockHTTPProber_Probe_Call) Return(n int, n1 int64, err error) *MockHTTPProber_Probe_Call { _c.Call.Return(n, n1, err) return _c } func (_c *MockHTTPProber_Probe_Call) RunAndReturn(run func(ctx context.Context, url string) (int, int64, error)) *MockHTTPProber_Probe_Call { _c.Call.Return(run) return _c }