From e4399a07a798ea83156f300116d33e39f6b6c19f Mon Sep 17 00:00:00 2001
From: Daniel Watkins <daniel.watkins@canonical.com>
Date: Thu, 26 Feb 2015 15:26:15 +0000
Subject: Fix traceback with no arguments on Python 3.

Fixes bug 1424277.
---
 bin/cloud-init              |  2 ++
 tests/unittests/test_cli.py | 48 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 tests/unittests/test_cli.py

diff --git a/bin/cloud-init b/bin/cloud-init
index 6c83c2e7..e95fea28 100755
--- a/bin/cloud-init
+++ b/bin/cloud-init
@@ -609,6 +609,8 @@ def main():
     # Setup signal handlers before running
     signal_handler.attach_handlers()
 
+    if not hasattr(args, 'action'):
+        parser.error('too few arguments')
     (name, functor) = args.action
     if name in ("modules", "init"):
         functor = status_wrapper
diff --git a/tests/unittests/test_cli.py b/tests/unittests/test_cli.py
new file mode 100644
index 00000000..ba447f87
--- /dev/null
+++ b/tests/unittests/test_cli.py
@@ -0,0 +1,48 @@
+import imp
+import sys
+
+import six
+
+from . import helpers as test_helpers
+
+try:
+    from unittest import mock
+except ImportError:
+    import mock
+
+
+class TestCLI(test_helpers.FilesystemMockingTestCase):
+
+    def setUp(self):
+        super(TestCLI, self).setUp()
+        self.stderr = six.StringIO()
+        self.patchStdoutAndStderr(stderr=self.stderr)
+        self.sys_exit = mock.MagicMock()
+        self.patched_funcs.enter_context(
+            mock.patch.object(sys, 'exit', self.sys_exit))
+
+    def _call_main(self):
+        self.patched_funcs.enter_context(
+            mock.patch.object(sys, 'argv', ['cloud-init']))
+        cli = imp.load_module(
+            'cli', open('bin/cloud-init'), '', ('', 'r', imp.PY_SOURCE))
+        try:
+            return cli.main()
+        except:
+            pass
+
+    def test_no_arguments_shows_usage(self):
+        self._call_main()
+        self.assertIn('usage: cloud-init', self.stderr.getvalue())
+
+    def test_no_arguments_exits_2(self):
+        exit_code = self._call_main()
+        if self.sys_exit.call_count:
+            self.assertEqual(mock.call(2), self.sys_exit.call_args)
+        else:
+            self.assertEqual(2, exit_code)
+
+    def test_no_arguments_shows_error_message(self):
+        self._call_main()
+        self.assertIn('cloud-init: error: too few arguments',
+                      self.stderr.getvalue())
-- 
cgit v1.2.3