X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/93d38f3cb3d7874ed3723409357d8dd5e8d618a5..093ec98e4a065acfc537ea22c08c337c115fe273:/sdk/python/tests/test_sdk.py?ds=sidebyside

diff --git a/sdk/python/tests/test_sdk.py b/sdk/python/tests/test_sdk.py
index d8b37d8a3c..41add57c0e 100644
--- a/sdk/python/tests/test_sdk.py
+++ b/sdk/python/tests/test_sdk.py
@@ -1,3 +1,7 @@
+# Copyright (C) The Arvados Authors. All rights reserved.
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import mock
 import os
 import unittest
@@ -7,10 +11,11 @@ import arvados.collection
 
 class TestSDK(unittest.TestCase):
 
-    @mock.patch('arvados.apisetup.api_from_config')
     @mock.patch('arvados.current_task')
     @mock.patch('arvados.current_job')
-    def test_one_task_per_input_file_normalize(self, mock_job, mock_task, mock_api):
+    def test_one_task_per_input_file_normalize(self, mock_job, mock_task):
+        mock_api = mock.MagicMock()
+
         # This manifest will be reduced from three lines to one when it is
         # normalized.
         nonnormalized_manifest = """. 5348b82a029fd9e971a811ce1f71360b+43 0:43:md5sum.txt
@@ -30,7 +35,7 @@ class TestSDK(unittest.TestCase):
             'sequence': 0,
         }
         # mock the API client to return a collection with a nonnormalized manifest.
-        mock_api('v1').collections().get().execute.return_value = {
+        mock_api.collections().get().execute.return_value = {
             'uuid': 'zzzzz-4zz18-mockcollection0',
             'portable_data_hash': dummy_hash,
             'manifest_text': nonnormalized_manifest,
@@ -38,5 +43,5 @@ class TestSDK(unittest.TestCase):
 
         # Because one_task_per_input_file normalizes this collection,
         # it should now create only one job task and not three.
-        arvados.job_setup.one_task_per_input_file(and_end_task=False)
-        mock_api('v1').job_tasks().create().execute.assert_called_once_with()
+        arvados.job_setup.one_task_per_input_file(and_end_task=False, api_client=mock_api)
+        mock_api.job_tasks().create().execute.assert_called_once_with()