Exemple #1
0
def update_progress():
    print('updating progress...')
    for idx, conn in children.items():
        if conn:
            send_msg(conn, b'ready')
            data = pickle.loads(recv_msg(conn))
            print(data)
            progress[idx].update(data)
    print('progress updated!')
Exemple #2
0
def get_progress_thread(conn, idx):
    print('thread created')
    send_msg(conn, b'ready')

    while conn:
        recv_msg(conn)
        print('sending data')
        update_progress()
        send_msg(conn, pickle.dumps(progress))

    print('closing connection')
    conn.close()
Exemple #3
0
def test_unsplit_messaging():
    """ Test if the message sending and receiving work as expected.

    Test for unsplit messages. (msg < BUFFER_SIZE)
    """

    msg = ('test-msg', 'test-data')

    send_msg(msg[0], msg[1], ('127.0.0.1', 6667))

    received = RECEIVER.receive_msg()

    assert unpack_msg(received[0]) == (msg[0], msg[1])
Exemple #4
0
def client_thread(conn, idx):
    print('thread created')
    progress = {}
    all_progress[idx] = progress

    while conn:
        data = conn.recv(1024)
        if not data:
            break

        data = pickle.loads(data)
        progress.update(data)
        print(all_progress)
        send_msg(conn, b'good')
        sleep(.1)

    print('closing connection')
    conn.close()
Exemple #5
0
def test_split_messaging_2():
    """ Test if the message sending and receiving work as expected.

    Test for split (2-parts) messages. (msg > BUFFER_SIZE)
    """

    msg = ('test-msg', b'1' * (SERVER.buffersize + 10))

    send_msg(msg[0], msg[1], ('127.0.0.1', 6667))

    received = RECEIVER.receive_msg()

    assert received is None

    received = RECEIVER.receive_msg()

    # Check content
    assert unpack_msg(received[0]) == (msg[0], msg[1])