@ -65,6 +65,9 @@ func NewNodeTransport(transport *raft.NetworkTransport) *NodeTransport {
// Close closes the transport
func (n *NodeTransport) Close() error {
close(n.done)
if n.NetworkTransport == nil {
return nil
}
return n.NetworkTransport.Close()
@ -9,3 +9,13 @@ func Test_NewTransport(t *testing.T) {
t.Fatal("failed to create new Transport")
func Test_NewNodeTransport(t *testing.T) {
nt := NewNodeTransport(nil)
if nt == nil {
t.Fatal("failed to create new NodeTransport")
if err := nt.Close(); err != nil {
t.Fatalf("failed to close NodeTransport: %s", err.Error())