Example #1
0
File: tests.py Project: pmdz/smite
def test_default_handler():
    ipc_name = 'smite-test-{}'.format(uuid.uuid1().hex)
    default_handler = lambda t: t

    servant = Servant()
    servant.set_default_handler(default_handler)
    servant.bind_ipc(ipc_name)
    servant.run(True)

    msg_num = 10

    client = RClient('ipc://{}'.format(ipc_name))
    for _ in range(msg_num):
        msg_txt = uuid.uuid1().hex
        random_msg_name = uuid.uuid1().hex
        rep = client.send(random_msg_name, msg_txt)
        assert rep == msg_txt

    assert servant.stats['summary']['exceptions'] == 0
    assert servant.stats['summary']['malicious_messages'] == 0
    assert servant.stats['summary']['received_messages'] == msg_num
    assert servant.stats['summary']['processed_messages'] == msg_num

    client.close()
    servant.stop()
Example #2
0
File: tests.py Project: pmdz/smite
def test_rclient():
    ipc_name = 'smite-test-{}'.format(uuid.uuid1().hex)

    servant = Servant({'echo': lambda t: t})
    servant.bind_ipc(ipc_name)
    servant.run(True)

    msg_num = 10

    client = RClient('ipc://{}'.format(ipc_name))
    for _ in range(msg_num):
        echo_txt = uuid.uuid1().hex
        rep = client.send('echo', echo_txt)
        assert rep == echo_txt

    assert servant.stats['summary']['exceptions'] == 0
    assert servant.stats['summary']['malicious_messages'] == 0
    assert servant.stats['summary']['received_messages'] == msg_num
    assert servant.stats['summary']['processed_messages'] == msg_num

    client.close()
    servant.stop()