mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 06:55:27 +08:00
40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package reader
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestUtilFunctions_GetKey2Segments(t *testing.T) {
|
|
// TODO: Add GetKey2Segments test
|
|
}
|
|
|
|
func TestUtilFunctions_GetCollectionByCollectionName(t *testing.T) {
|
|
// 1. Construct node, and collections
|
|
node := NewQueryNode(0, 0)
|
|
var _ = node.NewCollection(0, "collection0", "")
|
|
|
|
// 2. Get collection by collectionName
|
|
var c0, err = node.GetCollectionByCollectionName("collection0")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, c0.CollectionName, "collection0")
|
|
c0 = node.GetCollectionByID(0)
|
|
assert.NotNil(t, c0)
|
|
assert.Equal(t, c0.CollectionID, 0)
|
|
}
|
|
|
|
func TestUtilFunctions_GetSegmentBySegmentID(t *testing.T) {
|
|
// 1. Construct node, collection, partition and segment
|
|
node := NewQueryNode(0, 0)
|
|
var collection = node.NewCollection(0, "collection0", "")
|
|
var partition = collection.NewPartition("partition0")
|
|
var segment = partition.NewSegment(0)
|
|
node.SegmentsMap[0] = segment
|
|
|
|
// 2. Get segment by segment id
|
|
var s0, err = node.GetSegmentBySegmentID(0)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, s0.SegmentId, int64(0))
|
|
}
|