Testing __main__

puppeter/__init__.py

@@ -16,8 +16,3 @@ def get_logger(cls):
 
 def __fullname(cls):
     return cls.__module__ + "." + cls.__name__
-
-
-if __name__ == '__main__':
-    from puppeter.main import main
-    main()

puppeter/__main__.py

@@ -1,2 +1,4 @@
 from puppeter.main import main
-main()
+
+if __name__ == '__main__':
+    main()

setup.py

@@ -41,7 +41,7 @@ def find_version(*file_paths):
     # Versions should comply with PEP440.  For a discussion on single-sourcing
     # the version across setup.py and the project code, see
     # https://packaging.python.org/en/latest/single_source_version.html
-    version=find_version('puppeter', '__init__.py'),
+    version=find_version('puppeter', '__main__.py'),
 
     description='Puppeter - an automatic puppet installer',
     # Get the long description from the README file

tests/__init__.py

@@ -1,3 +1,4 @@
 # the inclusion of the tests module is not meant to offer best practices for
 # testing in general, but rather to support the `find_packages` example in
 # setup.py that excludes installing the "tests" package
+

tests/test__main__.py

@@ -0,0 +1,21 @@
+import re
+
+import pytest
+
+from tests.helpers.commandline import captured_output
+
+
+def test_usage_printout():
+    from puppeter import __main__
+
+    with pytest.raises(SystemExit) as sysexit:
+        with captured_output() as (out, err):
+            __main__.main(['puppeter', '--help'])
+
+    stdout = out.getvalue().strip()
+    stderr = err.getvalue().strip()
+    assert sysexit.value.code == 0
+    assert stdout == ''
+    assert re.compile('Puppeter', re.MULTILINE).search(stderr)
+    assert re.compile('--answers FILE, -a FILE', re.MULTILINE).search(stderr)
+    assert re.compile('--execute, -e', re.MULTILINE).search(stderr)