def test_prepare_run_of_already_failed_run_mock(self, mock_resolve,
mock_prepare_run,
mock_start_run):
spec_run = MagicMock(cache=V1Cache(disable=True))
mock_resolve.return_value = (None, spec_run)
experiment = RunFactory(project=self.project,
user=self.user,
raw_content="test",
is_managed=True)
# We are patching the automatic call and executing prepare manually
runs_prepare(run_id=experiment.id)
experiment.refresh_from_db()
assert experiment.status == V1Statuses.COMPILED
assert mock_prepare_run.call_count == 1 # Automatic call from executor
assert mock_start_run.call_count == 1
def test_prepare_run_of_already_skipped_run(self, mock_resolve):
spec_run = MagicMock(cache=V1Cache(disable=False))
mock_resolve.return_value = (None, spec_run)
experiment = RunFactory(project=self.project, user=self.user)
new_run_status(
run=experiment,
condition=V1StatusCondition.get_condition(type=V1Statuses.SKIPPED,
status=True),
)
new_experiment = RunFactory(project=self.project, user=self.user)
runs_prepare(run_id=new_experiment.id)
new_experiment.refresh_from_db()
assert new_experiment.status == V1Statuses.COMPILED
def test_cache_config(self):
config_dict = {"disable": True}
assert config_dict == V1Cache.from_dict(config_dict).to_dict()
config_dict = {"disable": True, "ttl": 12, "inputs": ["in1", "in2"]}
assert config_dict == V1Cache.from_dict(config_dict).to_dict()