def test_get_rank_with_imcoming(): server = TChannel('server') server.listen() connection = yield TornadoConnection.outgoing(server.hostport) connection.direction = INCOMING peer = Peer(TChannel('test'), '10.10.101.21:230') calculator = PreferIncomingCalculator() peer.register_incoming_conn(connection) assert sys.maxint != calculator.get_rank(peer)
def test_get_rank_with_outgoing(): server = TChannel('server') server.listen() connection = yield TornadoConnection.outgoing(server.hostport) peer = Peer(TChannel('test'), '10.10.101.21:230') calculator = PreferIncomingCalculator() peer.register_outgoing_conn(connection) assert PreferIncomingCalculator.TIERS[1] == calculator.get_rank(peer)
def peer(): return Peer( tchannel=TChannel('peer'), hostport='127.0.0.1:21300', )
def test_get_rank_no_connection(): server = TChannel('server') server.listen() peer = Peer(TChannel('test'), '10.10.101.21:230') calculator = PreferIncomingCalculator() assert sys.maxint == calculator.get_rank(peer)
def peer(tchannel): return Peer(tchannel, "localhost:4040")