gistfile1.txt

# -*- Mode: python -*-

import pprint

from twisted.trial import unittest
from twisted.internet import reactor, protocol

import server

class ConnectClient( protocol.DatagramProtocol ):
    def connect_to( self, address ):
        self.transport.write( 'foo', ( address.host, address.port ) )
        reactor.callLater( 1, reactor.stop )

class TestPeerManagement( unittest.TestCase ):

    def test_oneClient( self ):
        peers = server.PeerManagement()
        reactor.listenUDP( 0, peers )
        address = peers.transport.getHost()

        connect_client = ConnectClient()
        reactor.listenUDP( 0, connect_client )
        reactor.callWhenRunning( connect_client.connect_to, address )

        reactor.run()