From f75c5340d2abfd9df91aee9ca23a65b6c5578e63 Mon Sep 17 00:00:00 2001
From: Manfred Karrer <mk@nucleo.io>
Date: Fri, 24 Oct 2014 14:18:10 +0200
Subject: [PATCH] Fix bug in relay mode

---
 .../java/io/bitsquare/msg/BasicUsecasesInWANTest.java    | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/test/java/io/bitsquare/msg/BasicUsecasesInWANTest.java b/src/test/java/io/bitsquare/msg/BasicUsecasesInWANTest.java
index 39c1125cf2..08854ae00e 100644
--- a/src/test/java/io/bitsquare/msg/BasicUsecasesInWANTest.java
+++ b/src/test/java/io/bitsquare/msg/BasicUsecasesInWANTest.java
@@ -241,10 +241,12 @@ No future set beforehand, probably an early shutdown / timeout, or use setFailed
                 FutureNAT futureNAT = peerNAT.startSetupPortforwarding(futureDiscover);
                 futureNAT.awaitUninterruptibly();
                 if (futureNAT.isSuccess()) {
+                    log.info("Automatic port forwarding is setup. Address = " +
+                            futureNAT.peerAddress());
                     FutureDiscover futureDiscover2 = peer.discover().peerAddress(masterNodeAddress).start();
                     futureDiscover2.awaitUninterruptibly();
                     if (futureDiscover2.isSuccess()) {
-                        log.info("Discover with direct connection successful. Address = " + futureDiscover2
+                        log.info("Discover with automatic port forwarding successful. Address = " + futureDiscover2
                                 .peerAddress());
 
                         log.info("Automatic port forwarding is setup. Address = " +
@@ -261,10 +263,7 @@ No future set beforehand, probably an early shutdown / timeout, or use setFailed
                     FutureRelayNAT futureRelayNAT = peerNAT.startRelay(futureDiscover, futureNAT);
                     futureRelayNAT.awaitUninterruptibly();
                     if (futureRelayNAT.isSuccess()) {
-                        log.info("Bootstrap using relay successful. Address = " +
-                                peer.peerAddress());
-                        futureRelayNAT.shutdown();
-                        peer.shutdown().awaitUninterruptibly();
+                        log.info("Bootstrap using relay successful. Address = " + peer.peerAddress());
                         return peerDHT;
                     }
                     else {