commit 2677bf6cf3de02c9838d3d846abb06cbfd3c27c8
author Michael Mortensen <mmortensen@google.com> Tue Oct 29 08:31:25 2019 -0600
committer Commit Bot <commit-bot@chromium.org> Tue Oct 29 21:50:49 2019 +0000
tree ded2f7d94cb577f05b6b09487886f56d9d65a5ed
parent 417bd662d425c696ce6e6b67331024716c1fbdad

BuildAPI: Finish Package Mock responses.

BUG=chromium:1000854
TEST=run_tests

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

api/controller/packages.py

diff --git a/api/controller/packages.py b/api/controller/packages.py
index 158fa93..62d51f7 100644
--- a/api/controller/packages.py
+++ b/api/controller/packages.py
@@ -25,8 +25,13 @@
     binhost_pb2.OVERLAYTYPE_BOTH: constants.BOTH_OVERLAYS,
 }
 
+def _UprevResponse(_input_proto, output_proto, _config):
+  """Add fake paths to a successful uprev response."""
+  output_proto.modified_ebuilds.add().path = '/fake/path1'
+  output_proto.modified_ebuilds.add().path = '/fake/path2'
 
-@faux.all_empty
+@faux.success(_UprevResponse)
+@faux.empty_error
 @validate.require('overlay_type')
 @validate.is_in('overlay_type', _OVERLAY_TYPE_TO_NAME)
 @validate.validation_complete
@@ -49,7 +54,14 @@
     output_proto.modified_ebuilds.add().path = path
 
 
-@faux.all_empty
+def _UprevVersionedPackageResponse(_input_proto, output_proto, _config):
+  """Add fake paths to a successful uprev versioned package response."""
+  uprev_response = output_proto.responses.add()
+  uprev_response.modified_ebuilds.add().path = '/uprev/response/path'
+
+
+@faux.success(_UprevVersionedPackageResponse)
+@faux.empty_error
 @validate.require('versions')
 @validate.require('package_info.package_name', 'package_info.category')
 @validate.validation_complete
@@ -83,7 +95,18 @@
       uprev_response.modified_ebuilds.add().path = path
 
 
-@faux.all_empty
+def _GetBestVisibleResponse(_input_proto, output_proto, _config):
+  """Add fake paths to a successful GetBestVisible response."""
+  package_info = common_pb2.PackageInfo(
+      category='category',
+      package_name='name',
+      version='1.01',
+  )
+  output_proto.package_info.CopyFrom(package_info)
+
+
+@faux.success(_GetBestVisibleResponse)
+@faux.empty_error
 @validate.require('atom')
 @validate.validation_complete
 def GetBestVisible(input_proto, output_proto, _config):
@@ -113,7 +136,19 @@
   output_proto.version = packages.determine_chrome_version(build_target)
 
 
-@faux.all_empty
+def _GetTargetVersionsResponse(_input_proto, output_proto, _config):
+  """Add fake target version fields to a successful response."""
+  output_proto.android_version = '5812377'
+  output_proto.android_branch_version = 'git_nyc-mr1-arc'
+  output_proto.android_target_version = 'cheets'
+  output_proto.chrome_version = '78.0.3900.0'
+  output_proto.platform_version = '12438.0.0'
+  output_proto.milestone_version = '78'
+  output_proto.full_version = 'R78-12438.0.0'
+
+
+@faux.success(_GetTargetVersionsResponse)
+@faux.empty_error
 @validate.require('build_target.name')
 @validate.validation_complete
 def GetTargetVersions(input_proto, output_proto, _config):