From 17e61a34c4ad58a4684db363bd9fa307473c404d Mon Sep 17 00:00:00 2001
From: Yuxin Wang <yuxinwang.dev@gmail.com>
Date: Sat, 13 Oct 2018 19:04:45 -0400
Subject: [PATCH] Yield data after registration.

---
 p2pfs/core/peer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/p2pfs/core/peer.py b/p2pfs/core/peer.py
index 8e2d22d..df51dd4 100644
--- a/p2pfs/core/peer.py
+++ b/p2pfs/core/peer.py
@@ -200,8 +200,6 @@ class DownloadManager:
                         self._read_tasks[asyncio.ensure_future(read_message(reader))] = peer_address
                         continue
 
-                    yield number, raw_data
-
                     # remove successfully-received chunk from pending and download plans
                     del self._file_chunk_info[number]
                     del self._pending_chunknum[number]
@@ -216,6 +214,8 @@ class DownloadManager:
                     except (ConnectionError, RuntimeError):
                         # stop querying tracker
                         pass
+                    
+                    yield number, raw_data
 
                     # send out request chunk
                     if len(self._to_download_chunk) > 0: