Ejemplo n.º 1
0
def test_exception_status_code_is_set(server, ThriftTest, server_ttypes):

    # Given this test server:

    @server.thrift.register(ThriftTest)
    def testException(request):
        raise server_ttypes.Xception(
            errorCode=1001,
            message=request.body.arg
        )

    # Make a call:

    conn = yield TornadoConnection.outgoing(server.hostport)
    res = yield conn.send(
        CallRequestMessage(
            service=b'service',
            headers={b'cn': b'client', b'as': b'thrift'},
            args=[
                b'ThriftTest::testException',
                b'',
                bytearray([
                    0x0B,        # type = string
                    0x00, 0x01,  # field ID 1

                    0x00, 0x00, 0x00, 0x00,  # empty string

                    0x00,  # STOP
                ]),
            ],
        )
    )

    assert 1 == res.status_code
Ejemplo n.º 2
0
def test_exception_status_code_is_set(server, ThriftTest, server_ttypes):

    # Given this test server:

    @server.thrift.register(ThriftTest)
    def testException(request):
        raise server_ttypes.Xception(
            errorCode=1001,
            message=request.body.arg
        )

    # Make a call:

    conn = yield TornadoConnection.outgoing(server.hostport)
    res = yield conn.send(
        CallRequestMessage(
            service=b'service',
            headers={b'cn': b'client', b'as': b'thrift'},
            args=[
                b'ThriftTest::testException',
                b'',
                bytearray([
                    0x0B,        # type = string
                    0x00, 0x01,  # field ID 1

                    0x00, 0x00, 0x00, 0x00,  # empty string

                    0x00,  # STOP
                ]),
            ],
        )
    )

    assert 1 == res.status_code
Ejemplo n.º 3
0
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)
Ejemplo n.º 4
0
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)
Ejemplo n.º 5
0
def test_init_req_header():
    with mock.patch.object(
            TornadoConnection,
            '_extract_handshake_headers',
            side_effect=verify_init_header,
    ) as mock_extract:
        mock_extract.return_value = None
        server = TChannel('test_server')
        server.listen()
        yield TornadoConnection.outgoing(server.hostport)
Ejemplo n.º 6
0
def test_init_req_header():
    with mock.patch.object(
        TornadoConnection,
        '_extract_handshake_headers',
        side_effect=verify_init_header,
    ) as mock_extract:
        mock_extract.return_value = None
        server = TChannel('test_server')
        server.listen()
        yield TornadoConnection.outgoing(server.hostport)
Ejemplo n.º 7
0
def test_outbound_pending_change():
    server = TChannel('server')
    server.listen()
    connection = yield TornadoConnection.outgoing(server.hostport)
    c = [0]

    def outbound_pending_change_callback():
        c[0] += 1

    connection.set_outbound_pending_change_callback(
        outbound_pending_change_callback)

    connection.add_pending_outbound()
    assert c[0] == 1
    connection.add_pending_outbound()
    assert c[0] == 2
    connection.remove_pending_outbound()
    assert c[0] == 3
    connection.remove_pending_outbound()
    assert c[0] == 4
Ejemplo n.º 8
0
def test_outbound_pending_change_propagate(peer):
    server = TChannel('server')
    server.listen()
    connection = yield TornadoConnection.outgoing(server.hostport)

    peer.register_incoming_conn(connection)
    b = [0]

    def conn_change_db(peer):
        b[0] += 1

    peer.set_on_conn_change_callback(conn_change_db)

    connection.add_pending_outbound()
    assert b[0] == 1
    connection.add_pending_outbound()
    assert b[0] == 2
    connection.remove_pending_outbound()
    assert b[0] == 3
    connection.remove_pending_outbound()
    assert b[0] == 4
Ejemplo n.º 9
0
def test_outbound_pending_change_propagate(peer):
    server = TChannel('server')
    server.listen()
    connection = yield TornadoConnection.outgoing(server.hostport)

    peer.register_incoming_conn(connection)
    b = [0]

    def conn_change_db(peer):
        b[0] += 1

    peer.set_on_conn_change_callback(conn_change_db)

    connection.add_pending_outbound()
    assert b[0] == 1
    connection.add_pending_outbound()
    assert b[0] == 2
    connection.remove_pending_outbound()
    assert b[0] == 3
    connection.remove_pending_outbound()
    assert b[0] == 4
Ejemplo n.º 10
0
def test_outbound_pending_change():
    server = TChannel('server')
    server.listen()
    connection = yield TornadoConnection.outgoing(server.hostport)
    c = [0]

    def outbound_pending_change_callback():
        c[0] += 1

    connection.set_outbound_pending_change_callback(
        outbound_pending_change_callback
    )

    connection.add_pending_outbound()
    assert c[0] == 1
    connection.add_pending_outbound()
    assert c[0] == 2
    connection.remove_pending_outbound()
    assert c[0] == 3
    connection.remove_pending_outbound()
    assert c[0] == 4