From 590afebc0ae7e6dda22a559c4ddbca6315f25860 Mon Sep 17 00:00:00 2001
From: Jakob Borg <jakob@nym.se>
Date: Mon, 6 Oct 2014 11:13:00 +0200
Subject: [PATCH] Handle connection closed in reconnect test

---
 test/reconnect_test.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/test/reconnect_test.go b/test/reconnect_test.go
index 21ec8c32..837ce17b 100644
--- a/test/reconnect_test.go
+++ b/test/reconnect_test.go
@@ -19,6 +19,7 @@ package integration_test
 
 import (
 	"log"
+	"strings"
 	"sync"
 	"testing"
 	"time"
@@ -82,6 +83,10 @@ func testRestartDuringTransfer(t *testing.T, restartSender, restartReceiver bool
 	for {
 		comp, err := sender.peerCompletion()
 		if err != nil {
+			if strings.Contains(err.Error(), "use of closed network connection") {
+				time.Sleep(250 * time.Millisecond)
+				continue
+			}
 			sender.stop()
 			receiver.stop()
 			t.Fatal(err)