File tree Expand file tree Collapse file tree 2 files changed +11
-2
lines changed Expand file tree Collapse file tree 2 files changed +11
-2
lines changed Original file line number Diff line number Diff line change 5
5
6
6
from jupyter_scheduler .orm import create_session , create_tables
7
7
from jupyter_scheduler .scheduler import Scheduler
8
- from jupyter_scheduler .tests .mocks import MockEnvironmentManager
8
+ from jupyter_scheduler .tests .mocks import MockDownloadManager , MockEnvironmentManager
9
9
10
10
pytest_plugins = ("jupyter_server.pytest_plugin" ,)
11
11
@@ -48,5 +48,8 @@ def jp_scheduler_db():
48
48
@pytest .fixture
49
49
def jp_scheduler ():
50
50
return Scheduler (
51
- db_url = DB_URL , root_dir = str (TEST_ROOT_DIR ), environments_manager = MockEnvironmentManager ()
51
+ db_url = DB_URL ,
52
+ root_dir = (TEST_ROOT_DIR ),
53
+ environments_manager = MockEnvironmentManager (),
54
+ download_manager = MockDownloadManager (DB_URL ),
52
55
)
Original file line number Diff line number Diff line change 1
1
from typing import Dict , List
2
2
3
+ from jupyter_scheduler .download_manager import DownloadManager
3
4
from jupyter_scheduler .environments import EnvironmentManager
4
5
from jupyter_scheduler .executors import ExecutionManager
5
6
from jupyter_scheduler .models import JobFeature , RuntimeEnvironment , UpdateJobDefinition
@@ -73,3 +74,8 @@ def pause_jobs(self, job_definition_id: str):
73
74
74
75
def resume_jobs (self , job_definition_id : str ):
75
76
pass
77
+
78
+
79
+ class MockDownloadManager (DownloadManager ):
80
+ def __init__ (self , db_url : str ):
81
+ pass
You can’t perform that action at this time.
0 commit comments