// Code generated by mockery v1.0.0. DO NOT EDIT.

package mocks

import mock "github.com/stretchr/testify/mock"

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

// BatchGet provides a mock function with given fields: keys
func (_m *KV) BatchGet(keys [][]byte) ([][]byte, error) {
	ret := _m.Called(keys)

	var r0 [][]byte
	if rf, ok := ret.Get(0).(func([][]byte) [][]byte); ok {
		r0 = rf(keys)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).([][]byte)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func([][]byte) error); ok {
		r1 = rf(keys)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Begin provides a mock function with given fields:
func (_m *KV) Begin() {
	_m.Called()
}

// Commit provides a mock function with given fields:
func (_m *KV) Commit() {
	_m.Called()
}

// Get provides a mock function with given fields: key
func (_m *KV) Get(key []byte) ([]byte, error) {
	ret := _m.Called(key)

	var r0 []byte
	if rf, ok := ret.Get(0).(func([]byte) []byte); ok {
		r0 = rf(key)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).([]byte)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func([]byte) error); ok {
		r1 = rf(key)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Rollback provides a mock function with given fields:
func (_m *KV) Rollback() {
	_m.Called()
}

// Set provides a mock function with given fields: key, value
func (_m *KV) Set(key []byte, value []byte) error {
	ret := _m.Called(key, value)

	var r0 error
	if rf, ok := ret.Get(0).(func([]byte, []byte) error); ok {
		r0 = rf(key, value)
	} else {
		r0 = ret.Error(0)
	}

	return r0
}