mirror of https://github.com/milvus-io/milvus.git
generate-mockery for querycoordv2, querynode and rootcoort (#21714)
Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>pull/21737/head
congqixia
GitHub
parent
837e3162d7
commit
5f3d3dc4fc
14
Makefile
14
Makefile
|
|
@ -39,6 +39,7 @@ get-build-deps:
|
|||
getdeps:
|
||||
@mkdir -p ${GOPATH}/bin
|
||||
@which golangci-lint 1>/dev/null || (echo "Installing golangci-lint" && curl -sSfL https://raw.githubusercontent.com/golangci/golangci-lint/master/install.sh | sh -s -- -b $(GOPATH)/bin v1.46.2)
|
||||
@$(PWD)/bin/mockery --version 2>&1 1>/dev/null || (echo "Installing mockery v2.16.0 to ./bin/" && GOBIN=$(PWD)/bin/ go install github.com/vektra/mockery/v2@v2.16.0)
|
||||
|
||||
tools/bin/revive: tools/check/go.mod
|
||||
cd tools/check; \
|
||||
|
|
@ -303,5 +304,18 @@ mock-datacoord:
|
|||
mock-tnx-kv:
|
||||
mockery --name=TxnKV --dir=$(PWD)/internal/kv --output=$(PWD)/internal/kv/mocks --filename=TxnKV.go --with-expecter
|
||||
|
||||
generate-mockery: getdeps
|
||||
# internal/querycoordv2
|
||||
$(PWD)/bin/mockery --name=QueryNodeServer --dir=$(PWD)/internal/proto/querypb/ --output=$(PWD)/internal/querycoordv2/mocks --filename=mock_querynode.go --with-expecter --structname=MockQueryNodeServer
|
||||
$(PWD)/bin/mockery --name=Broker --dir=$(PWD)/internal/querycoordv2/meta --output=$(PWD)/internal/querycoordv2/meta --filename=mock_broker.go --with-expecter --structname=MockBroker --outpkg=meta
|
||||
$(PWD)/bin/mockery --name=Scheduler --dir=$(PWD)/internal/querycoordv2/task --output=$(PWD)/internal/querycoordv2/task --filename=mock_scheduler.go --with-expecter --structname=MockScheduler --outpkg=task --inpackage
|
||||
$(PWD)/bin/mockery --name=Cluster --dir=$(PWD)/internal/querycoordv2/session --output=$(PWD)/internal/querycoordv2/session --filename=mock_cluster.go --with-expecter --structname=MockCluster --outpkg=session --inpackage
|
||||
$(PWD)/bin/mockery --name=Store --dir=$(PWD)/internal/querycoordv2/meta --output=$(PWD)/internal/querycoordv2/meta --filename=mock_store.go --with-expecter --structname=MockStore --outpkg=meta --inpackage
|
||||
$(PWD)/bin/mockery --name=Balance --dir=$(PWD)/internal/querycoordv2/balance --output=$(PWD)/internal/querycoordv2/balance --filename=mock_balancer.go --with-expecter --structname=MockBalancer --outpkg=balance --inpackage
|
||||
# internal/querynode
|
||||
$(PWD)/bin/mockery --name=TSafeReplicaInterface --dir=$(PWD)/internal/querynode --output=$(PWD)/internal/querynode --filename=mock_tsafe_replica_test.go --with-expecter --structname=MockTSafeReplicaInterface --outpkg=querynode --inpackage
|
||||
# internal/rootcoord
|
||||
$(PWD)/bin/mockery --name=IMetaTable --dir=$(PWD)/internal/rootcoord --output=$(PWD)/internal/rootcoord/mocks --filename=meta_table.go --with-expecter --outpkg=mockrootcoord
|
||||
$(PWD)/bin/mockery --name=GarbageCollector --dir=$(PWD)/internal/rootcoord --output=$(PWD)/internal/rootcoord/mocks --filename=garbage_collector.go --with-expecter --outpkg=mockrootcoord
|
||||
|
||||
ci-ut: build-cpp-with-coverage generated-proto-go-without-cpp codecov-cpp codecov-go
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
// Code generated by mockery v2.14.0. DO NOT EDIT.
|
||||
// Code generated by mockery v2.16.0. DO NOT EDIT.
|
||||
|
||||
package balance
|
||||
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
// Code generated by mockery v2.14.0. DO NOT EDIT.
|
||||
// Code generated by mockery v2.16.0. DO NOT EDIT.
|
||||
|
||||
package meta
|
||||
|
||||
|
|
@ -6,6 +6,7 @@ import (
|
|||
context "context"
|
||||
|
||||
datapb "github.com/milvus-io/milvus/internal/proto/datapb"
|
||||
|
||||
mock "github.com/stretchr/testify/mock"
|
||||
|
||||
querypb "github.com/milvus-io/milvus/internal/proto/querypb"
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
// Code generated by mockery v2.14.0. DO NOT EDIT.
|
||||
// Code generated by mockery v2.16.0. DO NOT EDIT.
|
||||
|
||||
package meta
|
||||
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
// Code generated by mockery v2.14.0. DO NOT EDIT.
|
||||
// Code generated by mockery v2.16.0. DO NOT EDIT.
|
||||
|
||||
package mocks
|
||||
|
||||
|
|
@ -59,7 +59,7 @@ type MockQueryNodeServer_GetComponentStates_Call struct {
|
|||
|
||||
// GetComponentStates is a helper method to define mock.On call
|
||||
// - _a0 context.Context
|
||||
// - _a1 *internalpb.GetComponentStatesRequest
|
||||
// - _a1 *milvuspb.GetComponentStatesRequest
|
||||
func (_e *MockQueryNodeServer_Expecter) GetComponentStates(_a0 interface{}, _a1 interface{}) *MockQueryNodeServer_GetComponentStates_Call {
|
||||
return &MockQueryNodeServer_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", _a0, _a1)}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
// Code generated by mockery v2.14.0. DO NOT EDIT.
|
||||
// Code generated by mockery v2.16.0. DO NOT EDIT.
|
||||
|
||||
package session
|
||||
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
// Code generated by mockery v2.14.1. DO NOT EDIT.
|
||||
// Code generated by mockery v2.16.0. DO NOT EDIT.
|
||||
|
||||
package task
|
||||
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
// Code generated by mockery v2.14.0. DO NOT EDIT.
|
||||
// Code generated by mockery v2.16.0. DO NOT EDIT.
|
||||
|
||||
package querynode
|
||||
|
||||
|
|
|
|||
|
|
@ -1,87 +0,0 @@
|
|||
// Code generated by mockery v2.14.0. DO NOT EDIT.
|
||||
|
||||
package mockrootcoord
|
||||
|
||||
import (
|
||||
context "context"
|
||||
|
||||
model "github.com/milvus-io/milvus/internal/metastore/model"
|
||||
mock "github.com/stretchr/testify/mock"
|
||||
)
|
||||
|
||||
// GarbageCollector is an autogenerated mock type for the GarbageCollector type
|
||||
type GarbageCollector struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
// GcCollectionData provides a mock function with given fields: ctx, coll
|
||||
func (_m *GarbageCollector) GcCollectionData(ctx context.Context, coll *model.Collection) (uint64, error) {
|
||||
ret := _m.Called(ctx, coll)
|
||||
|
||||
var r0 uint64
|
||||
if rf, ok := ret.Get(0).(func(context.Context, *model.Collection) uint64); ok {
|
||||
r0 = rf(ctx, coll)
|
||||
} else {
|
||||
r0 = ret.Get(0).(uint64)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(context.Context, *model.Collection) error); ok {
|
||||
r1 = rf(ctx, coll)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GcPartitionData provides a mock function with given fields: ctx, pChannels, partition
|
||||
func (_m *GarbageCollector) GcPartitionData(ctx context.Context, pChannels []string, partition *model.Partition) (uint64, error) {
|
||||
ret := _m.Called(ctx, pChannels, partition)
|
||||
|
||||
var r0 uint64
|
||||
if rf, ok := ret.Get(0).(func(context.Context, []string, *model.Partition) uint64); ok {
|
||||
r0 = rf(ctx, pChannels, partition)
|
||||
} else {
|
||||
r0 = ret.Get(0).(uint64)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(context.Context, []string, *model.Partition) error); ok {
|
||||
r1 = rf(ctx, pChannels, partition)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// ReDropCollection provides a mock function with given fields: collMeta, ts
|
||||
func (_m *GarbageCollector) ReDropCollection(collMeta *model.Collection, ts uint64) {
|
||||
_m.Called(collMeta, ts)
|
||||
}
|
||||
|
||||
// ReDropPartition provides a mock function with given fields: pChannels, partition, ts
|
||||
func (_m *GarbageCollector) ReDropPartition(pChannels []string, partition *model.Partition, ts uint64) {
|
||||
_m.Called(pChannels, partition, ts)
|
||||
}
|
||||
|
||||
// RemoveCreatingCollection provides a mock function with given fields: collMeta
|
||||
func (_m *GarbageCollector) RemoveCreatingCollection(collMeta *model.Collection) {
|
||||
_m.Called(collMeta)
|
||||
}
|
||||
|
||||
type mockConstructorTestingTNewGarbageCollector interface {
|
||||
mock.TestingT
|
||||
Cleanup(func())
|
||||
}
|
||||
|
||||
// NewGarbageCollector creates a new instance of GarbageCollector. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
|
||||
func NewGarbageCollector(t mockConstructorTestingTNewGarbageCollector) *GarbageCollector {
|
||||
mock := &GarbageCollector{}
|
||||
mock.Mock.Test(t)
|
||||
|
||||
t.Cleanup(func() { mock.AssertExpectations(t) })
|
||||
|
||||
return mock
|
||||
}
|
||||
|
|
@ -1,645 +0,0 @@
|
|||
// Code generated by mockery v2.14.0. DO NOT EDIT.
|
||||
|
||||
package mockrootcoord
|
||||
|
||||
import (
|
||||
context "context"
|
||||
|
||||
etcdpb "github.com/milvus-io/milvus/internal/proto/etcdpb"
|
||||
internalpb "github.com/milvus-io/milvus/internal/proto/internalpb"
|
||||
|
||||
milvuspb "github.com/milvus-io/milvus-proto/go-api/milvuspb"
|
||||
|
||||
mock "github.com/stretchr/testify/mock"
|
||||
|
||||
model "github.com/milvus-io/milvus/internal/metastore/model"
|
||||
)
|
||||
|
||||
// IMetaTable is an autogenerated mock type for the IMetaTable type
|
||||
type IMetaTable struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
// AddCollection provides a mock function with given fields: ctx, coll
|
||||
func (_m *IMetaTable) AddCollection(ctx context.Context, coll *model.Collection) error {
|
||||
ret := _m.Called(ctx, coll)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, *model.Collection) error); ok {
|
||||
r0 = rf(ctx, coll)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// AddCredential provides a mock function with given fields: credInfo
|
||||
func (_m *IMetaTable) AddCredential(credInfo *internalpb.CredentialInfo) error {
|
||||
ret := _m.Called(credInfo)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(*internalpb.CredentialInfo) error); ok {
|
||||
r0 = rf(credInfo)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// AddPartition provides a mock function with given fields: ctx, partition
|
||||
func (_m *IMetaTable) AddPartition(ctx context.Context, partition *model.Partition) error {
|
||||
ret := _m.Called(ctx, partition)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, *model.Partition) error); ok {
|
||||
r0 = rf(ctx, partition)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// AlterAlias provides a mock function with given fields: ctx, alias, collectionName, ts
|
||||
func (_m *IMetaTable) AlterAlias(ctx context.Context, alias string, collectionName string, ts uint64) error {
|
||||
ret := _m.Called(ctx, alias, collectionName, ts)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) error); ok {
|
||||
r0 = rf(ctx, alias, collectionName, ts)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// AlterCollection provides a mock function with given fields: ctx, oldColl, newColl, ts
|
||||
func (_m *IMetaTable) AlterCollection(ctx context.Context, oldColl *model.Collection, newColl *model.Collection, ts uint64) error {
|
||||
ret := _m.Called(ctx, oldColl, newColl, ts)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, *model.Collection, *model.Collection, uint64) error); ok {
|
||||
r0 = rf(ctx, oldColl, newColl, ts)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// AlterCredential provides a mock function with given fields: credInfo
|
||||
func (_m *IMetaTable) AlterCredential(credInfo *internalpb.CredentialInfo) error {
|
||||
ret := _m.Called(credInfo)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(*internalpb.CredentialInfo) error); ok {
|
||||
r0 = rf(credInfo)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// ChangeCollectionState provides a mock function with given fields: ctx, collectionID, state, ts
|
||||
func (_m *IMetaTable) ChangeCollectionState(ctx context.Context, collectionID int64, state etcdpb.CollectionState, ts uint64) error {
|
||||
ret := _m.Called(ctx, collectionID, state, ts)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int64, etcdpb.CollectionState, uint64) error); ok {
|
||||
r0 = rf(ctx, collectionID, state, ts)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// ChangePartitionState provides a mock function with given fields: ctx, collectionID, partitionID, state, ts
|
||||
func (_m *IMetaTable) ChangePartitionState(ctx context.Context, collectionID int64, partitionID int64, state etcdpb.PartitionState, ts uint64) error {
|
||||
ret := _m.Called(ctx, collectionID, partitionID, state, ts)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int64, int64, etcdpb.PartitionState, uint64) error); ok {
|
||||
r0 = rf(ctx, collectionID, partitionID, state, ts)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// CreateAlias provides a mock function with given fields: ctx, alias, collectionName, ts
|
||||
func (_m *IMetaTable) CreateAlias(ctx context.Context, alias string, collectionName string, ts uint64) error {
|
||||
ret := _m.Called(ctx, alias, collectionName, ts)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) error); ok {
|
||||
r0 = rf(ctx, alias, collectionName, ts)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// CreateRole provides a mock function with given fields: tenant, entity
|
||||
func (_m *IMetaTable) CreateRole(tenant string, entity *milvuspb.RoleEntity) error {
|
||||
ret := _m.Called(tenant, entity)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity) error); ok {
|
||||
r0 = rf(tenant, entity)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// DeleteCredential provides a mock function with given fields: username
|
||||
func (_m *IMetaTable) DeleteCredential(username string) error {
|
||||
ret := _m.Called(username)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(string) error); ok {
|
||||
r0 = rf(username)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// DropAlias provides a mock function with given fields: ctx, alias, ts
|
||||
func (_m *IMetaTable) DropAlias(ctx context.Context, alias string, ts uint64) error {
|
||||
ret := _m.Called(ctx, alias, ts)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, string, uint64) error); ok {
|
||||
r0 = rf(ctx, alias, ts)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// DropGrant provides a mock function with given fields: tenant, role
|
||||
func (_m *IMetaTable) DropGrant(tenant string, role *milvuspb.RoleEntity) error {
|
||||
ret := _m.Called(tenant, role)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity) error); ok {
|
||||
r0 = rf(tenant, role)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// DropRole provides a mock function with given fields: tenant, roleName
|
||||
func (_m *IMetaTable) DropRole(tenant string, roleName string) error {
|
||||
ret := _m.Called(tenant, roleName)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(string, string) error); ok {
|
||||
r0 = rf(tenant, roleName)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// GetCollectionByID provides a mock function with given fields: ctx, collectionID, ts, allowUnavailable
|
||||
func (_m *IMetaTable) GetCollectionByID(ctx context.Context, collectionID int64, ts uint64, allowUnavailable bool) (*model.Collection, error) {
|
||||
ret := _m.Called(ctx, collectionID, ts, allowUnavailable)
|
||||
|
||||
var r0 *model.Collection
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int64, uint64, bool) *model.Collection); ok {
|
||||
r0 = rf(ctx, collectionID, ts, allowUnavailable)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*model.Collection)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(context.Context, int64, uint64, bool) error); ok {
|
||||
r1 = rf(ctx, collectionID, ts, allowUnavailable)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GetCollectionByName provides a mock function with given fields: ctx, collectionName, ts
|
||||
func (_m *IMetaTable) GetCollectionByName(ctx context.Context, collectionName string, ts uint64) (*model.Collection, error) {
|
||||
ret := _m.Called(ctx, collectionName, ts)
|
||||
|
||||
var r0 *model.Collection
|
||||
if rf, ok := ret.Get(0).(func(context.Context, string, uint64) *model.Collection); ok {
|
||||
r0 = rf(ctx, collectionName, ts)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*model.Collection)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(context.Context, string, uint64) error); ok {
|
||||
r1 = rf(ctx, collectionName, ts)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GetCollectionVirtualChannels provides a mock function with given fields: colID
|
||||
func (_m *IMetaTable) GetCollectionVirtualChannels(colID int64) []string {
|
||||
ret := _m.Called(colID)
|
||||
|
||||
var r0 []string
|
||||
if rf, ok := ret.Get(0).(func(int64) []string); ok {
|
||||
r0 = rf(colID)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]string)
|
||||
}
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// GetCredential provides a mock function with given fields: username
|
||||
func (_m *IMetaTable) GetCredential(username string) (*internalpb.CredentialInfo, error) {
|
||||
ret := _m.Called(username)
|
||||
|
||||
var r0 *internalpb.CredentialInfo
|
||||
if rf, ok := ret.Get(0).(func(string) *internalpb.CredentialInfo); ok {
|
||||
r0 = rf(username)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*internalpb.CredentialInfo)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(string) error); ok {
|
||||
r1 = rf(username)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GetPartitionByName provides a mock function with given fields: collID, partitionName, ts
|
||||
func (_m *IMetaTable) GetPartitionByName(collID int64, partitionName string, ts uint64) (int64, error) {
|
||||
ret := _m.Called(collID, partitionName, ts)
|
||||
|
||||
var r0 int64
|
||||
if rf, ok := ret.Get(0).(func(int64, string, uint64) int64); ok {
|
||||
r0 = rf(collID, partitionName, ts)
|
||||
} else {
|
||||
r0 = ret.Get(0).(int64)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int64, string, uint64) error); ok {
|
||||
r1 = rf(collID, partitionName, ts)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GetPartitionNameByID provides a mock function with given fields: collID, partitionID, ts
|
||||
func (_m *IMetaTable) GetPartitionNameByID(collID int64, partitionID int64, ts uint64) (string, error) {
|
||||
ret := _m.Called(collID, partitionID, ts)
|
||||
|
||||
var r0 string
|
||||
if rf, ok := ret.Get(0).(func(int64, int64, uint64) string); ok {
|
||||
r0 = rf(collID, partitionID, ts)
|
||||
} else {
|
||||
r0 = ret.Get(0).(string)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int64, int64, uint64) error); ok {
|
||||
r1 = rf(collID, partitionID, ts)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// IsAlias provides a mock function with given fields: name
|
||||
func (_m *IMetaTable) IsAlias(name string) bool {
|
||||
ret := _m.Called(name)
|
||||
|
||||
var r0 bool
|
||||
if rf, ok := ret.Get(0).(func(string) bool); ok {
|
||||
r0 = rf(name)
|
||||
} else {
|
||||
r0 = ret.Get(0).(bool)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// ListAbnormalCollections provides a mock function with given fields: ctx, ts
|
||||
func (_m *IMetaTable) ListAbnormalCollections(ctx context.Context, ts uint64) ([]*model.Collection, error) {
|
||||
ret := _m.Called(ctx, ts)
|
||||
|
||||
var r0 []*model.Collection
|
||||
if rf, ok := ret.Get(0).(func(context.Context, uint64) []*model.Collection); ok {
|
||||
r0 = rf(ctx, ts)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*model.Collection)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(context.Context, uint64) error); ok {
|
||||
r1 = rf(ctx, ts)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// ListAliasesByID provides a mock function with given fields: collID
|
||||
func (_m *IMetaTable) ListAliasesByID(collID int64) []string {
|
||||
ret := _m.Called(collID)
|
||||
|
||||
var r0 []string
|
||||
if rf, ok := ret.Get(0).(func(int64) []string); ok {
|
||||
r0 = rf(collID)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]string)
|
||||
}
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// ListCollectionPhysicalChannels provides a mock function with given fields:
|
||||
func (_m *IMetaTable) ListCollectionPhysicalChannels() map[int64][]string {
|
||||
ret := _m.Called()
|
||||
|
||||
var r0 map[int64][]string
|
||||
if rf, ok := ret.Get(0).(func() map[int64][]string); ok {
|
||||
r0 = rf()
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(map[int64][]string)
|
||||
}
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// ListCollections provides a mock function with given fields: ctx, ts
|
||||
func (_m *IMetaTable) ListCollections(ctx context.Context, ts uint64) ([]*model.Collection, error) {
|
||||
ret := _m.Called(ctx, ts)
|
||||
|
||||
var r0 []*model.Collection
|
||||
if rf, ok := ret.Get(0).(func(context.Context, uint64) []*model.Collection); ok {
|
||||
r0 = rf(ctx, ts)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*model.Collection)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(context.Context, uint64) error); ok {
|
||||
r1 = rf(ctx, ts)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// ListCredentialUsernames provides a mock function with given fields:
|
||||
func (_m *IMetaTable) ListCredentialUsernames() (*milvuspb.ListCredUsersResponse, error) {
|
||||
ret := _m.Called()
|
||||
|
||||
var r0 *milvuspb.ListCredUsersResponse
|
||||
if rf, ok := ret.Get(0).(func() *milvuspb.ListCredUsersResponse); ok {
|
||||
r0 = rf()
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*milvuspb.ListCredUsersResponse)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func() error); ok {
|
||||
r1 = rf()
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// ListPolicy provides a mock function with given fields: tenant
|
||||
func (_m *IMetaTable) ListPolicy(tenant string) ([]string, error) {
|
||||
ret := _m.Called(tenant)
|
||||
|
||||
var r0 []string
|
||||
if rf, ok := ret.Get(0).(func(string) []string); ok {
|
||||
r0 = rf(tenant)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]string)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(string) error); ok {
|
||||
r1 = rf(tenant)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// ListUserRole provides a mock function with given fields: tenant
|
||||
func (_m *IMetaTable) ListUserRole(tenant string) ([]string, error) {
|
||||
ret := _m.Called(tenant)
|
||||
|
||||
var r0 []string
|
||||
if rf, ok := ret.Get(0).(func(string) []string); ok {
|
||||
r0 = rf(tenant)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]string)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(string) error); ok {
|
||||
r1 = rf(tenant)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// OperatePrivilege provides a mock function with given fields: tenant, entity, operateType
|
||||
func (_m *IMetaTable) OperatePrivilege(tenant string, entity *milvuspb.GrantEntity, operateType milvuspb.OperatePrivilegeType) error {
|
||||
ret := _m.Called(tenant, entity, operateType)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(string, *milvuspb.GrantEntity, milvuspb.OperatePrivilegeType) error); ok {
|
||||
r0 = rf(tenant, entity, operateType)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// OperateUserRole provides a mock function with given fields: tenant, userEntity, roleEntity, operateType
|
||||
func (_m *IMetaTable) OperateUserRole(tenant string, userEntity *milvuspb.UserEntity, roleEntity *milvuspb.RoleEntity, operateType milvuspb.OperateUserRoleType) error {
|
||||
ret := _m.Called(tenant, userEntity, roleEntity, operateType)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(string, *milvuspb.UserEntity, *milvuspb.RoleEntity, milvuspb.OperateUserRoleType) error); ok {
|
||||
r0 = rf(tenant, userEntity, roleEntity, operateType)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// RemoveCollection provides a mock function with given fields: ctx, collectionID, ts
|
||||
func (_m *IMetaTable) RemoveCollection(ctx context.Context, collectionID int64, ts uint64) error {
|
||||
ret := _m.Called(ctx, collectionID, ts)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int64, uint64) error); ok {
|
||||
r0 = rf(ctx, collectionID, ts)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// RemovePartition provides a mock function with given fields: ctx, collectionID, partitionID, ts
|
||||
func (_m *IMetaTable) RemovePartition(ctx context.Context, collectionID int64, partitionID int64, ts uint64) error {
|
||||
ret := _m.Called(ctx, collectionID, partitionID, ts)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int64, int64, uint64) error); ok {
|
||||
r0 = rf(ctx, collectionID, partitionID, ts)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
return r0
|
||||
}
|
||||
|
||||
// SelectGrant provides a mock function with given fields: tenant, entity
|
||||
func (_m *IMetaTable) SelectGrant(tenant string, entity *milvuspb.GrantEntity) ([]*milvuspb.GrantEntity, error) {
|
||||
ret := _m.Called(tenant, entity)
|
||||
|
||||
var r0 []*milvuspb.GrantEntity
|
||||
if rf, ok := ret.Get(0).(func(string, *milvuspb.GrantEntity) []*milvuspb.GrantEntity); ok {
|
||||
r0 = rf(tenant, entity)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*milvuspb.GrantEntity)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(string, *milvuspb.GrantEntity) error); ok {
|
||||
r1 = rf(tenant, entity)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// SelectRole provides a mock function with given fields: tenant, entity, includeUserInfo
|
||||
func (_m *IMetaTable) SelectRole(tenant string, entity *milvuspb.RoleEntity, includeUserInfo bool) ([]*milvuspb.RoleResult, error) {
|
||||
ret := _m.Called(tenant, entity, includeUserInfo)
|
||||
|
||||
var r0 []*milvuspb.RoleResult
|
||||
if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity, bool) []*milvuspb.RoleResult); ok {
|
||||
r0 = rf(tenant, entity, includeUserInfo)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*milvuspb.RoleResult)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(string, *milvuspb.RoleEntity, bool) error); ok {
|
||||
r1 = rf(tenant, entity, includeUserInfo)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// SelectUser provides a mock function with given fields: tenant, entity, includeRoleInfo
|
||||
func (_m *IMetaTable) SelectUser(tenant string, entity *milvuspb.UserEntity, includeRoleInfo bool) ([]*milvuspb.UserResult, error) {
|
||||
ret := _m.Called(tenant, entity, includeRoleInfo)
|
||||
|
||||
var r0 []*milvuspb.UserResult
|
||||
if rf, ok := ret.Get(0).(func(string, *milvuspb.UserEntity, bool) []*milvuspb.UserResult); ok {
|
||||
r0 = rf(tenant, entity, includeRoleInfo)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*milvuspb.UserResult)
|
||||
}
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(string, *milvuspb.UserEntity, bool) error); ok {
|
||||
r1 = rf(tenant, entity, includeRoleInfo)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
type mockConstructorTestingTNewIMetaTable interface {
|
||||
mock.TestingT
|
||||
Cleanup(func())
|
||||
}
|
||||
|
||||
// NewIMetaTable creates a new instance of IMetaTable. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
|
||||
func NewIMetaTable(t mockConstructorTestingTNewIMetaTable) *IMetaTable {
|
||||
mock := &IMetaTable{}
|
||||
mock.Mock.Test(t)
|
||||
|
||||
t.Cleanup(func() { mock.AssertExpectations(t) })
|
||||
|
||||
return mock
|
||||
}
|
||||
|
|
@ -0,0 +1,216 @@
|
|||
// Code generated by mockery v2.16.0. DO NOT EDIT.
|
||||
|
||||
package mockrootcoord
|
||||
|
||||
import (
|
||||
context "context"
|
||||
|
||||
model "github.com/milvus-io/milvus/internal/metastore/model"
|
||||
mock "github.com/stretchr/testify/mock"
|
||||
)
|
||||
|
||||
// GarbageCollector is an autogenerated mock type for the GarbageCollector type
|
||||
type GarbageCollector struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
type GarbageCollector_Expecter struct {
|
||||
mock *mock.Mock
|
||||
}
|
||||
|
||||
func (_m *GarbageCollector) EXPECT() *GarbageCollector_Expecter {
|
||||
return &GarbageCollector_Expecter{mock: &_m.Mock}
|
||||
}
|
||||
|
||||
// GcCollectionData provides a mock function with given fields: ctx, coll
|
||||
func (_m *GarbageCollector) GcCollectionData(ctx context.Context, coll *model.Collection) (uint64, error) {
|
||||
ret := _m.Called(ctx, coll)
|
||||
|
||||
var r0 uint64
|
||||
if rf, ok := ret.Get(0).(func(context.Context, *model.Collection) uint64); ok {
|
||||
r0 = rf(ctx, coll)
|
||||
} else {
|
||||
r0 = ret.Get(0).(uint64)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(context.Context, *model.Collection) error); ok {
|
||||
r1 = rf(ctx, coll)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GarbageCollector_GcCollectionData_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GcCollectionData'
|
||||
type GarbageCollector_GcCollectionData_Call struct {
|
||||
*mock.Call
|
||||
}
|
||||
|
||||
// GcCollectionData is a helper method to define mock.On call
|
||||
// - ctx context.Context
|
||||
// - coll *model.Collection
|
||||
func (_e *GarbageCollector_Expecter) GcCollectionData(ctx interface{}, coll interface{}) *GarbageCollector_GcCollectionData_Call {
|
||||
return &GarbageCollector_GcCollectionData_Call{Call: _e.mock.On("GcCollectionData", ctx, coll)}
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_GcCollectionData_Call) Run(run func(ctx context.Context, coll *model.Collection)) *GarbageCollector_GcCollectionData_Call {
|
||||
_c.Call.Run(func(args mock.Arguments) {
|
||||
run(args[0].(context.Context), args[1].(*model.Collection))
|
||||
})
|
||||
return _c
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_GcCollectionData_Call) Return(ddlTs uint64, err error) *GarbageCollector_GcCollectionData_Call {
|
||||
_c.Call.Return(ddlTs, err)
|
||||
return _c
|
||||
}
|
||||
|
||||
// GcPartitionData provides a mock function with given fields: ctx, pChannels, partition
|
||||
func (_m *GarbageCollector) GcPartitionData(ctx context.Context, pChannels []string, partition *model.Partition) (uint64, error) {
|
||||
ret := _m.Called(ctx, pChannels, partition)
|
||||
|
||||
var r0 uint64
|
||||
if rf, ok := ret.Get(0).(func(context.Context, []string, *model.Partition) uint64); ok {
|
||||
r0 = rf(ctx, pChannels, partition)
|
||||
} else {
|
||||
r0 = ret.Get(0).(uint64)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(context.Context, []string, *model.Partition) error); ok {
|
||||
r1 = rf(ctx, pChannels, partition)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GarbageCollector_GcPartitionData_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GcPartitionData'
|
||||
type GarbageCollector_GcPartitionData_Call struct {
|
||||
*mock.Call
|
||||
}
|
||||
|
||||
// GcPartitionData is a helper method to define mock.On call
|
||||
// - ctx context.Context
|
||||
// - pChannels []string
|
||||
// - partition *model.Partition
|
||||
func (_e *GarbageCollector_Expecter) GcPartitionData(ctx interface{}, pChannels interface{}, partition interface{}) *GarbageCollector_GcPartitionData_Call {
|
||||
return &GarbageCollector_GcPartitionData_Call{Call: _e.mock.On("GcPartitionData", ctx, pChannels, partition)}
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_GcPartitionData_Call) Run(run func(ctx context.Context, pChannels []string, partition *model.Partition)) *GarbageCollector_GcPartitionData_Call {
|
||||
_c.Call.Run(func(args mock.Arguments) {
|
||||
run(args[0].(context.Context), args[1].([]string), args[2].(*model.Partition))
|
||||
})
|
||||
return _c
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_GcPartitionData_Call) Return(ddlTs uint64, err error) *GarbageCollector_GcPartitionData_Call {
|
||||
_c.Call.Return(ddlTs, err)
|
||||
return _c
|
||||
}
|
||||
|
||||
// ReDropCollection provides a mock function with given fields: collMeta, ts
|
||||
func (_m *GarbageCollector) ReDropCollection(collMeta *model.Collection, ts uint64) {
|
||||
_m.Called(collMeta, ts)
|
||||
}
|
||||
|
||||
// GarbageCollector_ReDropCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ReDropCollection'
|
||||
type GarbageCollector_ReDropCollection_Call struct {
|
||||
*mock.Call
|
||||
}
|
||||
|
||||
// ReDropCollection is a helper method to define mock.On call
|
||||
// - collMeta *model.Collection
|
||||
// - ts uint64
|
||||
func (_e *GarbageCollector_Expecter) ReDropCollection(collMeta interface{}, ts interface{}) *GarbageCollector_ReDropCollection_Call {
|
||||
return &GarbageCollector_ReDropCollection_Call{Call: _e.mock.On("ReDropCollection", collMeta, ts)}
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_ReDropCollection_Call) Run(run func(collMeta *model.Collection, ts uint64)) *GarbageCollector_ReDropCollection_Call {
|
||||
_c.Call.Run(func(args mock.Arguments) {
|
||||
run(args[0].(*model.Collection), args[1].(uint64))
|
||||
})
|
||||
return _c
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_ReDropCollection_Call) Return() *GarbageCollector_ReDropCollection_Call {
|
||||
_c.Call.Return()
|
||||
return _c
|
||||
}
|
||||
|
||||
// ReDropPartition provides a mock function with given fields: pChannels, partition, ts
|
||||
func (_m *GarbageCollector) ReDropPartition(pChannels []string, partition *model.Partition, ts uint64) {
|
||||
_m.Called(pChannels, partition, ts)
|
||||
}
|
||||
|
||||
// GarbageCollector_ReDropPartition_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ReDropPartition'
|
||||
type GarbageCollector_ReDropPartition_Call struct {
|
||||
*mock.Call
|
||||
}
|
||||
|
||||
// ReDropPartition is a helper method to define mock.On call
|
||||
// - pChannels []string
|
||||
// - partition *model.Partition
|
||||
// - ts uint64
|
||||
func (_e *GarbageCollector_Expecter) ReDropPartition(pChannels interface{}, partition interface{}, ts interface{}) *GarbageCollector_ReDropPartition_Call {
|
||||
return &GarbageCollector_ReDropPartition_Call{Call: _e.mock.On("ReDropPartition", pChannels, partition, ts)}
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_ReDropPartition_Call) Run(run func(pChannels []string, partition *model.Partition, ts uint64)) *GarbageCollector_ReDropPartition_Call {
|
||||
_c.Call.Run(func(args mock.Arguments) {
|
||||
run(args[0].([]string), args[1].(*model.Partition), args[2].(uint64))
|
||||
})
|
||||
return _c
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_ReDropPartition_Call) Return() *GarbageCollector_ReDropPartition_Call {
|
||||
_c.Call.Return()
|
||||
return _c
|
||||
}
|
||||
|
||||
// RemoveCreatingCollection provides a mock function with given fields: collMeta
|
||||
func (_m *GarbageCollector) RemoveCreatingCollection(collMeta *model.Collection) {
|
||||
_m.Called(collMeta)
|
||||
}
|
||||
|
||||
// GarbageCollector_RemoveCreatingCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RemoveCreatingCollection'
|
||||
type GarbageCollector_RemoveCreatingCollection_Call struct {
|
||||
*mock.Call
|
||||
}
|
||||
|
||||
// RemoveCreatingCollection is a helper method to define mock.On call
|
||||
// - collMeta *model.Collection
|
||||
func (_e *GarbageCollector_Expecter) RemoveCreatingCollection(collMeta interface{}) *GarbageCollector_RemoveCreatingCollection_Call {
|
||||
return &GarbageCollector_RemoveCreatingCollection_Call{Call: _e.mock.On("RemoveCreatingCollection", collMeta)}
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_RemoveCreatingCollection_Call) Run(run func(collMeta *model.Collection)) *GarbageCollector_RemoveCreatingCollection_Call {
|
||||
_c.Call.Run(func(args mock.Arguments) {
|
||||
run(args[0].(*model.Collection))
|
||||
})
|
||||
return _c
|
||||
}
|
||||
|
||||
func (_c *GarbageCollector_RemoveCreatingCollection_Call) Return() *GarbageCollector_RemoveCreatingCollection_Call {
|
||||
_c.Call.Return()
|
||||
return _c
|
||||
}
|
||||
|
||||
type mockConstructorTestingTNewGarbageCollector interface {
|
||||
mock.TestingT
|
||||
Cleanup(func())
|
||||
}
|
||||
|
||||
// NewGarbageCollector creates a new instance of GarbageCollector. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
|
||||
func NewGarbageCollector(t mockConstructorTestingTNewGarbageCollector) *GarbageCollector {
|
||||
mock := &GarbageCollector{}
|
||||
mock.Mock.Test(t)
|
||||
|
||||
t.Cleanup(func() { mock.AssertExpectations(t) })
|
||||
|
||||
return mock
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue