commit c3e8d0c802a0fcd4a101646874121d16135d9279
author Alex Klein <saklein@chromium.org> Fri May 15 11:20:22 2020 -0600
committer Commit Bot <commit-bot@chromium.org> Mon May 18 17:31:18 2020 +0000
tree 2a3f27fd7438eab18eadb615d8249f1f49a66900
parent 7a7c27ed2eaba1414a28162355e6bb268198f35a

artifacts: Fix pinned image endpoint.

BUG=None
TEST=cq

Change-Id: I0ebd7388c82d60951faa488b290155d407f6d276
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/2204577
Tested-by: Alex Klein <saklein@chromium.org>
Reviewed-by: Michael Mortensen <mmortensen@google.com>
Commit-Queue: Alex Klein <saklein@chromium.org>

api/controller/artifacts_unittest.py

diff --git a/api/controller/artifacts_unittest.py b/api/controller/artifacts_unittest.py
index 5f44e13..e231870 100644
--- a/api/controller/artifacts_unittest.py
+++ b/api/controller/artifacts_unittest.py
@@ -406,8 +406,9 @@
     self.assertFalse(self.response.artifacts)
 
 
-class FetchPinnedGuestImagesTest(cros_test_lib.MockTempDirTestCase,
-                                 api_config.ApiConfigMixin, BundleRequestMixin):
+class FetchPinnedGuestImageUrisTest(cros_test_lib.MockTempDirTestCase,
+                                    api_config.ApiConfigMixin,
+                                    BundleRequestMixin):
   """Unittests for FetchPinnedGuestImages."""
 
   def setUp(self):
@@ -427,15 +428,15 @@
   def testValidateOnly(self):
     """Sanity check that a validate only call does not execute any logic."""
     patch = self.PatchObject(artifacts_svc, 'FetchPinnedGuestImages')
-    artifacts.FetchPinnedGuestImages(self.input_request, self.response,
-                                     self.validate_only_config)
+    artifacts.FetchPinnedGuestImageUris(self.input_request, self.response,
+                                        self.validate_only_config)
     patch.assert_not_called()
 
   def testMockCall(self):
     """Test that a mock call does not execute logic, returns mocked value."""
     patch = self.PatchObject(artifacts_svc, 'FetchPinnedGuestImages')
-    artifacts.FetchPinnedGuestImages(self.input_request, self.response,
-                                     self.mock_call_config)
+    artifacts.FetchPinnedGuestImageUris(self.input_request, self.response,
+                                        self.mock_call_config)
     patch.assert_not_called()
     self.assertEqual(len(self.response.pinned_images), 1)
     self.assertEqual(self.response.pinned_images[0].filename,
@@ -450,8 +451,8 @@
         filename='my_pinned_file.tar.gz', uri='https://the_testuri.com'))
     self.PatchObject(artifacts_svc, 'FetchPinnedGuestImages',
                      return_value=pins)
-    artifacts.FetchPinnedGuestImages(self.input_request, self.response,
-                                     self.api_config)
+    artifacts.FetchPinnedGuestImageUris(self.input_request, self.response,
+                                        self.api_config)
     self.assertEqual(len(self.response.pinned_images), 1)
     self.assertEqual(self.response.pinned_images[0].filename,
                      'my_pinned_file.tar.gz')