test_pick_winner.py

#!/usr/bin/env python3

import asyncio

from p2p import peer


def test_pick_winner_ignores_round_without_eligible_validator():
    async def scenario():
        original_sleep = peer.asyncio.sleep
        real_sleep = asyncio.sleep
        try:
            async def fast_sleep(_):
                await real_sleep(0)

            peer.asyncio.sleep = fast_sleep
            peer.validators.clear()
            peer.tempblocks.clear()
            peer.Blockchain.clear()
            queue = asyncio.Queue()
            peer.tempblocks.append(
                {
                    "Validator": "missing",
                    "Index": 1,
                    "BPM": 60,
                    "Timestamp": "t",
                    "PrevHash": "h",
                    "Hash": "h2",
                }
            )

            task = asyncio.create_task(peer.pick_winner(queue))
            for _ in range(100):
                if peer.tempblocks == []:
                    break
                await real_sleep(0.001)

            assert not task.done()
            assert queue.empty()
            assert peer.tempblocks == []

            task.cancel()
            await asyncio.gather(task, return_exceptions=True)
        finally:
            peer.asyncio.sleep = original_sleep
            peer.validators.clear()
            peer.tempblocks.clear()
            peer.Blockchain.clear()

    asyncio.run(scenario())