Python AmqpTransportの例

コード例 #1

ファイル: test_amqp_transport.py プロジェクト: xdmiodz/tomodachi

def test_routing_key(monkeypatch: Any) -> None:
    routing_key = AmqpTransport.get_routing_key("test.topic", {})
    assert routing_key == "test.topic"

    routing_key = AmqpTransport.get_routing_key(
        "test.topic", {"options": {
            "amqp": {
                "routing_key_prefix": "prefix-"
            }
        }})
    assert routing_key == "prefix-test.topic"

コード例 #2

ファイル: test_amqp_transport.py プロジェクト: xdmiodz/tomodachi

def test_queue_name(monkeypatch: Any) -> None:
    _uuid = "5d0b530f-5c44-4981-b01f-342801bd48f5"
    queue_name = AmqpTransport.get_queue_name("test.topic", "func", _uuid,
                                              False, {})
    assert queue_name == "b444917b9b922e8c29235737c7775c823e092c2374d1bfde071d42c637e3b4fd"

    queue_name = AmqpTransport.get_queue_name("test.topic", "func2", _uuid,
                                              False, {})
    assert queue_name != "b444917b9b922e8c29235737c7775c823e092c2374d1bfde071d42c637e3b4fd"

    queue_name = AmqpTransport.get_queue_name(
        "test.topic", "func", _uuid, False,
        {"options": {
            "amqp": {
                "queue_name_prefix": "prefix-"
            }
        }})
    assert queue_name == "prefix-b444917b9b922e8c29235737c7775c823e092c2374d1bfde071d42c637e3b4fd"

    queue_name = AmqpTransport.get_queue_name("test.topic", "func", _uuid,
                                              True, {})
    assert queue_name == "540e8e5bc604e4ea618f7e0517a04f030ad1dcbff2e121e9466ddd1c811450bf"

    queue_name = AmqpTransport.get_queue_name("test.topic", "func2", _uuid,
                                              True, {})
    assert queue_name == "540e8e5bc604e4ea618f7e0517a04f030ad1dcbff2e121e9466ddd1c811450bf"

    queue_name = AmqpTransport.get_queue_name(
        "test.topic", "func", _uuid, True,
        {"options": {
            "amqp": {
                "queue_name_prefix": "prefix-"
            }
        }})
    assert queue_name == "prefix-540e8e5bc604e4ea618f7e0517a04f030ad1dcbff2e121e9466ddd1c811450bf"

コード例 #3

ファイル: test_amqp_transport.py プロジェクト: zhengyb/tomodachi

def test_publish_invalid_credentials(monkeypatch: Any, capsys: Any, loop: Any) -> None:
    services, future = start_service('tests/services/dummy_service.py', monkeypatch)

    instance = services.get('test_dummy')

    with pytest.raises(AmqpException):
        loop.run_until_complete(AmqpTransport.publish(instance, 'data', 'test.topic', wait=True))

    os.kill(os.getpid(), signal.SIGINT)
    loop.run_until_complete(future)

    out, err = capsys.readouterr()
    assert 'Unable to connect [amqp] to 127.0.0.1:54321' in err
    assert out == ''

コード例 #4

ファイル: test_amqp_transport.py プロジェクト: xdmiodz/tomodachi

def test_publish_invalid_credentials(monkeypatch: Any, capsys: Any,
                                     loop: Any) -> None:
    services, future = start_service("tests/services/dummy_service.py",
                                     monkeypatch)

    instance = services.get("test_dummy")

    with pytest.raises(AmqpException):
        loop.run_until_complete(
            AmqpTransport.publish(instance, "data", "test.topic", wait=True))

    async def _async_kill():
        os.kill(os.getpid(), signal.SIGINT)

    loop.create_task(_async_kill())
    loop.run_until_complete(future)

    out, err = capsys.readouterr()
    assert "Unable to connect [amqp] to 127.0.0.1:54321" in err
    assert out == ""

コード例 #5

ファイル: test_amqp_transport.py プロジェクト: zhengyb/tomodachi

def test_queue_name(monkeypatch: Any) -> None:
    _uuid = '5d0b530f-5c44-4981-b01f-342801bd48f5'
    queue_name = AmqpTransport.get_queue_name('test.topic', 'func', _uuid, False, {})
    assert queue_name == 'b444917b9b922e8c29235737c7775c823e092c2374d1bfde071d42c637e3b4fd'

    queue_name = AmqpTransport.get_queue_name('test.topic', 'func2', _uuid, False, {})
    assert queue_name != 'b444917b9b922e8c29235737c7775c823e092c2374d1bfde071d42c637e3b4fd'

    queue_name = AmqpTransport.get_queue_name('test.topic', 'func', _uuid, False, {'options': {'amqp': {'queue_name_prefix': 'prefix-'}}})
    assert queue_name == 'prefix-b444917b9b922e8c29235737c7775c823e092c2374d1bfde071d42c637e3b4fd'

    queue_name = AmqpTransport.get_queue_name('test.topic', 'func', _uuid, True, {})
    assert queue_name == '540e8e5bc604e4ea618f7e0517a04f030ad1dcbff2e121e9466ddd1c811450bf'

    queue_name = AmqpTransport.get_queue_name('test.topic', 'func2', _uuid, True, {})
    assert queue_name == '540e8e5bc604e4ea618f7e0517a04f030ad1dcbff2e121e9466ddd1c811450bf'

    queue_name = AmqpTransport.get_queue_name('test.topic', 'func', _uuid, True, {'options': {'amqp': {'queue_name_prefix': 'prefix-'}}})
    assert queue_name == 'prefix-540e8e5bc604e4ea618f7e0517a04f030ad1dcbff2e121e9466ddd1c811450bf'

コード例 #6

ファイル: test_amqp_transport.py プロジェクト: xdmiodz/tomodachi

def test_decode_routing_key(monkeypatch: Any) -> None:
    routing_key = AmqpTransport.decode_routing_key("test-topic")
    assert routing_key == "test-topic"

    routing_key = AmqpTransport.decode_routing_key("test.topic")
    assert routing_key == "test.topic"

コード例 #7

ファイル: test_amqp_transport.py プロジェクト: zhengyb/tomodachi

def test_routing_key(monkeypatch: Any) -> None:
    routing_key = AmqpTransport.get_routing_key('test.topic', {})
    assert routing_key == 'test.topic'

    routing_key = AmqpTransport.get_routing_key('test.topic', {'options': {'amqp': {'routing_key_prefix': 'prefix-'}}})
    assert routing_key == 'prefix-test.topic'

コード例 #8

ファイル: test_amqp_transport.py プロジェクト: zhengyb/tomodachi

def test_encode_routing_key(monkeypatch: Any) -> None:
    routing_key = AmqpTransport.encode_routing_key('test-topic')
    assert routing_key == 'test-topic'

    routing_key = AmqpTransport.encode_routing_key('test.topic')
    assert routing_key == 'test.topic'