74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
from app import config
|
|
from app.db import Session
|
|
from job_runner import get_jobs_to_run
|
|
from app.models import Job, JobState
|
|
import arrow
|
|
|
|
|
|
def test_get_jobs_to_run(flask_client):
|
|
now = arrow.now()
|
|
for job in Job.all():
|
|
Job.delete(job.id)
|
|
expected_jobs_to_run = [
|
|
# Jobs in ready state
|
|
Job.create(name="", payload=""),
|
|
Job.create(name="", payload="", run_at=now),
|
|
# Jobs in taken state
|
|
Job.create(
|
|
name="",
|
|
payload="",
|
|
state=JobState.taken.value,
|
|
taken_at=now.shift(minutes=-(config.JOB_TAKEN_RETRY_WAIT_MINS + 10)),
|
|
),
|
|
Job.create(
|
|
name="",
|
|
payload="",
|
|
state=JobState.taken.value,
|
|
taken_at=now.shift(minutes=-(config.JOB_TAKEN_RETRY_WAIT_MINS + 10)),
|
|
attempts=config.JOB_MAX_ATTEMPTS - 1,
|
|
),
|
|
Job.create(
|
|
name="",
|
|
payload="",
|
|
state=JobState.taken.value,
|
|
taken_at=now.shift(minutes=-(config.JOB_TAKEN_RETRY_WAIT_MINS + 10)),
|
|
run_at=now,
|
|
),
|
|
]
|
|
# Jobs not to run
|
|
# Job to run in the future
|
|
Job.create(name="", payload="", run_at=now.shift(hours=2))
|
|
# Job in done state
|
|
Job.create(name="", payload="", state=JobState.done.value)
|
|
# Job taken but not enough time has passed
|
|
Job.create(
|
|
name="",
|
|
payload="",
|
|
state=JobState.taken.value,
|
|
taken_at=now.shift(minutes=-(config.JOB_TAKEN_RETRY_WAIT_MINS - 10)),
|
|
)
|
|
# Job taken with enough time but out of run_at zone
|
|
Job.create(
|
|
name="",
|
|
payload="",
|
|
state=JobState.taken.value,
|
|
taken_at=now.shift(minutes=-(config.JOB_TAKEN_RETRY_WAIT_MINS + 10)),
|
|
run_at=now.shift(hours=3),
|
|
)
|
|
# Job out of attempts
|
|
(
|
|
Job.create(
|
|
name="",
|
|
payload="",
|
|
state=JobState.taken.value,
|
|
taken_at=now.shift(minutes=-(config.JOB_TAKEN_RETRY_WAIT_MINS + 10)),
|
|
attempts=config.JOB_MAX_ATTEMPTS + 1,
|
|
),
|
|
)
|
|
Session.commit()
|
|
jobs = get_jobs_to_run()
|
|
assert len(jobs) == len(expected_jobs_to_run)
|
|
job_ids = [job.id for job in jobs]
|
|
for job in expected_jobs_to_run:
|
|
assert job.id in job_ids
|