github.com/rohankumardubey/aresdb@v0.0.2-0.20190517170215-e54e3ca06b9c/client/mocks/Connector.go (about)

     1  //  Copyright (c) 2017-2018 Uber Technologies, Inc.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  
    15  // Code generated by mockery v1.0.0
    16  package mocks
    17  
    18  import (
    19  	mock "github.com/stretchr/testify/mock"
    20  	"github.com/uber/aresdb/client"
    21  	"github.com/uber/aresdb/memstore/common"
    22  )
    23  
    24  // Connector is an autogenerated mock type for the Connector type
    25  type Connector struct {
    26  	mock.Mock
    27  }
    28  
    29  // Insert provides a mock function with given fields: tableName, columnNames, rows
    30  func (_m *Connector) Insert(tableName string, columnNames []string, rows []client.Row, updateModes ...common.ColumnUpdateMode) (int, error) {
    31  	ret := _m.Called(tableName, columnNames, rows)
    32  
    33  	var r0 int
    34  	if rf, ok := ret.Get(0).(func(string, []string, []client.Row, ...common.ColumnUpdateMode) int); ok {
    35  		r0 = rf(tableName, columnNames, rows, updateModes...)
    36  	} else {
    37  		r0 = ret.Get(0).(int)
    38  	}
    39  
    40  	var r1 error
    41  	if rf, ok := ret.Get(1).(func(string, []string, []client.Row, ...common.ColumnUpdateMode) error); ok {
    42  		r1 = rf(tableName, columnNames, rows, updateModes...)
    43  	} else {
    44  		r1 = ret.Error(1)
    45  	}
    46  
    47  	return r0, r1
    48  }