mock_server.go

// Code generated by MockGen. DO NOT EDIT.
// Source: ./router/server/server.go

// Package mock is a generated GoMock package.
package mock

import (
	reflect "reflect"

	gomock "github.com/golang/mock/gomock"
	pgproto3 "github.com/jackc/pgx/v5/pgproto3"
	config "github.com/pg-sharding/spqr/pkg/config"
	kr "github.com/pg-sharding/spqr/pkg/models/kr"
	prepstatement "github.com/pg-sharding/spqr/pkg/prepstatement"
	shard "github.com/pg-sharding/spqr/pkg/shard"
	tsa "github.com/pg-sharding/spqr/pkg/tsa"
	txstatus "github.com/pg-sharding/spqr/pkg/txstatus"
)

// MockServer is a mock of Server interface.
type MockServer struct {
	ctrl     *gomock.Controller
	recorder *MockServerMockRecorder
}

// MockServerMockRecorder is the mock recorder for MockServer.
type MockServerMockRecorder struct {
	mock *MockServer
}

// NewMockServer creates a new mock instance.
func NewMockServer(ctrl *gomock.Controller) *MockServer {
	mock := &MockServer{ctrl: ctrl}
	mock.recorder = &MockServerMockRecorder{mock}
	return mock
}

// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockServer) EXPECT() *MockServerMockRecorder {
	return m.recorder
}

// AddDataShard mocks base method.
func (m *MockServer) AddDataShard(clid uint, shardKey kr.ShardKey, tsa tsa.TSA) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "AddDataShard", clid, shardKey, tsa)
	ret0, _ := ret[0].(error)
	return ret0
}

// AddDataShard indicates an expected call of AddDataShard.
func (mr *MockServerMockRecorder) AddDataShard(clid, shardKey, tsa interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddDataShard", reflect.TypeOf((*MockServer)(nil).AddDataShard), clid, shardKey, tsa)
}

// Cancel mocks base method.
func (m *MockServer) Cancel() error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Cancel")
	ret0, _ := ret[0].(error)
	return ret0
}

// Cancel indicates an expected call of Cancel.
func (mr *MockServerMockRecorder) Cancel() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Cancel", reflect.TypeOf((*MockServer)(nil).Cancel))
}

// DataPending mocks base method.
func (m *MockServer) DataPending() bool {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DataPending")
	ret0, _ := ret[0].(bool)
	return ret0
}

// DataPending indicates an expected call of DataPending.
func (mr *MockServerMockRecorder) DataPending() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DataPending", reflect.TypeOf((*MockServer)(nil).DataPending))
}

// Datashards mocks base method.
func (m *MockServer) Datashards() []shard.Shard {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Datashards")
	ret0, _ := ret[0].([]shard.Shard)
	return ret0
}

// Datashards indicates an expected call of Datashards.
func (mr *MockServerMockRecorder) Datashards() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Datashards", reflect.TypeOf((*MockServer)(nil).Datashards))
}

// HasPrepareStatement mocks base method.
func (m *MockServer) HasPrepareStatement(hash uint64, shardId uint) (bool, *prepstatement.PreparedStatementDescriptor) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "HasPrepareStatement", hash, shardId)
	ret0, _ := ret[0].(bool)
	ret1, _ := ret[1].(*prepstatement.PreparedStatementDescriptor)
	return ret0, ret1
}

// HasPrepareStatement indicates an expected call of HasPrepareStatement.
func (mr *MockServerMockRecorder) HasPrepareStatement(hash, shardId interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HasPrepareStatement", reflect.TypeOf((*MockServer)(nil).HasPrepareStatement), hash, shardId)
}

// Name mocks base method.
func (m *MockServer) Name() string {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Name")
	ret0, _ := ret[0].(string)
	return ret0
}

// Name indicates an expected call of Name.
func (mr *MockServerMockRecorder) Name() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Name", reflect.TypeOf((*MockServer)(nil).Name))
}

// Receive mocks base method.
func (m *MockServer) Receive() (pgproto3.BackendMessage, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Receive")
	ret0, _ := ret[0].(pgproto3.BackendMessage)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// Receive indicates an expected call of Receive.
func (mr *MockServerMockRecorder) Receive() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Receive", reflect.TypeOf((*MockServer)(nil).Receive))
}

// ReceiveShard mocks base method.
func (m *MockServer) ReceiveShard(shardId uint) (pgproto3.BackendMessage, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ReceiveShard", shardId)
	ret0, _ := ret[0].(pgproto3.BackendMessage)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ReceiveShard indicates an expected call of ReceiveShard.
func (mr *MockServerMockRecorder) ReceiveShard(shardId interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReceiveShard", reflect.TypeOf((*MockServer)(nil).ReceiveShard), shardId)
}

// Reset mocks base method.
func (m *MockServer) Reset() error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Reset")
	ret0, _ := ret[0].(error)
	return ret0
}

// Reset indicates an expected call of Reset.
func (mr *MockServerMockRecorder) Reset() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Reset", reflect.TypeOf((*MockServer)(nil).Reset))
}

// Send mocks base method.
func (m *MockServer) Send(query pgproto3.FrontendMessage) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Send", query)
	ret0, _ := ret[0].(error)
	return ret0
}

// Send indicates an expected call of Send.
func (mr *MockServerMockRecorder) Send(query interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockServer)(nil).Send), query)
}

// SendShard mocks base method.
func (m *MockServer) SendShard(query pgproto3.FrontendMessage, shardId uint) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "SendShard", query, shardId)
	ret0, _ := ret[0].(error)
	return ret0
}

// SendShard indicates an expected call of SendShard.
func (mr *MockServerMockRecorder) SendShard(query, shardId interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendShard", reflect.TypeOf((*MockServer)(nil).SendShard), query, shardId)
}

// SetTxStatus mocks base method.
func (m *MockServer) SetTxStatus(status txstatus.TXStatus) {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "SetTxStatus", status)
}

// SetTxStatus indicates an expected call of SetTxStatus.
func (mr *MockServerMockRecorder) SetTxStatus(status interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTxStatus", reflect.TypeOf((*MockServer)(nil).SetTxStatus), status)
}

// StorePrepareStatement mocks base method.
func (m *MockServer) StorePrepareStatement(hash uint64, shardId uint, d *prepstatement.PreparedStatementDefinition, rd *prepstatement.PreparedStatementDescriptor) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "StorePrepareStatement", hash, shardId, d, rd)
	ret0, _ := ret[0].(error)
	return ret0
}

// StorePrepareStatement indicates an expected call of StorePrepareStatement.
func (mr *MockServerMockRecorder) StorePrepareStatement(hash, shardId, d, rd interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StorePrepareStatement", reflect.TypeOf((*MockServer)(nil).StorePrepareStatement), hash, shardId, d, rd)
}

// Sync mocks base method.
func (m *MockServer) Sync() int64 {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Sync")
	ret0, _ := ret[0].(int64)
	return ret0
}

// Sync indicates an expected call of Sync.
func (mr *MockServerMockRecorder) Sync() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Sync", reflect.TypeOf((*MockServer)(nil).Sync))
}

// TxStatus mocks base method.
func (m *MockServer) TxStatus() txstatus.TXStatus {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "TxStatus")
	ret0, _ := ret[0].(txstatus.TXStatus)
	return ret0
}

// TxStatus indicates an expected call of TxStatus.
func (mr *MockServerMockRecorder) TxStatus() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TxStatus", reflect.TypeOf((*MockServer)(nil).TxStatus))
}

// UnRouteShard mocks base method.
func (m *MockServer) UnRouteShard(sh kr.ShardKey, rule *config.FrontendRule) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "UnRouteShard", sh, rule)
	ret0, _ := ret[0].(error)
	return ret0
}

// UnRouteShard indicates an expected call of UnRouteShard.
func (mr *MockServerMockRecorder) UnRouteShard(sh, rule interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnRouteShard", reflect.TypeOf((*MockServer)(nil).UnRouteShard), sh, rule)
}