mirror of https://github.com/milvus-io/milvus.git
Add insert node failed unittests for query node (#7914)
Signed-off-by: bigsheeper <yihao.dai@zilliz.com>pull/7939/head
parent
bc1f39bb55
commit
0b849d3089
|
@ -67,6 +67,28 @@ func TestFlowGraphInsertNode_insert(t *testing.T) {
|
|||
insertNode.insert(insertData, defaultSegmentID, wg)
|
||||
})
|
||||
|
||||
t.Run("test segment insert error", func(t *testing.T) {
|
||||
replica, err := genSimpleReplica()
|
||||
assert.NoError(t, err)
|
||||
insertNode := newInsertNode(replica)
|
||||
|
||||
err = replica.addSegment(defaultSegmentID,
|
||||
defaultPartitionID,
|
||||
defaultCollectionID,
|
||||
defaultVChannel,
|
||||
segmentTypeGrowing,
|
||||
true)
|
||||
assert.NoError(t, err)
|
||||
|
||||
insertData, err := genFlowGraphInsertData()
|
||||
assert.NoError(t, err)
|
||||
|
||||
wg := &sync.WaitGroup{}
|
||||
wg.Add(1)
|
||||
insertData.insertRecords[defaultSegmentID][0].Value = insertData.insertRecords[defaultSegmentID][0].Value[:len(insertData.insertRecords[defaultSegmentID][0].Value)/2]
|
||||
insertNode.insert(insertData, defaultSegmentID, wg)
|
||||
})
|
||||
|
||||
t.Run("test no target segment", func(t *testing.T) {
|
||||
replica, err := genSimpleReplica()
|
||||
assert.NoError(t, err)
|
||||
|
@ -119,4 +141,28 @@ func TestFlowGraphInsertNode_operate(t *testing.T) {
|
|||
msg := []flowgraph.Msg{&iMsg}
|
||||
insertNode.Operate(msg)
|
||||
})
|
||||
|
||||
t.Run("test invalid input length", func(t *testing.T) {
|
||||
replica, err := genSimpleReplica()
|
||||
assert.NoError(t, err)
|
||||
insertNode := newInsertNode(replica)
|
||||
|
||||
err = replica.addSegment(defaultSegmentID,
|
||||
defaultPartitionID,
|
||||
defaultCollectionID,
|
||||
defaultVChannel,
|
||||
segmentTypeGrowing,
|
||||
true)
|
||||
assert.NoError(t, err)
|
||||
|
||||
msgInsertMsg, err := genSimpleInsertMsg()
|
||||
assert.NoError(t, err)
|
||||
iMsg := insertMsg{
|
||||
insertMessages: []*msgstream.InsertMsg{
|
||||
msgInsertMsg,
|
||||
},
|
||||
}
|
||||
msg := []flowgraph.Msg{&iMsg, &iMsg}
|
||||
insertNode.Operate(msg)
|
||||
})
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue