erigon-pulse/gointerfaces/remote/mocks.go

// Code generated by moq; DO NOT EDIT.
// github.com/matryer/moq

package remote

import (
	context "context"
	types "github.com/ledgerwatch/erigon-lib/gointerfaces/types"
	grpc "google.golang.org/grpc"
	emptypb "google.golang.org/protobuf/types/known/emptypb"
	sync "sync"
)

// Ensure, that KVClientMock does implement KVClient.
// If this is not the case, regenerate this file with moq.
var _ KVClient = &KVClientMock{}

// KVClientMock is a mock implementation of KVClient.
//
// 	func TestSomethingThatUsesKVClient(t *testing.T) {
//
// 		// make and configure a mocked KVClient
// 		mockedKVClient := &KVClientMock{
// 			StateChangesFunc: func(ctx context.Context, in *StateChangeRequest, opts ...grpc.CallOption) (KV_StateChangesClient, error) {
// 				panic("mock out the StateChanges method")
// 			},
// 			TxFunc: func(ctx context.Context, opts ...grpc.CallOption) (KV_TxClient, error) {
// 				panic("mock out the Tx method")
// 			},
// 			VersionFunc: func(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*types.VersionReply, error) {
// 				panic("mock out the Version method")
// 			},
// 		}
//
// 		// use mockedKVClient in code that requires KVClient
// 		// and then make assertions.
//
// 	}
type KVClientMock struct {
	// StateChangesFunc mocks the StateChanges method.
	StateChangesFunc func(ctx context.Context, in *StateChangeRequest, opts ...grpc.CallOption) (KV_StateChangesClient, error)

	// TxFunc mocks the Tx method.
	TxFunc func(ctx context.Context, opts ...grpc.CallOption) (KV_TxClient, error)

	// VersionFunc mocks the Version method.
	VersionFunc func(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*types.VersionReply, error)

	// calls tracks calls to the methods.
	calls struct {
		// StateChanges holds details about calls to the StateChanges method.
		StateChanges []struct {
			// Ctx is the ctx argument value.
			Ctx context.Context
			// In is the in argument value.
			In *StateChangeRequest
			// Opts is the opts argument value.
			Opts []grpc.CallOption
		}
		// Tx holds details about calls to the Tx method.
		Tx []struct {
			// Ctx is the ctx argument value.
			Ctx context.Context
			// Opts is the opts argument value.
			Opts []grpc.CallOption
		}
		// Version holds details about calls to the Version method.
		Version []struct {
			// Ctx is the ctx argument value.
			Ctx context.Context
			// In is the in argument value.
			In *emptypb.Empty
			// Opts is the opts argument value.
			Opts []grpc.CallOption
		}
	}
	lockStateChanges sync.RWMutex
	lockTx           sync.RWMutex
	lockVersion      sync.RWMutex
}

// StateChanges calls StateChangesFunc.
func (mock *KVClientMock) StateChanges(ctx context.Context, in *StateChangeRequest, opts ...grpc.CallOption) (KV_StateChangesClient, error) {
	callInfo := struct {
		Ctx  context.Context
		In   *StateChangeRequest
		Opts []grpc.CallOption
	}{
		Ctx:  ctx,
		In:   in,
		Opts: opts,
	}
	mock.lockStateChanges.Lock()
	mock.calls.StateChanges = append(mock.calls.StateChanges, callInfo)
	mock.lockStateChanges.Unlock()
	if mock.StateChangesFunc == nil {
		var (
			kV_StateChangesClientOut KV_StateChangesClient
			errOut                   error
		)
		return kV_StateChangesClientOut, errOut
	}
	return mock.StateChangesFunc(ctx, in, opts...)
}

// StateChangesCalls gets all the calls that were made to StateChanges.
// Check the length with:
//     len(mockedKVClient.StateChangesCalls())
func (mock *KVClientMock) StateChangesCalls() []struct {
	Ctx  context.Context
	In   *StateChangeRequest
	Opts []grpc.CallOption
} {
	var calls []struct {
		Ctx  context.Context
		In   *StateChangeRequest
		Opts []grpc.CallOption
	}
	mock.lockStateChanges.RLock()
	calls = mock.calls.StateChanges
	mock.lockStateChanges.RUnlock()
	return calls
}

// Tx calls TxFunc.
func (mock *KVClientMock) Tx(ctx context.Context, opts ...grpc.CallOption) (KV_TxClient, error) {
	callInfo := struct {
		Ctx  context.Context
		Opts []grpc.CallOption
	}{
		Ctx:  ctx,
		Opts: opts,
	}
	mock.lockTx.Lock()
	mock.calls.Tx = append(mock.calls.Tx, callInfo)
	mock.lockTx.Unlock()
	if mock.TxFunc == nil {
		var (
			kV_TxClientOut KV_TxClient
			errOut         error
		)
		return kV_TxClientOut, errOut
	}
	return mock.TxFunc(ctx, opts...)
}

// TxCalls gets all the calls that were made to Tx.
// Check the length with:
//     len(mockedKVClient.TxCalls())
func (mock *KVClientMock) TxCalls() []struct {
	Ctx  context.Context
	Opts []grpc.CallOption
} {
	var calls []struct {
		Ctx  context.Context
		Opts []grpc.CallOption
	}
	mock.lockTx.RLock()
	calls = mock.calls.Tx
	mock.lockTx.RUnlock()
	return calls
}

// Version calls VersionFunc.
func (mock *KVClientMock) Version(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*types.VersionReply, error) {
	callInfo := struct {
		Ctx  context.Context
		In   *emptypb.Empty
		Opts []grpc.CallOption
	}{
		Ctx:  ctx,
		In:   in,
		Opts: opts,
	}
	mock.lockVersion.Lock()
	mock.calls.Version = append(mock.calls.Version, callInfo)
	mock.lockVersion.Unlock()
	if mock.VersionFunc == nil {
		var (
			versionReplyOut *types.VersionReply
			errOut          error
		)
		return versionReplyOut, errOut
	}
	return mock.VersionFunc(ctx, in, opts...)
}

// VersionCalls gets all the calls that were made to Version.
// Check the length with:
//     len(mockedKVClient.VersionCalls())
func (mock *KVClientMock) VersionCalls() []struct {
	Ctx  context.Context
	In   *emptypb.Empty
	Opts []grpc.CallOption
} {
	var calls []struct {
		Ctx  context.Context
		In   *emptypb.Empty
		Opts []grpc.CallOption
	}
	mock.lockVersion.RLock()
	calls = mock.calls.Version
	mock.lockVersion.RUnlock()
	return calls
}