raft_log5_test.py

import asyncio
import alog
from logging import INFO, ERROR
import raft_test
import secrets
import raft_election_test

NENTRIES = 5

async def check_entries(group,term,leader,entries,offset=0):
    # we only need to check leader logs b/c followers logs will be checked for
    # consistency during commitIndex checks

    log_good = False
    indices = list(range(offset+1, offset+1+len(entries)))
    if len(group.logs[leader]) != len(entries) + offset or not all(i in group.logs[leader] for i in indices):
        await alog.log(ERROR, f"### Expected leader log to have {offset+len(entries)} entries")
    elif not all(group.logs[leader][i][0] == term for i in indices):
        await alog.log(ERROR, f"### Expected leader log to have entries from term {term}")
    elif { group.logs[leader][i][1] for i in indices } != set(entries):
        await alog.log(ERROR, f"### Leader log contains incorrect entries")
    else:
        log_good = True

    if not log_good:
        await alog.log(ERROR, f"### Leader log: {group.logs[leader]}")
        raise RuntimeError("Leader log error")

async def main(n, group):
    term, leader = await raft_election_test.elect_leader(n, group)

    entries = [ secrets.token_urlsafe() for _ in range(NENTRIES) ]
    await alog.log(INFO, f"# Logging {entries}, waiting for it to be logged and committed")
    for entry in entries:
        group.processes[leader].log_entry(entry)
        await asyncio.sleep(0.5)

    def all_committed(group):
        return min(group.commitIndex.values()) == 5

    await group.wait_predicate(all_committed)

    await check_entries(group, term, leader, entries)
    await alog.log(INFO, f"### Log5 test passed")


if __name__ == "__main__":
    asyncio.run(raft_test.run_test(main))