fix pytests

Author: andrii-i

conftest.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+from unittest.mock import AsyncMock, Mock
 
 import pytest
 from sqlalchemy import create_engine
@@ -59,6 +60,7 @@ def jp_scheduler(jp_scheduler_db_url, jp_scheduler_root_dir, jp_scheduler_db):
         db_url=jp_scheduler_db_url,
         root_dir=str(jp_scheduler_root_dir),
         environments_manager=MockEnvironmentManager(),
+        dask_client_future=AsyncMock(),
     )
 
 

jupyter_scheduler/tests/test_job_files_manager.py

@@ -4,7 +4,7 @@
 import tarfile
 import time
 from pathlib import Path
-from unittest.mock import patch
+from unittest.mock import AsyncMock, patch
 
 import pytest
 
@@ -41,13 +41,16 @@ async def test_copy_from_staging():
     }
     output_dir = "jobs/1"
     with patch("jupyter_scheduler.job_files_manager.Downloader") as mock_downloader:
-        with patch("jupyter_scheduler.job_files_manager.Process") as mock_process:
+        with patch("jupyter_scheduler.scheduler.DaskClient") as mock_dask_future:
             with patch("jupyter_scheduler.scheduler.Scheduler") as mock_scheduler:
+                mock_downloader.return_value.download = AsyncMock()
                 mock_scheduler.get_job.return_value = job
                 mock_scheduler.get_staging_paths.return_value = staging_paths
                 mock_scheduler.get_local_output_path.return_value = output_dir
                 mock_scheduler.get_job_filenames.return_value = job_filenames
-                manager = JobFilesManager(scheduler=mock_scheduler)
+                manager = JobFilesManager(
+                    scheduler=mock_scheduler, dask_client_future=mock_dask_future
+                )
                 await manager.copy_from_staging(1)
 
                 mock_downloader.assert_called_once_with(