summaryrefslogtreecommitdiff
path: root/cli
diff options
context:
space:
mode:
authorRalph Nathan <ralphnathan@chromium.org>2015-06-08 11:03:46 -0700
committerChromeOS Commit Bot <chromeos-commit-bot@chromium.org>2015-06-09 01:26:17 +0000
commit4b59f47586c693c345df034f2daa39724265a73c (patch)
tree7f89f2d59424bac70feb6f352c105cfe7da562d1 /cli
parent5ccc829372cca5c0e685680632292d63edea968c (diff)
downloadchromite-4b59f47586c693c345df034f2daa39724265a73c.tar.gz
brillo image: Different summary message for brillo tools.
Currently, both brillo image and cros image display the same summary message. This CL changes it so brillo image displays a different summary message. It is done by passing '--toolset=brillo' to build_image. BUG=brillo:1189 TEST=unittests + manual tests CQ-DEPEND=CL:275748, CL:275702, CL:275726 Change-Id: I62b75c6018177f53f96c647bd3dc07a968ae5c38 Reviewed-on: https://chromium-review.googlesource.com/275867 Reviewed-by: Steve Fung <stevefung@chromium.org> Commit-Queue: Ralph Nathan <ralphnathan@chromium.org> Trybot-Ready: Ralph Nathan <ralphnathan@chromium.org> Tested-by: Ralph Nathan <ralphnathan@chromium.org>
Diffstat (limited to 'cli')
-rw-r--r--cli/cros/cros_image_unittest.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/cli/cros/cros_image_unittest.py b/cli/cros/cros_image_unittest.py
index dfa942bb1..1ab1a021e 100644
--- a/cli/cros/cros_image_unittest.py
+++ b/cli/cros/cros_image_unittest.py
@@ -136,6 +136,16 @@ class ImageCommandTest(cros_test_lib.WorkspaceTestCase,
self.assertEqual(1, op.call_count)
self.assertTrue('--progress_bar' in op.call_args[0][1])
+ def testToolsetBrillo(self):
+ """Test RunCommand called with --toolset=brillo when using brillo."""
+ self.PatchObject(command, 'GetToolset', return_value='brillo')
+ self.SetupCommandMock([])
+ self.cmd_mock.inst.Run()
+
+ # Test that build_image is called with the correct arguments.
+ cmds = '\n'.join(repr(x) for x in self.rc_mock.call_args_list)
+ self.assertIn('--toolset=brillo', cmds)
+
class ImageCommandParserTest(cros_test_lib.TestCase):
"""Test class for our ImageCommand's parser."""