test_job_list.py 841 B

123456789101112131415161718192021222324
  1. from time import sleep
  2. import pytest
  3. from facefusion.jobs.job_list import compose_job_list
  4. from facefusion.jobs.job_manager import clear_jobs, create_job, init_jobs
  5. from .helper import get_test_jobs_directory
  6. @pytest.fixture(scope = 'function', autouse = True)
  7. def before_each() -> None:
  8. clear_jobs(get_test_jobs_directory())
  9. init_jobs(get_test_jobs_directory())
  10. def test_compose_job_list() -> None:
  11. create_job('job-test-compose-job-list-1')
  12. sleep(0.5)
  13. create_job('job-test-compose-job-list-2')
  14. job_headers, job_contents = compose_job_list('drafted')
  15. assert job_headers == [ 'job id', 'steps', 'date created', 'date updated', 'job status' ]
  16. assert job_contents[0] == [ 'job-test-compose-job-list-1', 0, 'just now', None, 'drafted' ]
  17. assert job_contents[1] == [ 'job-test-compose-job-list-2', 0, 'just now', None, 'drafted' ]