diff --git a/src/network/tcp.py b/src/network/tcp.py index e06cf89f..5ebd6a21 100644 --- a/src/network/tcp.py +++ b/src/network/tcp.py @@ -381,14 +381,3 @@ class TCPServer(AdvancedDispatcher): TCPConnection(sock=sock)) except socket.error: pass - - -if __name__ == "__main__": - # initial fill - - for host in (("127.0.0.1", 8448),): - direct = TCPConnection(host) - while asyncore.socket_map: - print "loop, state = %s" % (direct.state) - asyncore.loop(timeout=10, count=1) - continue diff --git a/src/tests/core.py b/src/tests/core.py index ce8e7877..a323be83 100644 --- a/src/tests/core.py +++ b/src/tests/core.py @@ -14,6 +14,8 @@ import unittest import knownnodes import state from helper_msgcoding import MsgEncode, MsgDecode +from network import asyncore_pollchoose as asyncore +from network.tcp import TCPConnection from queues import excQueue knownnodes_file = os.path.join(state.appdata, 'knownnodes.dat') @@ -74,10 +76,21 @@ class TestCore(unittest.TestCase): MsgEncode({'body': 'A msg with no subject'}, 3) except Exception as e: self.fail( - 'Exception %s whyle trying to encode message' + 'Exception %s while trying to encode message' ' with no subject!' % e ) + def test_tcpconnection(self): + """initial fill script from network.tcp""" + try: + for peer in (state.Peer("127.0.0.1", 8448),): + direct = TCPConnection(peer) + while asyncore.socket_map: + print("loop, state = %s" % direct.state) + asyncore.loop(timeout=10, count=1) + except: + self.fail('Exception in test loop') + def _wipe_knownnodes(self): with knownnodes.knownNodesLock: knownnodes.knownNodes = {stream: {} for stream in range(1, 4)}