From 9601cb2ee6c7f573e4331aaa509544ded0daad2f Mon Sep 17 00:00:00 2001 From: bigsheeper Date: Fri, 17 Sep 2021 11:37:49 +0800 Subject: [PATCH] Add failed streming unittest for query node (#7960) Signed-off-by: bigsheeper --- internal/querynode/streaming_test.go | 43 ++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) diff --git a/internal/querynode/streaming_test.go b/internal/querynode/streaming_test.go index 86925ba21..1963366c6 100644 --- a/internal/querynode/streaming_test.go +++ b/internal/querynode/streaming_test.go @@ -117,6 +117,49 @@ func TestStreaming_search(t *testing.T) { Timestamp(0)) assert.Error(t, err) }) + + t.Run("test no partitions in collection", func(t *testing.T) { + streaming, err := genSimpleStreaming(ctx) + assert.NoError(t, err) + defer streaming.close() + + plan, searchReqs, err := genSimpleSearchPlanAndRequests() + assert.NoError(t, err) + + err = streaming.replica.removePartition(defaultPartitionID) + assert.NoError(t, err) + + res, err := streaming.search(searchReqs, + defaultCollectionID, + []UniqueID{}, + defaultVChannel, + plan, + Timestamp(0)) + assert.NoError(t, err) + assert.Nil(t, res) + }) + + t.Run("test search failed", func(t *testing.T) { + streaming, err := genSimpleStreaming(ctx) + assert.NoError(t, err) + defer streaming.close() + + plan, searchReqs, err := genSimpleSearchPlanAndRequests() + assert.NoError(t, err) + + seg, err := streaming.replica.getSegmentByID(defaultSegmentID) + assert.NoError(t, err) + + seg.segmentPtr = nil + + _, err = streaming.search(searchReqs, + defaultCollectionID, + []UniqueID{}, + defaultVChannel, + plan, + Timestamp(0)) + assert.Error(t, err) + }) } func TestStreaming_retrieve(t *testing.T) { -- GitLab