From: Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com>
Date: Sun, 22 Sep 2024 17:42:14 +0000 (+0200)
Subject: [3.12] gh-95468: Add more tests for "--" (double dash) in test_argparse (GH-124274... 
X-Git-Tag: v3.12.7~70
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=25312403de78bbc6896a1934fec46164e58a5965;p=thirdparty%2FPython%2Fcpython.git

[3.12] gh-95468: Add more tests for "--" (double dash) in test_argparse (GH-124274) (GH-124276)

(cherry picked from commit baa3550bc3a119f41cc4eaed5373f9d695208e8e)

Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
---

diff --git a/Lib/test/test_argparse.py b/Lib/test/test_argparse.py
index e1fe42b505da..42998d5c6ea0 100644
--- a/Lib/test/test_argparse.py
+++ b/Lib/test/test_argparse.py
@@ -5383,7 +5383,9 @@ class TestParseKnownArgs(TestCase):
         args = parser.parse_args([])
         self.assertEqual(NS(x=[]), args)
 
-    def test_double_dash(self):
+
+class TestDoubleDash(TestCase):
+    def test_single_argument_option(self):
         parser = argparse.ArgumentParser(exit_on_error=False)
         parser.add_argument('-f', '--foo')
         parser.add_argument('bar', nargs='*')
@@ -5407,6 +5409,7 @@ class TestParseKnownArgs(TestCase):
         args = parser.parse_args(['a', '--', 'b', '--', 'c', '--foo', 'd'])
         self.assertEqual(NS(foo=None, bar=['a', 'b', '--', 'c', '--foo', 'd']), args)
 
+    def test_multiple_argument_option(self):
         parser = argparse.ArgumentParser(exit_on_error=False)
         parser.add_argument('-f', '--foo', nargs='*')
         parser.add_argument('bar', nargs='*')
@@ -5429,6 +5432,7 @@ class TestParseKnownArgs(TestCase):
         self.assertEqual(NS(foo=['c'], bar=['a', 'b']), args)
         self.assertEqual(argv, ['--', 'd'])
 
+    def test_multiple_double_dashes(self):
         parser = argparse.ArgumentParser(exit_on_error=False)
         parser.add_argument('foo')
         parser.add_argument('bar', nargs='*')
@@ -5444,9 +5448,10 @@ class TestParseKnownArgs(TestCase):
         args = parser.parse_args(['--', '--', 'a', '--', 'b', 'c'])
         self.assertEqual(NS(foo='--', bar=['a', '--', 'b', 'c']), args)
 
+    def test_remainder(self):
         parser = argparse.ArgumentParser(exit_on_error=False)
         parser.add_argument('foo')
-        parser.add_argument('bar', nargs=argparse.REMAINDER)
+        parser.add_argument('bar', nargs='...')
 
         args = parser.parse_args(['--', 'a', 'b', 'c'])
         self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
@@ -5457,6 +5462,40 @@ class TestParseKnownArgs(TestCase):
         args = parser.parse_args(['a', '--', 'b', '--', 'c'])
         self.assertEqual(NS(foo='a', bar=['b', '--', 'c']), args)
 
+        parser = argparse.ArgumentParser(exit_on_error=False)
+        parser.add_argument('--foo')
+        parser.add_argument('bar', nargs='...')
+        args = parser.parse_args(['--foo', 'a', '--', 'b', '--', 'c'])
+        self.assertEqual(NS(foo='a', bar=['--', 'b', '--', 'c']), args)
+
+    def test_subparser(self):
+        parser = argparse.ArgumentParser(exit_on_error=False)
+        parser.add_argument('foo')
+        subparsers = parser.add_subparsers()
+        parser1 = subparsers.add_parser('run')
+        parser1.add_argument('-f')
+        parser1.add_argument('bar', nargs='*')
+
+        args = parser.parse_args(['x', 'run', 'a', 'b', '-f', 'c'])
+        self.assertEqual(NS(foo='x', f='c', bar=['a', 'b']), args)
+        args = parser.parse_args(['x', 'run', 'a', 'b', '--', '-f', 'c'])
+        self.assertEqual(NS(foo='x', f=None, bar=['a', 'b', '-f', 'c']), args)
+        args = parser.parse_args(['x', 'run', 'a', '--', 'b', '-f', 'c'])
+        self.assertEqual(NS(foo='x', f=None, bar=['a', 'b', '-f', 'c']), args)
+        args = parser.parse_args(['x', 'run', '--', 'a', 'b', '-f', 'c'])
+        self.assertEqual(NS(foo='x', f=None, bar=['a', 'b', '-f', 'c']), args)
+        args = parser.parse_args(['x', '--', 'run', 'a', 'b', '-f', 'c'])
+        self.assertEqual(NS(foo='x', f='c', bar=['a', 'b']), args)
+        args = parser.parse_args(['--', 'x', 'run', 'a', 'b', '-f', 'c'])
+        self.assertEqual(NS(foo='x', f='c', bar=['a', 'b']), args)
+        args = parser.parse_args(['x', 'run', '--', 'a', '--', 'b'])
+        self.assertEqual(NS(foo='x', f=None, bar=['a', '--', 'b']), args)
+        args = parser.parse_args(['x', '--', 'run', '--', 'a', '--', 'b'])
+        self.assertEqual(NS(foo='x', f=None, bar=['a', '--', 'b']), args)
+        self.assertRaisesRegex(argparse.ArgumentError,
+            "invalid choice: '--'",
+            parser.parse_args, ['--', 'x', '--', 'run', 'a', 'b'])
+
 
 # ===========================
 # parse_intermixed_args tests