[NOD-1208] Use netConnection.Disconnect in Routes.Disconnect

This commit is contained in:
Mike Zak 2020-08-05 15:46:32 +03:00
parent cd10de2dce
commit 9d60bb1ee7
2 changed files with 6 additions and 5 deletions

View File

@ -170,6 +170,7 @@ func generateRouteInitializer() (netadapter.RouterInitializer, <-chan *Routes) {
spawn("netAdapterMock-routeInitializer-sendRoutesToChan", func() {
routesChan <- &Routes{
netConnection: netConnection,
OutgoingRoute: router.OutgoingRoute(),
IncomingRoute: everythingElseRoute,
handshakeRoute: handshakeRoute,

View File

@ -3,6 +3,8 @@ package netadaptermock
import (
"time"
"github.com/kaspanet/kaspad/netadapter"
"github.com/pkg/errors"
"github.com/kaspanet/kaspad/netadapter/router"
@ -11,6 +13,7 @@ import (
// Routes holds the incoming and outgoing routes of a connection created by NetAdapterMock
type Routes struct {
netConnection *netadapter.NetConnection
IncomingRoute, OutgoingRoute *router.Route
handshakeRoute *router.Route
pingRoute *router.Route
@ -46,9 +49,6 @@ func (r *Routes) WaitForDisconnect(timeout time.Duration) error {
}
// Close closes all the routes in this Routes object
func (r *Routes) Close() {
r.IncomingRoute.Close()
r.OutgoingRoute.Close()
r.handshakeRoute.Close()
r.pingRoute.Close()
func (r *Routes) Disconnect() {
r.netConnection.Disconnect()
}