From 69c44551ffd70e2b2af442f68642b2dde882ef28 Mon Sep 17 00:00:00 2001 From: Chris MacNaughton Date: Wed, 10 Oct 2018 12:14:15 +0200 Subject: [PATCH] add unit testing and fix typo --- unit_tests/utilities/test_zaza_utilities_bundle.py | 13 +++++++++++++ zaza/utilities/bundle.py | 2 +- 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/unit_tests/utilities/test_zaza_utilities_bundle.py b/unit_tests/utilities/test_zaza_utilities_bundle.py index 5a0c175..eb30477 100644 --- a/unit_tests/utilities/test_zaza_utilities_bundle.py +++ b/unit_tests/utilities/test_zaza_utilities_bundle.py @@ -106,3 +106,16 @@ class TestUtilitiesBundle(ut_utils.BaseTestCase): expected = yaml.safe_load(TEST_BUNDLE_WITHOUT_PLACEMENT) self.assertEqual(expected, flattened) + + def test_parser(self): + args = bundle.parse_args([ + '-i', 'bundle.yaml']) + self.assertEqual(args.input, 'bundle.yaml') + self.assertEqual(args.output, '/dev/stdout') + + def test_parser_output(self): + args = bundle.parse_args([ + '-i', 'bundle.yaml', + '-o', 'bundle_out.yaml']) + self.assertEqual(args.input, 'bundle.yaml') + self.assertEqual(args.output, 'bundle_out.yaml') diff --git a/zaza/utilities/bundle.py b/zaza/utilities/bundle.py index 412050e..3e4e843 100644 --- a/zaza/utilities/bundle.py +++ b/zaza/utilities/bundle.py @@ -21,7 +21,7 @@ import sys def remove_machine_specification(input_yaml): """ - Will remove the machien specifications from a supplied bundle. + Will remove the machine specifications from a supplied bundle. :param input_yaml: Juju bundle to strip machinens from :type input_yaml: dict