cmd/swarm-smoke: fix waitToPushSynced connection closing (#1781)

ethersphere · Sep 19, 2019 · f185630 · f185630
1 parent d13ef0a
commit f185630
Showing 1 changed file with 17 additions and 16 deletions.
diff --git a/cmd/swarm-smoke/upload_and_sync.go b/cmd/swarm-smoke/upload_and_sync.go
@@ -333,24 +333,25 @@ func waitToPushSynced(tagname string) {
 	for {
 		time.Sleep(200 * time.Millisecond)
 
-		rpcClient, err := rpc.Dial(wsEndpoint(hosts[0]))
-		if rpcClient != nil {
-			defer rpcClient.Close()
-		}
-		if err != nil {
-			log.Error("error dialing host", "err", err)
-			continue
-		}
-
-		bzzClient := client.NewBzz(rpcClient)
+		if func() (synced bool) {
+			rpcClient, err := rpc.Dial(wsEndpoint(hosts[0]))
+			if rpcClient != nil {
+				defer rpcClient.Close()
+			}
+			if err != nil {
+				log.Error("error dialing host", "err", err)
+				return false
+			}
 
-		synced, err := bzzClient.IsPushSynced(tagname)
-		if err != nil {
-			log.Error(err.Error())
-			continue
-		}
+			bzzClient := client.NewBzz(rpcClient)
 
-		if synced {
+			synced, err = bzzClient.IsPushSynced(tagname)
+			if err != nil {
+				log.Error(err.Error())
+				return false
+			}
+			return synced
+		}() {
 			return
 		}
 	}