First commit of cros tool.
This commits the basis of the cros tool with unittests. Ability to
find subcmds taken from previous chromite shell work.
cros is the base tool for cros-ui-v2. It finds cros_<tool_name> tools
from cros.commands. Currently, only the functionality for finding
and calling sub-tools has been implemented.
BUG=None
TEST=Unittests + ran tool with example tool + pylint
Change-Id: Ifeb1a8f7ea508cea8a2479f54f9bdd1bd52a3030
Reviewed-on: https://gerrit.chromium.org/gerrit/34837
Commit-Ready: Chris Sosa <sosa@chromium.org>
Reviewed-by: Chris Sosa <sosa@chromium.org>
Tested-by: Chris Sosa <sosa@chromium.org>
diff --git a/cros/commands/command_unittest.py b/cros/commands/command_unittest.py
new file mode 100755
index 0000000..4364ead
--- /dev/null
+++ b/cros/commands/command_unittest.py
@@ -0,0 +1,59 @@
+#!/usr/bin/python
+
+# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""This module tests the command module."""
+
+import argparse
+import os
+import sys
+import unittest
+
+sys.path.insert(0, os.path.abspath('%s/../../..' % os.path.dirname(__file__)))
+
+from chromite.lib import cros_test_lib
+from chromite.cros.commands import command
+
+_COMMAND_NAME = 'superAwesomeCommandOfFunness'
+
+
+@command.CommandDecorator(_COMMAND_NAME)
+class TestCommand(command.CrosCommand):
+ """A fake command."""
+ def Run(self):
+ print 'Just testing'
+
+
+class CommandTest(unittest.TestCase):
+ """This test class tests that Commands method."""
+
+ def testParserSetsCrosClass(self):
+ """Tests that our parser sets cros_class correctly."""
+ my_parser = argparse.ArgumentParser()
+ command.CrosCommand.AddParser(my_parser)
+ ns = my_parser.parse_args([])
+ self.assertEqual(ns.cros_class, command.CrosCommand)
+
+ def testCommandDecorator(self):
+ """Tests that our decorator correctly adds TestCommand to _commands."""
+ # Note this exposes an implementation detail of _commands.
+ self.assertEqual(command._commands[_COMMAND_NAME], TestCommand)
+
+ def testBadUseOfCommandDecorator(self):
+ """Tests that our decorator correctly rejects bad test commands."""
+ try:
+ @command.CommandDecorator('bad')
+ class BadTestCommand(object):
+ """A command that wasn't implement correctly."""
+ pass
+
+ except command.InvalidCommandError:
+ pass
+ else:
+ self.fail('Invalid command was accepted by the CommandDecorator')
+
+
+if __name__ == '__main__':
+ cros_test_lib.main()