Ejemplo n.º 1
0
def multi_hub_stress_test(stress=5, messages=1000):
    hubs = []
    for i in range(stress):
        h = Hub(timeout=TIMEOUT)
        hubs.append(h)
    report("Hubs created")
    for h1 in hubs:
        for h2 in hubs:
            h2.connect("Test", "127.0.0.1", h1.port)
    report("Hubs linked")
    s()
    for h in hubs:
        assert len(h.transports) == 2 * stress
    report("Hub transports verified")
    for h in hubs:
        for i in range(messages):
            h.write_all("Test{}".format(i), text)
    report("Writing finished")
    s()
    for h in hubs:
        for i in range(messages):
            assert h.get_all("Test{}".format(i)) == [text] * 2 * stress
    report("Hub messages verified")
    for h in hubs:
        h.close()
Ejemplo n.º 2
0
def multi_transports(num_conn=10, messages=10, bidirectional=False):
    h1 = Hub(timeout=TIMEOUT)
    h2 = Hub(timeout=TIMEOUT)
    for i in range(num_conn):
        h1.connect("Test{}".format(i), "127.0.0.1", h2.port)
    s()
    assert len(h1.transports) == len(h2.transports)
    for i in range(messages):
        h1.write_all("Test{}".format(i), text)
        if bidirectional:
            h2.write_all("Test{}".format(i), text)
    s()
    for i in range(messages):
        if bidirectional:
            assert h1.get_all("Test{}".format(i)) == [text] * num_conn
        assert h2.get_all("Test{}".format(i)) == [text] * num_conn
    h1.close()
    h2.close()
Ejemplo n.º 3
0
def stress_test(stress=5, messages=1000):
    h1 = Hub(timeout=TIMEOUT)
    h2 = Hub(timeout=TIMEOUT)
    for i in range(stress):
        if i % 2 == 0:
            h1.connect("Test{}".format(i), "127.0.0.1", h2.port)
        else:
            h2.connect("Test{}".format(i), "127.0.0.1", h1.port)
    s()
    assert len(h1.transports) == len(h2.transports)
    for i in range(messages):
        h1.write_all("Test{}".format(i), text)
        h2.write_all("Test{}".format(i), text)
    s()
    for i in range(messages):
        assert h1.get_all("Test{}".format(i)) == [text] * stress
        assert h2.get_all("Test{}".format(i)) == [text] * stress
    h1.close()
    h2.close()