Merge 1dceb2b into b4694f9

linkedin · Jan 5, 2019 · b7a2b35 · b7a2b35
2 parents b4694f9 + 1dceb2b
commit b7a2b35
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/src/shiv/cli.py b/src/shiv/cli.py
@@ -205,4 +205,4 @@ def main(
 
 
 if __name__ == "__main__":
-    main()
+    main()  # pragma: no cover
diff --git a/test/package/setup.py b/test/package/setup.py
@@ -5,4 +5,5 @@
 setup(
     name='hello',
     packages=['hello'],
+    entry_points={"console_scripts": ["hello = hello:main"]},
 )
diff --git a/test/test_cli.py b/test/test_cli.py
@@ -10,8 +10,9 @@
 
 from click.testing import CliRunner
 
-from shiv.cli import main, _interpreter_path
+from shiv.cli import main, find_entry_point, _interpreter_path
 from shiv.constants import DISALLOWED_PIP_ARGS, NO_PIP_ARGS_OR_SITE_PACKAGES, NO_OUTFILE, DISALLOWED_ARGS
+from shiv.pip import install
 
 
 @contextlib.contextmanager
@@ -40,6 +41,11 @@ def runner(self):
 
         return lambda args: CliRunner().invoke(main, args)
 
+    def test_find_entry_point(self, tmpdir, package_location):
+        """Test that we can find console_script metadata."""
+        install(['-t', str(tmpdir), str(package_location)])
+        assert find_entry_point(Path(tmpdir), "hello") == "hello:main"
+
     def test_no_args(self, runner):
         """This should fail with a warning about supplying pip arguments"""