fix existing pytests

Author: andrii-i

conftest.py

@@ -5,7 +5,7 @@
 
 from jupyter_scheduler.orm import create_session, create_tables
 from jupyter_scheduler.scheduler import Scheduler
-from jupyter_scheduler.tests.mocks import MockEnvironmentManager
+from jupyter_scheduler.tests.mocks import MockDownloadManager, MockEnvironmentManager
 
 pytest_plugins = ("jupyter_server.pytest_plugin",)
 
@@ -46,5 +46,8 @@ def jp_scheduler_db():
 @pytest.fixture
 def jp_scheduler(jp_data_dir):
     return Scheduler(
-        db_url=DB_URL, root_dir=str(jp_data_dir), environments_manager=MockEnvironmentManager()
+        db_url=DB_URL,
+        root_dir=str(jp_data_dir),
+        environments_manager=MockEnvironmentManager(),
+        download_manager=MockDownloadManager(DB_URL),
     )

jupyter_scheduler/tests/mocks.py

@@ -1,5 +1,6 @@
 from typing import Dict, List
 
+from jupyter_scheduler.download_manager import DownloadManager
 from jupyter_scheduler.environments import EnvironmentManager
 from jupyter_scheduler.executors import ExecutionManager
 from jupyter_scheduler.models import JobFeature, RuntimeEnvironment, UpdateJobDefinition
@@ -73,3 +74,8 @@ def pause_jobs(self, job_definition_id: str):
 
     def resume_jobs(self, job_definition_id: str):
         pass
+
+
+class MockDownloadManager(DownloadManager):
+    def __init__(self, db_url: str):
+        pass

jupyter_scheduler/tests/test_job_files_manager.py

@@ -56,6 +56,7 @@ async def test_copy_from_staging():
                     staging_paths=staging_paths,
                     output_dir=output_dir,
                     redownload=False,
+                    include_staging_files=None,
                 )
 
 

jupyter_scheduler/tests/test_scheduler.py

@@ -49,6 +49,7 @@ def test_create_job_definition(jp_scheduler):
     "update_time": 1,
     "create_time": 1,
     "active": True,
+    "packaged_files": [],
 }
 
 job_definition_2 = {
@@ -62,6 +63,7 @@ def test_create_job_definition(jp_scheduler):
     "create_time": 2,
     "active": True,
     "tags": ["tag_2"],
+    "packaged_files": [],
 }
 
 job_definition_3 = {
@@ -75,6 +77,7 @@ def test_create_job_definition(jp_scheduler):
     "create_time": 3,
     "active": False,
     "tags": ["tag_3"],
+    "packaged_files": [],
 }