From fdcb76490a945f1188e45340f8d4146bd817d890 Mon Sep 17 00:00:00 2001
From: Christian Grothoff <christian@grothoff.org>
Date: Mon, 23 Jan 2017 14:18:35 +0100
Subject: [PATCH] handle disconnect properly in test

---
 src/cadet/test_cadet_single.c | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/cadet/test_cadet_single.c b/src/cadet/test_cadet_single.c
index 9bfaeda0d..636def35d 100644
--- a/src/cadet/test_cadet_single.c
+++ b/src/cadet/test_cadet_single.c
@@ -190,13 +190,18 @@ channel_end (void *cls, const struct GNUNET_CADET_Channel *channel,
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
               "incoming channel closed at peer %ld\n",
               id);
-  if (REPETITIONS == repetition && channel == ch2)
+  if ( (REPETITIONS == repetition) &&
+       (channel == ch2) )
   {
-    ch2 = NULL;
-    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "everything fine! finishing!\n");
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "everything fine! finishing!\n");
     result = GNUNET_OK;
     GNUNET_SCHEDULER_shutdown ();
   }
+  if (channel == ch2)
+    ch2 = NULL;
+  if (channel == ch1)
+    ch1 = NULL;
 }
 
 
-- 
2.25.1