3
3
4
4
import pytest
5
5
6
+ from alfalfa_worker .jobs .step_run_base import ClockSource
6
7
from alfalfa_worker .lib .enums import RunStatus , SimType
7
8
from alfalfa_worker .lib .job import JobStatus
8
9
from tests .worker .lib .mock_dispatcher import MockDispatcher
@@ -33,14 +34,14 @@ def test_simple_internal_clock(mock_dispatcher: MockDispatcher, model_path: Path
33
34
"start_datetime" : str (datetime .datetime (2019 , 1 , 2 , 0 , 0 , 0 )),
34
35
"end_datetime" : str (datetime .datetime (2019 , 1 , 3 , 0 , 0 , 0 )),
35
36
"timescale" : "5" ,
36
- "realtime" : None
37
+ "realtime" : False
37
38
}
38
39
39
40
if run .sim_type == SimType .OPENSTUDIO :
40
41
step_run_job = mock_dispatcher .start_job ("alfalfa_worker.jobs.openstudio.step_run.StepRun" , params )
41
42
else :
42
43
step_run_job = mock_dispatcher .start_job ("alfalfa_worker.jobs.modelica.step_run.StepRun" , params )
43
- assert step_run_job .step_sim_type == "timescale"
44
+ assert step_run_job .options . clock_source == ClockSource . INTERNAL
44
45
run = step_run_job .run
45
46
wait_for_job_status (step_run_job , JobStatus .RUNNING )
46
47
wait_for_run_status (run , RunStatus .RUNNING )
@@ -68,7 +69,7 @@ def test_simple_external_clock(mock_dispatcher: MockDispatcher, model_path: Path
68
69
"start_datetime" : str (start_dt ),
69
70
"end_datetime" : str (datetime .datetime (2019 , 1 , 3 , 0 , 0 , 0 )),
70
71
"timescale" : "1" ,
71
- "realtime" : None
72
+ "realtime" : False
72
73
}
73
74
74
75
if run .sim_type == SimType .OPENSTUDIO :
@@ -77,7 +78,7 @@ def test_simple_external_clock(mock_dispatcher: MockDispatcher, model_path: Path
77
78
step_run_job = mock_dispatcher .start_job ("alfalfa_worker.jobs.modelica.step_run.StepRun" , params )
78
79
run = step_run_job .run
79
80
80
- wait_for_run_status (run , RunStatus .RUNNING , timeout = 30 )
81
+ wait_for_run_status (run , RunStatus .RUNNING , timeout = 60 )
81
82
wait_for_job_status (step_run_job , JobStatus .WAITING )
82
83
83
84
# -- Assert model gets to expected start time
0 commit comments