diff --git a/internal/querynode/streaming_test.go b/internal/querynode/streaming_test.go
index 86925ba218f2afd69e780e900a33a2a9de2bd340..1963366c613eebc340a9add6960ace89c60ad0ab 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) {