fix: unittest

Signed-off-by: chyezh <[email protected]>
milvus-io · Dec 26, 2024 · be3fa87 · be3fa87
1 parent 16224f3
commit be3fa87
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/datanode/services_test.go b/internal/datanode/services_test.go
@@ -329,6 +329,10 @@ func (s *DataNodeServicesSuite) TestCompaction() {
 
 func (s *DataNodeServicesSuite) TestFlushSegments() {
 	dmChannelName := "fake-by-dev-rootcoord-dml-channel-test-FlushSegments"
+	stream, err := s.node.factory.NewTtMsgStream(context.Background())
+	s.NoError(err)
+	s.NotNil(stream)
+	stream.AsProducer(context.Background(), []string{dmChannelName})
 	schema := &schemapb.CollectionSchema{
 		Name: "test_collection",
 		Fields: []*schemapb.FieldSchema{