Add insert node failed unittests for query node (#7914)

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
pull/7939/head
bigsheeper 2021-09-15 10:11:48 +08:00 committed by GitHub
parent bc1f39bb55
commit 0b849d3089
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 46 additions and 0 deletions

View File

@ -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)
})
}