コード例 #1
0
def test_send_to_all():
    conns = [mock.Mock() for i in range(10)]
    c = Connections('test', 'dummy')
    for i in conns:
        c._add(i, id(i))

    c.send_to_all('foo')
    for i in conns:
        assert i.send_concurrent.called
        i.send_concurrent.assert_called_with(('test', 'dummy', 'foo'))
コード例 #2
0
def test_send_to_many():
    conns = [(i, mock.Mock()) for i in range(10)]
    c = Connections('test', 'dummy')
    for target, sock in conns:
        c._add(sock, target)

    targets = [2, 3, 5, 9]
    c.send(targets, 'foo')
    for i, sock in conns:
        assert sock.send_concurrent.called == (i in targets)
コード例 #3
0
def test_heartbeat():
    conns = [mock.Mock() for i in range(10)]
    c = Connections('test', 'dummy')
    for i in conns:
        c._add(i, id(i))

    c.send_heartbeat()
    for i in conns:
        assert i.send_concurrent.called
        i.send_concurrent.assert_called_with(('test', 'dummy', HEARTBEAT))