def test_run_fail(container_mock, get_image_mock): container_mock().is_success = mock.Mock(return_value=False) try: steps.run('docker_client', 'image_id', cmd=['cmd'], mount_point=steps.MOUNT_POINT) finally: assert_run(container_mock)
def test_run_ok(container_mock, get_image_mock): container_mock().is_success = mock.Mock(return_value=True) steps.run('docker_client', 'image_id', cmd=['cmd']) assert_run(container_mock)