Ejemplo n.º 1
0
async def test_worker_chat_not_started(event_loop):
    with mock.patch('blackhole.worker.Worker.start') as mock_start:
        worker = Worker('1', [])
    assert mock_start.called is True
    worker._started = False
    with mock.patch('blackhole.worker.Worker.start') as mock_chat_start, \
            mock.patch('blackhole.worker.Worker.stop') as mock_chat_stop:
        await worker.chat(None)
    assert mock_chat_start.called is False
    assert mock_chat_stop.called is False
Ejemplo n.º 2
0
async def test_worker_heartbeat_started_bad_ping(event_loop):
    with mock.patch('blackhole.worker.Worker.start') as mock_start:
        worker = Worker('1', [])
    assert mock_start.called is True
    worker._started = True
    worker.ping = time.monotonic() - 120
    with mock.patch('blackhole.worker.Worker.start') as mock_heartbeat_start, \
            mock.patch('blackhole.worker.Worker.stop') as mock_heartbeat_stop:
        await worker.heartbeat(None)
    assert mock_heartbeat_start.called is True
    assert mock_heartbeat_stop.called is True
Ejemplo n.º 3
0
async def test_worker_chat_started_restart(event_loop):
    with mock.patch('blackhole.worker.Worker.start') as mock_start:
        worker = Worker('1', [])
    assert mock_start.called is True
    worker._started = True
    with mock.patch('blackhole.worker.Worker.start') as mock_chat_start, \
        mock.patch('blackhole.worker.Worker.stop') as mock_chat_stop, \
            mock.patch('asyncio.StreamReader.read', side_effect=Exception):
        reader = asyncio.StreamReader()
        await worker.chat(reader)
    assert mock_chat_start.called is True
    assert mock_chat_stop.called is True
Ejemplo n.º 4
0
async def test_worker_heartbeat_started_good_ping(event_loop):
    with mock.patch('blackhole.worker.Worker.start') as mock_start:
        worker = Worker('1', [])
    assert mock_start.called is True
    worker._started = True
    worker.ping = time.monotonic() + 120
    writer = BytesIO()
    async def reset():
        await asyncio.sleep(20)
        worker._started = False
    reset_task = asyncio.Task(reset())
    await worker.heartbeat(writer)
    assert worker._started is False
    reset_task.cancel()
Ejemplo n.º 5
0
async def test_worker_chat_started_good_message(event_loop):
    with mock.patch('blackhole.worker.Worker.start') as mock_start:
        worker = Worker('1', [], loop=event_loop)
    assert mock_start.called is True
    worker._started = True
    worker.ping = 987654321
    reader = asyncio.StreamReader()
    async def reset():
        reader.feed_data(protocols.PONG)
        worker._started = False
    reset_task = asyncio.Task(reset())
    with mock.patch('time.monotonic', return_value=123456789):
        await worker.chat(reader)
    assert worker._started is False
    assert worker.ping == 123456789
    reset_task.cancel()