// Code generated by mockery; DO NOT EDIT. // github.com/vektra/mockery // template: testify package atspi import ( "context" mock "github.com/stretchr/testify/mock" ) // newMockconnector creates a new instance of mockconnector. 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 newMockconnector(t interface { mock.TestingT Cleanup(func()) }) *mockconnector { if helper, ok := t.(interface{ Helper() }); ok { helper.Helper() } mock := &mockconnector{} mock.Mock.Test(t) t.Cleanup(func() { if helper, ok := t.(interface{ Helper() }); ok { helper.Helper() } mock.AssertExpectations(t) }) return mock } // mockconnector is an autogenerated mock type for the connector type type mockconnector struct { mock.Mock } type mockconnector_Expecter struct { mock *mock.Mock } func (_m *mockconnector) EXPECT() *mockconnector_Expecter { return &mockconnector_Expecter{mock: &_m.Mock} } // Connect provides a mock function for the type mockconnector func (_mock *mockconnector) Connect(ctx context.Context) (transport, error) { ret := _mock.Called(ctx) if len(ret) == 0 { panic("no return value specified for Connect") } var r0 transport var r1 error if returnFunc, ok := ret.Get(0).(func(context.Context) (transport, error)); ok { return returnFunc(ctx) } if returnFunc, ok := ret.Get(0).(func(context.Context) transport); ok { r0 = returnFunc(ctx) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(transport) } } if returnFunc, ok := ret.Get(1).(func(context.Context) error); ok { r1 = returnFunc(ctx) } else { r1 = ret.Error(1) } return r0, r1 } // mockconnector_Connect_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Connect' type mockconnector_Connect_Call struct { *mock.Call } // Connect is a helper method to define mock.On call // - ctx context.Context func (_e *mockconnector_Expecter) Connect(ctx any) *mockconnector_Connect_Call { return &mockconnector_Connect_Call{Call: _e.mock.On("Connect", ctx)} } func (_c *mockconnector_Connect_Call) Run(run func(ctx context.Context)) *mockconnector_Connect_Call { _c.Call.Run(func(args mock.Arguments) { var arg0 context.Context if args[0] != nil { arg0 = args[0].(context.Context) } run( arg0, ) }) return _c } func (_c *mockconnector_Connect_Call) Return(transportMoqParam transport, err error) *mockconnector_Connect_Call { _c.Call.Return(transportMoqParam, err) return _c } func (_c *mockconnector_Connect_Call) RunAndReturn(run func(ctx context.Context) (transport, error)) *mockconnector_Connect_Call { _c.Call.Return(run) return _c }