Return True/False for start method.

p2pfs/core/peer.py

@@ -41,8 +41,9 @@ class Peer(MessageServer):
             self._server_sock = self._connect(*self._serverconfig)
         except ConnectionRefusedError:
             logger.error('Server connection refused!')
-            exit(1)
+            return False, 'Server connection refused!'
         super().start()
+        return True, None
 
     def publish(self, file):
         path, filename = os.path.split(file)

tests/test_main.py

@@ -15,6 +15,12 @@ def fmd5(fname):
     return hash_md5.hexdigest()
 
 
+def test_server_refused():
+    peer = Peer('localhost', 0, 'localhost', 8880)
+    started, _ = peer.start()
+    assert not started
+
+
 def test_main():
     peer_1, peer_2 = Peer('localhost', 0, 'localhost', 8880), Peer('localhost', 0, 'localhost', 8880)
     tracker = Tracker('localhost', 8880)
@@ -26,9 +32,10 @@ def test_main():
         for _ in range(500):
             fout.write(os.urandom(1000 * 1000))
 
-    tracker.start()
-    peer_1.start()
-    peer_2.start()
+    tracker_started, _ = tracker.start()
+    peer_1_started, _ = peer_1.start()
+    peer_2_started, _ = peer_2.start()
+    assert tracker_started and peer_1_started and peer_2_started
 
     # peer1 publish small file and peer2 downloads it
     peer_1.publish('test_small_file')