Skip to content

Commit 551561e

Browse files
committed
fix existing pytests
1 parent bc4bc88 commit 551561e

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

conftest.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
from jupyter_scheduler.orm import create_session, create_tables
77
from jupyter_scheduler.scheduler import Scheduler
8-
from jupyter_scheduler.tests.mocks import MockEnvironmentManager
8+
from jupyter_scheduler.tests.mocks import MockDownloadManager, MockEnvironmentManager
99

1010
pytest_plugins = ("jupyter_server.pytest_plugin",)
1111

@@ -48,5 +48,8 @@ def jp_scheduler_db():
4848
@pytest.fixture
4949
def jp_scheduler():
5050
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),
5255
)

jupyter_scheduler/tests/mocks.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from typing import Dict, List
22

3+
from jupyter_scheduler.download_manager import DownloadManager
34
from jupyter_scheduler.environments import EnvironmentManager
45
from jupyter_scheduler.executors import ExecutionManager
56
from jupyter_scheduler.models import JobFeature, RuntimeEnvironment, UpdateJobDefinition
@@ -73,3 +74,8 @@ def pause_jobs(self, job_definition_id: str):
7374

7475
def resume_jobs(self, job_definition_id: str):
7576
pass
77+
78+
79+
class MockDownloadManager(DownloadManager):
80+
def __init__(self, db_url: str):
81+
pass

0 commit comments

Comments
 (0)