// Code generated by MockGen. DO NOT EDIT. // Source: github.com/influxdata/influxdb-client-go/api (interfaces: WriteAPIBlocking,QueryAPI) // Package mock_api is a generated GoMock package. package influx import ( context "context" reflect "reflect" gomock "github.com/golang/mock/gomock" api "github.com/influxdata/influxdb-client-go/v2/api" write "github.com/influxdata/influxdb-client-go/v2/api/write" domain "github.com/influxdata/influxdb-client-go/v2/domain" ) // MockWriteAPIBlocking is a mock of WriteAPIBlocking interface. type MockWriteAPIBlocking struct { ctrl *gomock.Controller recorder *MockWriteAPIBlockingMockRecorder } // MockWriteAPIBlockingMockRecorder is the mock recorder for MockWriteAPIBlocking. type MockWriteAPIBlockingMockRecorder struct { mock *MockWriteAPIBlocking } // NewMockWriteAPIBlocking creates a new mock instance. func NewMockWriteAPIBlocking(ctrl *gomock.Controller) *MockWriteAPIBlocking { mock := &MockWriteAPIBlocking{ctrl: ctrl} mock.recorder = &MockWriteAPIBlockingMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockWriteAPIBlocking) EXPECT() *MockWriteAPIBlockingMockRecorder { return m.recorder } // EnableBatching mocks base method. func (m *MockWriteAPIBlocking) EnableBatching() { m.ctrl.T.Helper() m.ctrl.Call(m, "EnableBatching") } // EnableBatching indicates an expected call of EnableBatching. func (mr *MockWriteAPIBlockingMockRecorder) EnableBatching() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnableBatching", reflect.TypeOf((*MockWriteAPIBlocking)(nil).EnableBatching)) } // Flush mocks base method. func (m *MockWriteAPIBlocking) Flush(arg0 context.Context) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Flush", arg0) ret0, _ := ret[0].(error) return ret0 } func (mr *MockWriteAPIBlockingMockRecorder) Flush(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Flush", reflect.TypeOf((*MockWriteAPIBlocking)(nil).Flush), arg0) } // WritePoint mocks base method. func (m *MockWriteAPIBlocking) WritePoint(arg0 context.Context, arg1 ...*write.Point) error { m.ctrl.T.Helper() varargs := []interface{}{arg0} for _, a := range arg1 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "WritePoint", varargs...) ret0, _ := ret[0].(error) return ret0 } // WritePoint indicates an expected call of WritePoint. func (mr *MockWriteAPIBlockingMockRecorder) WritePoint(arg0 interface{}, arg1 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0}, arg1...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WritePoint", reflect.TypeOf((*MockWriteAPIBlocking)(nil).WritePoint), varargs...) } // WriteRecord mocks base method. func (m *MockWriteAPIBlocking) WriteRecord(arg0 context.Context, arg1 ...string) error { m.ctrl.T.Helper() varargs := []interface{}{arg0} for _, a := range arg1 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "WriteRecord", varargs...) ret0, _ := ret[0].(error) return ret0 } // WriteRecord indicates an expected call of WriteRecord. func (mr *MockWriteAPIBlockingMockRecorder) WriteRecord(arg0 interface{}, arg1 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0}, arg1...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteRecord", reflect.TypeOf((*MockWriteAPIBlocking)(nil).WriteRecord), varargs...) } // MockQueryAPI is a mock of QueryAPI interface. type MockQueryAPI struct { ctrl *gomock.Controller recorder *MockQueryAPIMockRecorder } // MockQueryAPIMockRecorder is the mock recorder for MockQueryAPI. type MockQueryAPIMockRecorder struct { mock *MockQueryAPI } // NewMockQueryAPI creates a new mock instance. func NewMockQueryAPI(ctrl *gomock.Controller) *MockQueryAPI { mock := &MockQueryAPI{ctrl: ctrl} mock.recorder = &MockQueryAPIMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockQueryAPI) EXPECT() *MockQueryAPIMockRecorder { return m.recorder } // Query mocks base method. func (m *MockQueryAPI) Query(arg0 context.Context, arg1 string) (*api.QueryTableResult, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Query", arg0, arg1) ret0, _ := ret[0].(*api.QueryTableResult) ret1, _ := ret[1].(error) return ret0, ret1 } // Query indicates an expected call of Query. func (mr *MockQueryAPIMockRecorder) Query(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Query", reflect.TypeOf((*MockQueryAPI)(nil).Query), arg0, arg1) } // QueryRaw mocks base method. func (m *MockQueryAPI) QueryRaw(arg0 context.Context, arg1 string, arg2 *domain.Dialect) (string, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "QueryRaw", arg0, arg1, arg2) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // QueryRaw indicates an expected call of QueryRaw. func (mr *MockQueryAPIMockRecorder) QueryRaw(arg0, arg1, arg2 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryRaw", reflect.TypeOf((*MockQueryAPI)(nil).QueryRaw), arg0, arg1, arg2) } // QueryRawWithParams executes flux parametrized query on the InfluxDB server and returns complete query result as a string with table annotations according to dialect func (m *MockQueryAPI) QueryRawWithParams(ctx context.Context, query string, dialect *domain.Dialect, params interface{}) (string, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "QueryRaw", ctx, query, dialect) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // QueryRawWithParams indicates an expected call of QueryRawWithParams. func (mr *MockQueryAPIMockRecorder) QueryRawWithParams(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryRawWithParams", reflect.TypeOf((*MockQueryAPI)(nil).QueryRaw), arg0, arg1, arg2, arg3) } // QueryWithParams executes flux parametrized query on the InfluxDB server and returns QueryTableResult which parses streamed response into structures representing flux table parts func (m *MockQueryAPI) QueryWithParams(ctx context.Context, query string, params interface{}) (*api.QueryTableResult, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "QueryWithParams", ctx, query) ret0, _ := ret[0].(*api.QueryTableResult) ret1, _ := ret[1].(error) return ret0, ret1 } // QueryWithParms indicates an expected call of QueryWithParams. func (mr *MockQueryAPIMockRecorder) QueryWithParams(arg0, arg1, arg2 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryWithParams", reflect.TypeOf((*MockQueryAPI)(nil).QueryWithParams), arg0, arg1, arg2) }