Get the command name from the file path of the first command-line argument.

1 files changed, 24 insertions, 0 deletions

diff --git a/tests/test_cmdline.py b/tests/test_cmdline.py
index 2378887a..d7b5d307 100644
--- a/tests/test_cmdline.py
+++ b/tests/test_cmdline.py
@@ -642,6 +642,30 @@ class CmdLineStdoutTest(BaseCmdLineTest):
             self.assertIn("without C extension", out)
         self.assertLess(out.count("\n"), 4)
 
+    def test_help_contains_command_name(self):
+        """ Command name should be present in help output. """
+        fake_command_path = "lorem/ipsum/dolor"
+        expected_command_name = "dolor"
+        fake_argv = [fake_command_path, "sit", "amet"]
+        with mock.patch.object(sys, 'argv', new=fake_argv):
+            self.command_line("help")
+        self.assertIn(expected_command_name, out)
+
+    def test_help_contains_command_name_from_package(self):
+        """
+        Command package name should be present in help output.
+
+        When the main module is actually a package's `__main__` module, the resulting command line
+        has the `__main__.py` file's patch as the command name. Instead, the command name should be
+        derived from the package name.
+        """
+        fake_command_path = "lorem/ipsum/dolor/__main__.py"
+        expected_command_name = "dolor"
+        fake_argv = [fake_command_path, "sit", "amet"]
+        with mock.patch.object(sys, 'argv', new=fake_argv):
+            self.command_line("help")
+        self.assertIn(expected_command_name, out)
+
     def test_help(self):
         self.command_line("help")
         out = self.stdout()