mirror of
https://github.com/simple-login/app.git
synced 2024-11-10 21:27:10 +01:00
0da1811311
* Cleanup tasks * Update * Added tests * Create cron job * Delete old data cron * Fix import * import fix * Added delete + script to disable pgp for proton mboxes
73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
import arrow
|
|
|
|
from app import config
|
|
from app.models import Job, JobState
|
|
from tasks.cleanup_old_jobs import cleanup_old_jobs
|
|
|
|
|
|
def test_cleanup_old_jobs():
|
|
Job.filter().delete()
|
|
now = arrow.now()
|
|
delete_ids = [
|
|
Job.create(
|
|
updated_at=now.shift(minutes=-1),
|
|
state=JobState.done.value,
|
|
name="",
|
|
payload="",
|
|
flush=True,
|
|
).id,
|
|
Job.create(
|
|
updated_at=now.shift(minutes=-1),
|
|
state=JobState.error.value,
|
|
name="",
|
|
payload="",
|
|
flush=True,
|
|
).id,
|
|
Job.create(
|
|
updated_at=now.shift(minutes=-1),
|
|
state=JobState.taken.value,
|
|
attempts=config.JOB_MAX_ATTEMPTS,
|
|
name="",
|
|
payload="",
|
|
flush=True,
|
|
).id,
|
|
]
|
|
|
|
keep_ids = [
|
|
Job.create(
|
|
updated_at=now.shift(minutes=+1),
|
|
state=JobState.done.value,
|
|
name="",
|
|
payload="",
|
|
flush=True,
|
|
).id,
|
|
Job.create(
|
|
updated_at=now.shift(minutes=+1),
|
|
state=JobState.error.value,
|
|
name="",
|
|
payload="",
|
|
flush=True,
|
|
).id,
|
|
Job.create(
|
|
updated_at=now.shift(minutes=+1),
|
|
state=JobState.taken.value,
|
|
attempts=config.JOB_MAX_ATTEMPTS,
|
|
name="",
|
|
payload="",
|
|
flush=True,
|
|
).id,
|
|
Job.create(
|
|
updated_at=now.shift(minutes=-1),
|
|
state=JobState.taken.value,
|
|
attempts=config.JOB_MAX_ATTEMPTS - 1,
|
|
name="",
|
|
payload="",
|
|
flush=True,
|
|
).id,
|
|
]
|
|
cleanup_old_jobs(now)
|
|
for delete_id in delete_ids:
|
|
assert Job.get(id=delete_id) is None
|
|
for keep_id in keep_ids:
|
|
assert Job.get(id=keep_id) is not None
|