def test_removal(self): run = ContainerRun(id=42, state=ContainerRun.COMPLETE) expected_plan = {'ContainerRuns': {run}} plan = run.build_removal_plan() self.assertEqual(expected_plan, strip_removal_plan(plan))
def test_removal_skips_inputs(self): run = ContainerRun(id=42, state=ContainerRun.COMPLETE) dataset = Dataset(id=43) argument = ContainerArgument(type=ContainerArgument.INPUT) run.datasets.create(dataset=dataset, argument=argument) expected_plan = {'ContainerRuns': {run}} plan = run.build_removal_plan() self.assertEqual(expected_plan, strip_removal_plan(plan))
def test_remove_running(self): run = ContainerRun(id=42, state=ContainerRun.RUNNING) with self.assertRaisesRegex(ValueError, r'ContainerRun id 42 is still active.'): run.build_removal_plan()
def test_remove_running(self): run = ContainerRun(id=42, state=ContainerRun.RUNNING) with self.assertRaisesRegexp(ValueError, r'ContainerRun id 42 is still active.'): run.build_removal_plan()