return create_job_definition

Author: andrii-i

jupyter_scheduler/tests/test_scheduler.py

@@ -22,15 +22,23 @@
     "output_formats": ["ipynb"],
 }
 
+event_type_parameters = {"name": "type1", "parameters": {"param1": "value1"}}
 
-def test_create_job_definition(jp_scheduler):
+
+def create_job_definition(jp_scheduler, job_def_params):
     with patch("jupyter_scheduler.scheduler.fsspec") as mock_fsspec:
         with patch("jupyter_scheduler.scheduler.Scheduler.file_exists") as mock_file_exists:
             mock_file_exists.return_value = True
             job_definition_id = jp_scheduler.create_job_definition(
-                CreateJobDefinition(**test_job_def_params)
+                CreateJobDefinition(**job_def_params)
             )
 
+            return job_definition_id
+
+
+def test_create_job_definition(jp_scheduler):
+    job_definition_id = create_job_definition(jp_scheduler, test_job_def_params)
+
     with jp_scheduler.db_session() as session:
         definitions = session.query(JobDefinition).all()
         assert 1 == len(definitions)
@@ -43,9 +51,6 @@ def test_create_job_definition(jp_scheduler):
         assert [] == definition.on_events
 
 
-event_type_parameters = {"name": "type1", "parameters": {"param1": "value1"}}
-
-
 def test_create_job_definition_with_on_events(jp_scheduler):
     params_with_on_events = {
         **test_job_def_params,