mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-06 17:18:35 +08:00
5ad8a29c0b
Make CDC watch etcd replicate pchannel meta instead of listing them periodically. issue: https://github.com/milvus-io/milvus/issues/44123 --------- Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
//go:build test
|
|
// +build test
|
|
|
|
package resource
|
|
|
|
import (
|
|
"testing"
|
|
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
|
|
"github.com/milvus-io/milvus/internal/cdc/cluster"
|
|
"github.com/milvus-io/milvus/internal/cdc/controller"
|
|
"github.com/milvus-io/milvus/internal/cdc/replication"
|
|
"github.com/milvus-io/milvus/internal/mocks/mock_metastore"
|
|
"github.com/milvus-io/milvus/pkg/v2/mocks/mock_kv"
|
|
)
|
|
|
|
// InitForTest initializes the singleton of resources for test.
|
|
func InitForTest(t *testing.T, opts ...optResourceInit) {
|
|
r = &resourceImpl{}
|
|
for _, opt := range opts {
|
|
opt(r)
|
|
}
|
|
if r.metaKV == nil {
|
|
r.metaKV = mock_kv.NewMockMetaKv(t)
|
|
}
|
|
if r.etcdClient == nil {
|
|
r.etcdClient = &clientv3.Client{}
|
|
}
|
|
if r.catalog == nil {
|
|
r.catalog = mock_metastore.NewMockReplicationCatalog(t)
|
|
}
|
|
if r.clusterClient == nil {
|
|
r.clusterClient = cluster.NewMockClusterClient(t)
|
|
}
|
|
if r.replicateManagerClient == nil {
|
|
r.replicateManagerClient = replication.NewMockReplicateManagerClient(t)
|
|
}
|
|
if r.controller == nil {
|
|
r.controller = controller.NewMockController(t)
|
|
}
|
|
}
|