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

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/influxdata/idpe/influxql/query (interfaces: ShardGroup)
     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  // MockShardGroup is a mock of ShardGroup interface
    17  type MockShardGroup struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockShardGroupMockRecorder
    20  }
    21  
    22  // MockShardGroupMockRecorder is the mock recorder for MockShardGroup
    23  type MockShardGroupMockRecorder struct {
    24  	mock *MockShardGroup
    25  }
    26  
    27  // NewMockShardGroup creates a new mock instance
    28  func NewMockShardGroup(ctrl *gomock.Controller) *MockShardGroup {
    29  	mock := &MockShardGroup{ctrl: ctrl}
    30  	mock.recorder = &MockShardGroupMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use
    35  func (m *MockShardGroup) EXPECT() *MockShardGroupMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // Close mocks base method
    40  func (m *MockShardGroup) Close() error {
    41  	m.ctrl.T.Helper()
    42  	ret := m.ctrl.Call(m, "Close")
    43  	ret0, _ := ret[0].(error)
    44  	return ret0
    45  }
    46  
    47  // Close indicates an expected call of Close
    48  func (mr *MockShardGroupMockRecorder) Close() *gomock.Call {
    49  	mr.mock.ctrl.T.Helper()
    50  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockShardGroup)(nil).Close))
    51  }
    52  
    53  // CreateIterator mocks base method
    54  func (m *MockShardGroup) CreateIterator(arg0 context.Context, arg1 *influxql.Measurement, arg2 query.IteratorOptions) (query.Iterator, error) {
    55  	m.ctrl.T.Helper()
    56  	ret := m.ctrl.Call(m, "CreateIterator", arg0, arg1, arg2)
    57  	ret0, _ := ret[0].(query.Iterator)
    58  	ret1, _ := ret[1].(error)
    59  	return ret0, ret1
    60  }
    61  
    62  // CreateIterator indicates an expected call of CreateIterator
    63  func (mr *MockShardGroupMockRecorder) CreateIterator(arg0, arg1, arg2 interface{}) *gomock.Call {
    64  	mr.mock.ctrl.T.Helper()
    65  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateIterator", reflect.TypeOf((*MockShardGroup)(nil).CreateIterator), arg0, arg1, arg2)
    66  }
    67  
    68  // FieldDimensions mocks base method
    69  func (m *MockShardGroup) FieldDimensions(arg0 context.Context, arg1 *influxql.Measurement) (map[string]influxql.DataType, map[string]struct{}, error) {
    70  	m.ctrl.T.Helper()
    71  	ret := m.ctrl.Call(m, "FieldDimensions", arg0, arg1)
    72  	ret0, _ := ret[0].(map[string]influxql.DataType)
    73  	ret1, _ := ret[1].(map[string]struct{})
    74  	ret2, _ := ret[2].(error)
    75  	return ret0, ret1, ret2
    76  }
    77  
    78  // FieldDimensions indicates an expected call of FieldDimensions
    79  func (mr *MockShardGroupMockRecorder) FieldDimensions(arg0, arg1 interface{}) *gomock.Call {
    80  	mr.mock.ctrl.T.Helper()
    81  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FieldDimensions", reflect.TypeOf((*MockShardGroup)(nil).FieldDimensions), arg0, arg1)
    82  }
    83  
    84  // IteratorCost mocks base method
    85  func (m *MockShardGroup) IteratorCost(arg0 context.Context, arg1 *influxql.Measurement, arg2 query.IteratorOptions) (query.IteratorCost, error) {
    86  	m.ctrl.T.Helper()
    87  	ret := m.ctrl.Call(m, "IteratorCost", arg0, arg1, arg2)
    88  	ret0, _ := ret[0].(query.IteratorCost)
    89  	ret1, _ := ret[1].(error)
    90  	return ret0, ret1
    91  }
    92  
    93  // IteratorCost indicates an expected call of IteratorCost
    94  func (mr *MockShardGroupMockRecorder) IteratorCost(arg0, arg1, arg2 interface{}) *gomock.Call {
    95  	mr.mock.ctrl.T.Helper()
    96  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IteratorCost", reflect.TypeOf((*MockShardGroup)(nil).IteratorCost), arg0, arg1, arg2)
    97  }
    98  
    99  // MapType mocks base method
   100  func (m *MockShardGroup) MapType(arg0 context.Context, arg1 *influxql.Measurement, arg2 string) influxql.DataType {
   101  	m.ctrl.T.Helper()
   102  	ret := m.ctrl.Call(m, "MapType", arg0, arg1, arg2)
   103  	ret0, _ := ret[0].(influxql.DataType)
   104  	return ret0
   105  }
   106  
   107  // MapType indicates an expected call of MapType
   108  func (mr *MockShardGroupMockRecorder) MapType(arg0, arg1, arg2 interface{}) *gomock.Call {
   109  	mr.mock.ctrl.T.Helper()
   110  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MapType", reflect.TypeOf((*MockShardGroup)(nil).MapType), arg0, arg1, arg2)
   111  }