Merge pull request #14284 from AustinBenoit/test/server_transport_test_remove

rafthttp: test multiple transport removes
This commit is contained in:
Benjamin Wang 2022-08-01 07:12:05 +08:00 committed by GitHub
commit 21995a3b76
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -142,6 +142,25 @@ func TestTransportRemove(t *testing.T) {
}
}
func TestTransportRemoveIsIdempotent(t *testing.T) {
tr := &Transport{
LeaderStats: stats.NewLeaderStats(zaptest.NewLogger(t), ""),
streamRt: &roundTripperRecorder{},
peers: make(map[types.ID]Peer),
pipelineProber: probing.NewProber(nil),
streamProber: probing.NewProber(nil),
}
tr.AddPeer(1, []string{"http://localhost:2380"})
tr.RemovePeer(types.ID(1))
tr.RemovePeer(types.ID(1))
defer tr.Stop()
if _, ok := tr.peers[types.ID(1)]; ok {
t.Fatalf("senders[1] exists, want removed")
}
}
func TestTransportUpdate(t *testing.T) {
peer := newFakePeer()
tr := &Transport{