From 11ba8237405d2424bd08da61be75e5ff73883f18 Mon Sep 17 00:00:00 2001 From: Tom Clegg Date: Fri, 9 Dec 2022 02:09:06 -0500 Subject: [PATCH] 19840: Cleanup link added during test. Arvados-DCO-1.1-Signed-off-by: Tom Clegg --- sdk/python/tests/test_arv_keepdocker.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/sdk/python/tests/test_arv_keepdocker.py b/sdk/python/tests/test_arv_keepdocker.py index 8dea29a44c..526fd68727 100644 --- a/sdk/python/tests/test_arv_keepdocker.py +++ b/sdk/python/tests/test_arv_keepdocker.py @@ -168,15 +168,19 @@ class ArvKeepdockerTestCase(unittest.TestCase, tutil.VersionChecker): find_image_mock.assert_called_with('[::1]:8888/repo/img', 'tag') def test_list_images_with_host_and_port(self): - arvados.api('v1').links().create(body={'link': { + api = arvados.api('v1') + taglink = api.links().create(body={'link': { 'link_class': 'docker_image_repo+tag', 'name': 'registry.example:1234/repo:latest', 'head_uuid': 'zzzzz-4zz18-1v45jub259sjjgb', }}).execute() - out = tutil.StringIO() - with self.assertRaises(SystemExit): - self.run_arv_keepdocker([], sys.stderr, stdout=out) - self.assertRegex(out.getvalue(), '\nregistry.example:1234/repo +latest ') + try: + out = tutil.StringIO() + with self.assertRaises(SystemExit): + self.run_arv_keepdocker([], sys.stderr, stdout=out) + self.assertRegex(out.getvalue(), '\nregistry.example:1234/repo +latest ') + finally: + api.links().delete(uuid=taglink['uuid']).execute() @mock.patch('arvados.commands.keepdocker.list_images_in_arv', return_value=[]) -- 2.30.2