X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/ffb304afad93f50fd6ee43ecda6584dd0ac000c4..b9fd7e3f374248a61159e4750a84e38d1c48d5dd:/sdk/python/tests/test_retry_job_helpers.py diff --git a/sdk/python/tests/test_retry_job_helpers.py b/sdk/python/tests/test_retry_job_helpers.py index e31a9a0513..76c62cb0ce 100644 --- a/sdk/python/tests/test_retry_job_helpers.py +++ b/sdk/python/tests/test_retry_job_helpers.py @@ -1,3 +1,7 @@ +# Copyright (C) The Arvados Authors. All rights reserved. +# +# SPDX-License-Identifier: Apache-2.0 + from __future__ import absolute_import from builtins import object import mock @@ -72,34 +76,3 @@ class CurrentJobTestCase(ApiClientRetryTestMixin, unittest.TestCase): def run_method(self): arvados.current_job() - - -class CurrentTaskTestCase(ApiClientRetryTestMixin, unittest.TestCase): - - DEFAULT_EXCEPTION = arvados.errors.ApiError - - def setUp(self): - super(CurrentTaskTestCase, self).setUp() - os.environ['TASK_UUID'] = 'zzzzz-zzzzz-zzzzzzzzzzzzzzz' - os.environ['TASK_WORK'] = '.' - - def tearDown(self): - del os.environ['TASK_UUID'] - del os.environ['TASK_WORK'] - arvados._current_task = None - super(CurrentTaskTestCase, self).tearDown() - - def run_method(self): - arvados.current_task() - - -class TaskSetOutputTestCase(CurrentTaskTestCase, unittest.TestCase): - - DEFAULT_EXCEPTION = arvados.errors.ApiError - - def tearDown(self): - super(TaskSetOutputTestCase, self).tearDown() - run_test_server.reset() - - def run_method(self, locator=ApiClientRetryTestMixin.TEST_LOCATOR): - arvados.task_set_output({'uuid':self.TEST_UUID},s=locator)