// Code generated by mockery; DO NOT EDIT.
// github.com/vektra/mockery
// template: testify

package mocks

import (
	"github.com/aerospike/aerospike-client-go/v8"
	mock "github.com/stretchr/testify/mock"
)

// NewMockNodeGetter creates a new instance of MockNodeGetter. 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 NewMockNodeGetter(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockNodeGetter {
	mock := &MockNodeGetter{}
	mock.Mock.Test(t)

	t.Cleanup(func() { mock.AssertExpectations(t) })

	return mock
}

// MockNodeGetter is an autogenerated mock type for the NodeGetter type
type MockNodeGetter struct {
	mock.Mock
}

type MockNodeGetter_Expecter struct {
	mock *mock.Mock
}

func (_m *MockNodeGetter) EXPECT() *MockNodeGetter_Expecter {
	return &MockNodeGetter_Expecter{mock: &_m.Mock}
}

// GetNodeByName provides a mock function for the type MockNodeGetter
func (_mock *MockNodeGetter) GetNodeByName(name string) (*aerospike.Node, aerospike.Error) {
	ret := _mock.Called(name)

	if len(ret) == 0 {
		panic("no return value specified for GetNodeByName")
	}

	var r0 *aerospike.Node
	var r1 aerospike.Error
	if returnFunc, ok := ret.Get(0).(func(string) (*aerospike.Node, aerospike.Error)); ok {
		return returnFunc(name)
	}
	if returnFunc, ok := ret.Get(0).(func(string) *aerospike.Node); ok {
		r0 = returnFunc(name)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*aerospike.Node)
		}
	}
	if returnFunc, ok := ret.Get(1).(func(string) aerospike.Error); ok {
		r1 = returnFunc(name)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(aerospike.Error)
		}
	}
	return r0, r1
}

// MockNodeGetter_GetNodeByName_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetNodeByName'
type MockNodeGetter_GetNodeByName_Call struct {
	*mock.Call
}

// GetNodeByName is a helper method to define mock.On call
//   - name
func (_e *MockNodeGetter_Expecter) GetNodeByName(name interface{}) *MockNodeGetter_GetNodeByName_Call {
	return &MockNodeGetter_GetNodeByName_Call{Call: _e.mock.On("GetNodeByName", name)}
}

func (_c *MockNodeGetter_GetNodeByName_Call) Run(run func(name string)) *MockNodeGetter_GetNodeByName_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(string))
	})
	return _c
}

func (_c *MockNodeGetter_GetNodeByName_Call) Return(node *aerospike.Node, error aerospike.Error) *MockNodeGetter_GetNodeByName_Call {
	_c.Call.Return(node, error)
	return _c
}

func (_c *MockNodeGetter_GetNodeByName_Call) RunAndReturn(run func(name string) (*aerospike.Node, aerospike.Error)) *MockNodeGetter_GetNodeByName_Call {
	_c.Call.Return(run)
	return _c
}

// GetNodes provides a mock function for the type MockNodeGetter
func (_mock *MockNodeGetter) GetNodes() []*aerospike.Node {
	ret := _mock.Called()

	if len(ret) == 0 {
		panic("no return value specified for GetNodes")
	}

	var r0 []*aerospike.Node
	if returnFunc, ok := ret.Get(0).(func() []*aerospike.Node); ok {
		r0 = returnFunc()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).([]*aerospike.Node)
		}
	}
	return r0
}

// MockNodeGetter_GetNodes_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetNodes'
type MockNodeGetter_GetNodes_Call struct {
	*mock.Call
}

// GetNodes is a helper method to define mock.On call
func (_e *MockNodeGetter_Expecter) GetNodes() *MockNodeGetter_GetNodes_Call {
	return &MockNodeGetter_GetNodes_Call{Call: _e.mock.On("GetNodes")}
}

func (_c *MockNodeGetter_GetNodes_Call) Run(run func()) *MockNodeGetter_GetNodes_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *MockNodeGetter_GetNodes_Call) Return(nodes []*aerospike.Node) *MockNodeGetter_GetNodes_Call {
	_c.Call.Return(nodes)
	return _c
}

func (_c *MockNodeGetter_GetNodes_Call) RunAndReturn(run func() []*aerospike.Node) *MockNodeGetter_GetNodes_Call {
	_c.Call.Return(run)
	return _c
}

// GetRandomNode provides a mock function for the type MockNodeGetter
func (_mock *MockNodeGetter) GetRandomNode() (*aerospike.Node, aerospike.Error) {
	ret := _mock.Called()

	if len(ret) == 0 {
		panic("no return value specified for GetRandomNode")
	}

	var r0 *aerospike.Node
	var r1 aerospike.Error
	if returnFunc, ok := ret.Get(0).(func() (*aerospike.Node, aerospike.Error)); ok {
		return returnFunc()
	}
	if returnFunc, ok := ret.Get(0).(func() *aerospike.Node); ok {
		r0 = returnFunc()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*aerospike.Node)
		}
	}
	if returnFunc, ok := ret.Get(1).(func() aerospike.Error); ok {
		r1 = returnFunc()
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(aerospike.Error)
		}
	}
	return r0, r1
}

// MockNodeGetter_GetRandomNode_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetRandomNode'
type MockNodeGetter_GetRandomNode_Call struct {
	*mock.Call
}

// GetRandomNode is a helper method to define mock.On call
func (_e *MockNodeGetter_Expecter) GetRandomNode() *MockNodeGetter_GetRandomNode_Call {
	return &MockNodeGetter_GetRandomNode_Call{Call: _e.mock.On("GetRandomNode")}
}

func (_c *MockNodeGetter_GetRandomNode_Call) Run(run func()) *MockNodeGetter_GetRandomNode_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *MockNodeGetter_GetRandomNode_Call) Return(node *aerospike.Node, error aerospike.Error) *MockNodeGetter_GetRandomNode_Call {
	_c.Call.Return(node, error)
	return _c
}

func (_c *MockNodeGetter_GetRandomNode_Call) RunAndReturn(run func() (*aerospike.Node, aerospike.Error)) *MockNodeGetter_GetRandomNode_Call {
	_c.Call.Return(run)
	return _c
}
