Increase code coverage by testing CLI functions

bin/documentation-builder

@@ -2,6 +2,7 @@
 
 # Core modules
 import locale
+import sys
 
 # Local modules
 from ubuntudesign.documentation_builder import cli
@@ -13,5 +14,5 @@ if 'UTF-8' not in locale.getlocale():
 
 if __name__ == "__main__":
     # Run the CLI
-    cli.main()
+    cli.main(sys.argv[1:])
 

tests/test_cli.py

@@ -0,0 +1,23 @@
+# System
+from shutil import rmtree
+
+# Local modules
+from ubuntudesign.documentation_builder.cli import main, parse_arguments
+
+
+def test_parse_arguments():
+    arguments = parse_arguments([])
+
+    assert type(arguments) == dict
+
+def test_main():
+    main(
+        [
+             '--base-directory',
+             'tests/fixtures/builder/base/',
+             '--output-path',
+             'doesnt/matter'
+        ]
+    )
+    rmtree('doesnt')
+

ubuntudesign/documentation_builder/cli.py

@@ -7,7 +7,7 @@
 from .builder import Builder
 
 
-def parse_arguments():
+def parse_arguments(arguments):
     """
     Parse command-line options for documentation-parser command-line script
     """
@@ -114,7 +114,7 @@ def parse_arguments():
         help="Show the currently installed version of documentation-builder."
     )
 
-    arguments = vars(parser.parse_args())
+    arguments = vars(parser.parse_args(arguments))
 
     if arguments['version']:
         print(
@@ -130,15 +130,15 @@ def parse_arguments():
     return {name: value for name, value in arguments.items() if value}
 
 
-def main():
+def main(system_arguments):
     """
     The starting point for the documentation-parser.
     Intended to be run through the command-line.
     """
 
-    arguments = parse_arguments()
+    arguments = parse_arguments(system_arguments)
     Builder(**arguments)
 
 
 if __name__ == "__main__":
-    main()
+    main(sys.argv[1:])