Browse Source

etcdserver/api/rafthttp: configure "streamProber" in tests

Signed-off-by: Gyuho Lee <leegyuho@amazon.com>
Gyuho Lee 7 years ago
parent
commit
884a8bd36b
1 changed files with 4 additions and 0 deletions
  1. 4 0
      etcdserver/api/rafthttp/transport_test.go

+ 4 - 0
etcdserver/api/rafthttp/transport_test.go

@@ -101,6 +101,7 @@ func TestTransportAdd(t *testing.T) {
 		streamRt:       &roundTripperRecorder{},
 		streamRt:       &roundTripperRecorder{},
 		peers:          make(map[types.ID]Peer),
 		peers:          make(map[types.ID]Peer),
 		pipelineProber: probing.NewProber(nil),
 		pipelineProber: probing.NewProber(nil),
+		streamProber:   probing.NewProber(nil),
 	}
 	}
 	tr.AddPeer(1, []string{"http://localhost:2380"})
 	tr.AddPeer(1, []string{"http://localhost:2380"})
 
 
@@ -129,6 +130,7 @@ func TestTransportRemove(t *testing.T) {
 		streamRt:       &roundTripperRecorder{},
 		streamRt:       &roundTripperRecorder{},
 		peers:          make(map[types.ID]Peer),
 		peers:          make(map[types.ID]Peer),
 		pipelineProber: probing.NewProber(nil),
 		pipelineProber: probing.NewProber(nil),
+		streamProber:   probing.NewProber(nil),
 	}
 	}
 	tr.AddPeer(1, []string{"http://localhost:2380"})
 	tr.AddPeer(1, []string{"http://localhost:2380"})
 	tr.RemovePeer(types.ID(1))
 	tr.RemovePeer(types.ID(1))
@@ -144,6 +146,7 @@ func TestTransportUpdate(t *testing.T) {
 	tr := &Transport{
 	tr := &Transport{
 		peers:          map[types.ID]Peer{types.ID(1): peer},
 		peers:          map[types.ID]Peer{types.ID(1): peer},
 		pipelineProber: probing.NewProber(nil),
 		pipelineProber: probing.NewProber(nil),
+		streamProber:   probing.NewProber(nil),
 	}
 	}
 	u := "http://localhost:2380"
 	u := "http://localhost:2380"
 	tr.UpdatePeer(types.ID(1), []string{u})
 	tr.UpdatePeer(types.ID(1), []string{u})
@@ -163,6 +166,7 @@ func TestTransportErrorc(t *testing.T) {
 		pipelineRt:     newRespRoundTripper(http.StatusForbidden, nil),
 		pipelineRt:     newRespRoundTripper(http.StatusForbidden, nil),
 		peers:          make(map[types.ID]Peer),
 		peers:          make(map[types.ID]Peer),
 		pipelineProber: probing.NewProber(nil),
 		pipelineProber: probing.NewProber(nil),
+		streamProber:   probing.NewProber(nil),
 	}
 	}
 	tr.AddPeer(1, []string{"http://localhost:2380"})
 	tr.AddPeer(1, []string{"http://localhost:2380"})
 	defer tr.Stop()
 	defer tr.Stop()