Refactors test structure

test/unit/test_cleaner.py

@@ -0,0 +1,87 @@
+import random
+from os import urandom
+from uuid import uuid4
+from binascii import hexlify
+
+from pisa import logging
+from pisa.responder import Job
+from pisa.cleaner import Cleaner
+from pisa.appointment import Appointment
+
+CONFIRMATIONS = 6
+ITEMS = 10
+MAX_ITEMS = 100
+ITERATIONS = 1000
+
+
+def set_up_appointments(total_appointments):
+    appointments = dict()
+    locator_uuid_map = dict()
+
+    for _ in range(total_appointments):
+        uuid = uuid4().hex
+        locator = hexlify(urandom(64))
+
+        appointments[uuid] = Appointment(locator, None, None, None, None, None, None)
+        locator_uuid_map[locator] = [uuid]
+
+        # Each locator can have more than one uuid assigned to it. Do a coin toss to add multiple ones
+        while random.randint(0, 1):
+            uuid = uuid4().hex
+
+            appointments[uuid] = Appointment(locator, None, None, None, None, None, None)
+            locator_uuid_map[locator].append(uuid)
+
+    return appointments, locator_uuid_map
+
+
+def set_up_jobs(total_jobs):
+    jobs = dict()
+    tx_job_map = dict()
+
+    for _ in range(total_jobs):
+        uuid = uuid4().hex
+        txid = hexlify(urandom(64))
+
+        # Assign both justice_txid and dispute_txid the same id (it shouldn't matter)
+        jobs[uuid] = Job(txid, txid, None, None, None)
+        tx_job_map[txid] = [uuid]
+
+        # Each justice_txid can have more than one uuid assigned to it. Do a coin toss to add multiple ones
+        while random.randint(0, 1):
+            uuid = uuid4().hex
+
+            jobs[uuid] = Job(txid, txid, None, None, None)
+            tx_job_map[txid].append(uuid)
+
+    return jobs, tx_job_map
+
+
+def test_delete_expired_appointment():
+    appointments, locator_uuid_map = set_up_appointments(MAX_ITEMS)
+    expired_appointments = random.sample(list(appointments.keys()), k=ITEMS)
+
+    Cleaner.delete_expired_appointment(expired_appointments, appointments, locator_uuid_map)
+
+    assert not set(expired_appointments).issubset(appointments.keys())
+
+
+def test_delete_completed_jobs():
+    jobs, tx_job_map = set_up_jobs(MAX_ITEMS)
+    selected_jobs = random.sample(list(jobs.keys()), k=ITEMS)
+
+    completed_jobs = [(job, 6) for job in selected_jobs]
+
+    Cleaner.delete_completed_jobs(jobs, tx_job_map, completed_jobs, 0)
+
+    assert not set(completed_jobs).issubset(jobs.keys())
+
+
+logging.getLogger().disabled = True
+
+for _ in range(ITERATIONS):
+    test_delete_expired_appointment()
+
+for _ in range(ITERATIONS):
+    test_delete_completed_jobs()
+