Return boolean value for start method.
This commit is contained in:
parent
97b2a34bef
commit
854d1676bb
3 changed files with 7 additions and 6 deletions
|
@ -40,7 +40,7 @@ class Peer(MessageServer):
|
|||
self._server_sock = self._connect(self._serverconfig)
|
||||
except ConnectionRefusedError:
|
||||
logger.error('Server connection refused!')
|
||||
return False, 'Server connection refused!'
|
||||
return False
|
||||
# start the internal server
|
||||
super().start()
|
||||
# send out register message
|
||||
|
@ -49,7 +49,7 @@ class Peer(MessageServer):
|
|||
'type': MessageType.REQUEST_REGISTER,
|
||||
'address': self._sock.getsockname()
|
||||
})
|
||||
return True, None
|
||||
return True
|
||||
|
||||
def publish(self, file):
|
||||
path, filename = os.path.split(file)
|
||||
|
|
|
@ -36,6 +36,7 @@ class MessageServer:
|
|||
thread = threading.Thread(target=self._listen)
|
||||
thread.start()
|
||||
self._threads.add(thread)
|
||||
return True
|
||||
|
||||
def stop(self):
|
||||
# shutdown the server
|
||||
|
|
|
@ -17,7 +17,7 @@ def fmd5(fname):
|
|||
|
||||
def test_server_refused():
|
||||
peer = Peer('localhost', 0, 'localhost', 8880)
|
||||
started, _ = peer.start()
|
||||
started = peer.start()
|
||||
assert not started
|
||||
|
||||
|
||||
|
@ -32,9 +32,9 @@ def test_main():
|
|||
for _ in range(500):
|
||||
fout.write(os.urandom(1000 * 1000))
|
||||
|
||||
tracker_started, _ = tracker.start()
|
||||
peer_1_started, _ = peer_1.start()
|
||||
peer_2_started, _ = 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
|
||||
|
|
Loading…
Reference in a new issue