@@ -140,8 +140,9 @@ func (suite *EmbeddingNodeSuite) TestCreateEmbeddingNode() {
140
140
collSchema := proto .Clone (suite .collectionSchema ).(* schemapb.CollectionSchema )
141
141
collection := segments .NewCollectionWithoutSegcoreForTest (suite .collectionID , collSchema )
142
142
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (collection ).Once ()
143
- _ , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
143
+ node , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
144
144
suite .NoError (err )
145
+ defer node .Close ()
145
146
})
146
147
}
147
148
@@ -151,6 +152,7 @@ func (suite *EmbeddingNodeSuite) TestOperator() {
151
152
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (collection ).Once ()
152
153
node , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
153
154
suite .NoError (err )
155
+ defer node .Close ()
154
156
155
157
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (nil ).Once ()
156
158
suite .Panics (func () {
@@ -163,6 +165,7 @@ func (suite *EmbeddingNodeSuite) TestOperator() {
163
165
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (collection ).Times (2 )
164
166
node , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
165
167
suite .NoError (err )
168
+ defer node .Close ()
166
169
167
170
suite .Panics (func () {
168
171
node .Operate (& insertNodeMsg {
@@ -192,6 +195,7 @@ func (suite *EmbeddingNodeSuite) TestOperator() {
192
195
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (collection ).Times (2 )
193
196
node , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
194
197
suite .NoError (err )
198
+ defer node .Close ()
195
199
196
200
suite .NotPanics (func () {
197
201
output := node .Operate (& insertNodeMsg {
@@ -213,6 +217,7 @@ func (suite *EmbeddingNodeSuite) TestAddInsertData() {
213
217
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (collection ).Once ()
214
218
node , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
215
219
suite .NoError (err )
220
+ defer node .Close ()
216
221
217
222
// transfer insert msg failed because rowbase data not support sparse vector
218
223
insertDatas := make (map [int64 ]* delegator.InsertData )
@@ -237,6 +242,7 @@ func (suite *EmbeddingNodeSuite) TestAddInsertData() {
237
242
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (collection ).Once ()
238
243
node , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
239
244
suite .NoError (err )
245
+ defer node .Close ()
240
246
241
247
insertDatas := make (map [int64 ]* delegator.InsertData )
242
248
err = node .addInsertData (insertDatas , suite .msgs [0 ], collection )
@@ -250,6 +256,7 @@ func (suite *EmbeddingNodeSuite) TestBM25Embedding() {
250
256
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (collection ).Once ()
251
257
node , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
252
258
suite .NoError (err )
259
+ defer node .Close ()
253
260
254
261
runner := function .NewMockFunctionRunner (suite .T ())
255
262
runner .EXPECT ().BatchRun (mock .Anything ).Return (nil , errors .New ("mock error" ))
@@ -265,6 +272,7 @@ func (suite *EmbeddingNodeSuite) TestBM25Embedding() {
265
272
suite .colManager .EXPECT ().Get (suite .collectionID ).Return (collection ).Once ()
266
273
node , err := newEmbeddingNode (suite .collectionID , suite .channel , suite .manager , 128 )
267
274
suite .NoError (err )
275
+ defer node .Close ()
268
276
269
277
runner := function .NewMockFunctionRunner (suite .T ())
270
278
runner .EXPECT ().BatchRun (mock .Anything ).Return ([]interface {}{1 }, nil )
0 commit comments