commit cde48580e93da01999f0032e32ebf7fbca0e474e
author LaMont Jones <lamontjones@chromium.org> Mon Nov 26 15:49:48 2018 -0700
committer chrome-bot <chrome-bot@chromium.org> Tue Nov 27 19:14:03 2018 -0800
tree ad32253edee2e0ed84ea09ccb99e7ddebe19111e
parent 23537c6f8297964489dd0a8ba316246bc6fd22ae

Add missing unit tests.

Last week's CLs missed a function in unittesting.

BUG=chromium:813829
TEST=Updated unit tests pass.

Change-Id: I1c5663f8aa7a240e7fdc795d2a221b1135bdde65
Reviewed-on: https://chromium-review.googlesource.com/1351630
Commit-Ready: LaMont Jones <lamontjones@chromium.org>
Tested-by: LaMont Jones <lamontjones@chromium.org>
Reviewed-by: Jason Clinton <jclinton@chromium.org>

signing/lib/keys_unittest.py

diff --git a/signing/lib/keys_unittest.py b/signing/lib/keys_unittest.py
index 8b81137..6d77729 100644
--- a/signing/lib/keys_unittest.py
+++ b/signing/lib/keys_unittest.py
@@ -431,9 +431,9 @@
 class TestKeyset(cros_test_lib.TempDirTestCase):
   """Test Keyset class."""
 
-  def _get_keyset(self):
+  def _get_keyset(self, has_loem_ini=True):
     """Returns keyset with a few keys."""
-    kc = KeysetMock(self.tempdir, has_loem_ini=True)
+    kc = KeysetMock(self.tempdir, has_loem_ini=has_loem_ini)
 
     # Add a few more keys for this build.
     for key_name in ['key1', 'key2', 'key3', 'key4']:
@@ -498,6 +498,18 @@
 
     self.assertEqual(ks0._buildtarget_keys['loem9']['root_key'], k9)
 
+  def testGetBuildtargetKeysWithLoemIni(self):
+    ks0 = self._get_keyset()
+    expected_keys = ks0.GetBuildtargetKeys('root_key')
+    expected = {k: ks0._buildtarget_keys[k]['root_key']
+                for k in ks0._buildtarget_keys.keys()}
+    self.assertDictEqual(expected, expected_keys)
+
+  def testGetBuildtargetKeysWithoutLoemIni(self):
+    ks0 = self._get_keyset(has_loem_ini=False)
+    expected_keys = ks0.GetBuildtargetKeys('root_key')
+    self.assertDictEqual({'root_key': ks0.keys['root_key']}, expected_keys)
+
   def testGetBuildKeysetMissmatch(self):
     ks0 = self._get_keyset()