milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Check server test error (#9954) 

Signed-off-by: godchen <qingxiang.chen@zilliz.com>
pull/9968/head
godchen 2021-10-15 17:12:35 +08:00 committed by GitHub
parent bf71965f58
commit 3f713a1104
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
internal/datacoord/server_test.go
View File

@ -867,7 +867,8 @@ func TestDataNodeTtChannel(t *testing.T) {
Address: "localhost:7777",
NodeID: 0,
}
svr.cluster.Register(info)
err = svr.cluster.Register(info)
assert.Nil(t, err)
resp, err := svr.AssignSegmentID(context.TODO(), &datapb.AssignSegmentIDRequest{
NodeID: 0,
@ -930,7 +931,8 @@ func TestDataNodeTtChannel(t *testing.T) {
Address: "localhost:7777",
NodeID: 0,
}
svr.cluster.Register(info)
err = svr.cluster.Register(info)
assert.Nil(t, err)
resp, err := svr.AssignSegmentID(context.TODO(), &datapb.AssignSegmentIDRequest{
NodeID: 0,
PeerRole: "",
@ -1007,7 +1009,8 @@ func TestDataNodeTtChannel(t *testing.T) {
NodeID: 0,
Address: "localhost:7777",
}
svr.cluster.Register(node)
err = svr.cluster.Register(node)
assert.Nil(t, err)
resp, err := svr.AssignSegmentID(context.TODO(), &datapb.AssignSegmentIDRequest{
NodeID: 0,
@ -1349,7 +1352,8 @@ func TestHandleSessionEvent(t *testing.T) {
cluster := NewCluster(sessionManager, channelManager)
assert.Nil(t, err)
cluster.Startup(nil)
err = cluster.Startup(nil)
assert.Nil(t, err)
defer cluster.Close()
svr := newTestServer(t, nil, SetCluster(cluster))
@ -1365,7 +1369,8 @@ func TestHandleSessionEvent(t *testing.T) {
Exclusive: false,
},
}
svr.handleSessionEvent(context.Background(), evt)
err = svr.handleSessionEvent(context.Background(), evt)
assert.Nil(t, err)
evt = &sessionutil.SessionEvent{
EventType: sessionutil.SessionAddEvent,
@ -1376,7 +1381,8 @@ func TestHandleSessionEvent(t *testing.T) {
Exclusive: false,
},
}
svr.handleSessionEvent(context.Background(), evt)
err = svr.handleSessionEvent(context.Background(), evt)
assert.Nil(t, err)
dataNodes := svr.cluster.GetSessions()
assert.EqualValues(t, 1, len(dataNodes))
assert.EqualValues(t, "DN127.0.0.101", dataNodes[0].info.Address)
@ -1390,14 +1396,16 @@ func TestHandleSessionEvent(t *testing.T) {
Exclusive: false,
},
}
svr.handleSessionEvent(context.Background(), evt)
err = svr.handleSessionEvent(context.Background(), evt)
assert.Nil(t, err)
dataNodes = svr.cluster.GetSessions()
assert.EqualValues(t, 0, len(dataNodes))
})
t.Run("nil evt", func(t *testing.T) {
assert.NotPanics(t, func() {
svr.handleSessionEvent(context.Background(), nil)
err = svr.handleSessionEvent(context.Background(), nil)
assert.Nil(t, err)
})
})
}