github.com/influxdata/influxdb/v2@v2.7.6/influxql/query/mocks/ShardMapper.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/influxdata/idpe/influxql/query (interfaces: ShardMapper)
     3  
     4  // Package mocks is a generated GoMock package.
     5  package mocks
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	query "github.com/influxdata/influxdb/v2/influxql/query"
    13  	influxql "github.com/influxdata/influxql"
    14  )
    15  
    16  // MockShardMapper is a mock of ShardMapper interface
    17  type MockShardMapper struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockShardMapperMockRecorder
    20  }
    21  
    22  // MockShardMapperMockRecorder is the mock recorder for MockShardMapper
    23  type MockShardMapperMockRecorder struct {
    24  	mock *MockShardMapper
    25  }
    26  
    27  // NewMockShardMapper creates a new mock instance
    28  func NewMockShardMapper(ctrl *gomock.Controller) *MockShardMapper {
    29  	mock := &MockShardMapper{ctrl: ctrl}
    30  	mock.recorder = &MockShardMapperMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use
    35  func (m *MockShardMapper) EXPECT() *MockShardMapperMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // MapShards mocks base method
    40  func (m *MockShardMapper) MapShards(arg0 context.Context, arg1 influxql.Sources, arg2 influxql.TimeRange, arg3 query.SelectOptions) (query.ShardGroup, error) {
    41  	m.ctrl.T.Helper()
    42  	ret := m.ctrl.Call(m, "MapShards", arg0, arg1, arg2, arg3)
    43  	ret0, _ := ret[0].(query.ShardGroup)
    44  	ret1, _ := ret[1].(error)
    45  	return ret0, ret1
    46  }
    47  
    48  // MapShards indicates an expected call of MapShards
    49  func (mr *MockShardMapperMockRecorder) MapShards(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
    50  	mr.mock.ctrl.T.Helper()
    51  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MapShards", reflect.TypeOf((*MockShardMapper)(nil).MapShards), arg0, arg1, arg2, arg3)
    52  }