commit b70e8a807b9946133d58bb386ea64455e8e01bc3
author Michael Mortensen <mmortensen@google.com> Thu Oct 10 18:43:41 2019 -0600
committer Commit Bot <commit-bot@chromium.org> Sun Oct 13 23:05:48 2019 +0000
tree 484800ebaa19e17e300562ad11f1a270e1b317fb
parent c28d6f1164724a9021a0611789b2a3fb10eb5bd0

Begin implementing GetTargetVersions by porting code to get android
versions.

Regenerated api/gen/chromite/api/packages_pb2.py by running
api/compile_build_api_proto.

BUG=chromium:1004438
TEST=manual, run_tests

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

service/packages_unittest.py

diff --git a/service/packages_unittest.py b/service/packages_unittest.py
index 47f5791..46a21f4 100644
--- a/service/packages_unittest.py
+++ b/service/packages_unittest.py
@@ -9,6 +9,7 @@
 
 from chromite.lib import build_target_util
 from chromite.lib import cros_test_lib
+from chromite.lib import osutils
 from chromite.lib import portage_util
 from chromite.lib.chroot_lib import Chroot
 from chromite.service import packages
@@ -100,3 +101,33 @@
   def test_empty_atom_fails(self):
     with self.assertRaises(AssertionError):
       packages.has_prebuilt('')
+
+
+class AndroidVersionsTest(cros_test_lib.MockTestCase):
+  """Tests getting android versions."""
+  def setUp(self):
+    package_result = [
+        'chromeos-base/android-container-nyc-4717008-r1',
+        'chromeos-base/update_engine-0.0.3-r3408']
+    self.PatchObject(portage_util, 'GetPackageDependencies',
+                     return_value=package_result)
+    self.board = 'board'
+    self.PatchObject(portage_util, 'FindEbuildForBoardPackage',
+                     return_value='chromeos-base/android-container-nyc')
+    FakeEnvironment = {
+        'ARM_TARGET': '3-linux-target'
+    }
+    self.PatchObject(osutils, 'SourceEnvironment',
+                     return_value=FakeEnvironment)
+
+  def test_determine_android_version(self):
+    version = packages.determine_android_version(self.board)
+    self.assertEqual(version, '4717008')
+
+  def test_determine_android_branch(self):
+    branch = packages.determine_android_branch(self.board)
+    self.assertEqual(branch, '3')
+
+  def test_determine_android_target(self):
+    target = packages.determine_android_target(self.board)
+    self.assertEqual(target, 'cheets')