mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 22:24:23 +01:00
87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
import random
|
|
from os import urandom
|
|
from uuid import uuid4
|
|
|
|
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 = urandom(32).hex()
|
|
|
|
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 = urandom(32).hex()
|
|
|
|
# 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()
|
|
|