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 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_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)
