gh-117829 : adding cmdline tests

Author: VietThan

Lib/test/test_zipapp.py

@@ -326,12 +326,14 @@ def test_create_archive_with_include_pattern(self):
         (source / 'bin' / 'qux').touch()
         (source / 'bin' / 'baz').touch()
         (source / '__main__.py').touch()
+
         target = io.BytesIO()
         zipapp.create_archive(
             source=str(source),
             target=target,
             include_pattern=re.compile(r".*\.py")
         )
+
         target.seek(0)
         with zipfile.ZipFile(target, 'r') as zf:
             self.assertEqual(zf.namelist(),
@@ -346,12 +348,14 @@ def test_create_archive_with_exclude_pattern(self):
         (source / 'bin' / 'qux').touch()
         (source / 'bin' / 'baz').touch()
         (source / '__main__.py').touch()
+
         target = io.BytesIO()
         zipapp.create_archive(
             source=str(source),
             target=target,
             exclude_pattern=re.compile(r".*\.py")
         )
+
         target.seek(0)
         with zipfile.ZipFile(target, 'r') as zf:
             self.assertEqual(zf.namelist(),
@@ -366,13 +370,15 @@ def test_create_archive_with_include_and_exclude_pattern(self):
         (source / 'bin' / 'qux').touch()
         (source / 'bin' / 'baz').touch()
         (source / '__main__.py').touch()
+
         target = io.BytesIO()
         zipapp.create_archive(
             source=str(source),
             target=target,
             include_pattern=re.compile(r".*\.py"),
             exclude_pattern=re.compile(r".*z.*")
         )
+
         target.seek(0)
         with zipfile.ZipFile(target, 'r') as zf:
             self.assertEqual(zf.namelist(),
@@ -480,7 +486,61 @@ def test_info_error(self):
         self.assertTrue(cm.exception.code)
 
     def test_cmdline_create_with_include_pattern(self):
+        source = self.tmpdir / 'source'
+        source.mkdir()
+        (source / '.DS_Store').touch()
+        (source / 'zed.py').touch()
+        (source / 'bin').mkdir()
+        (source / 'bin' / 'qux').touch()
+        (source / 'bin' / 'baz').touch()
+        (source / '__main__.py').touch()
+
+        args = [str(source), '--include-pattern', r'.*\.py']
+        zipapp.main(args)
+        target = source.with_suffix('.pyz')
+        self.assertTrue(target.is_file())
+
+        with zipfile.ZipFile(target, 'r') as zf:
+            self.assertEqual(zf.namelist(),
+                    ["__main__.py", "zed.py"])
+            
+    def test_cmdline_create_with_exclude_pattern(self):
+        source = self.tmpdir / 'source'
+        source.mkdir()
+        (source / '.DS_Store').touch()
+        (source / 'zed.py').touch()
+        (source / 'bin').mkdir()
+        (source / 'bin' / 'qux').touch()
+        (source / 'bin' / 'baz').touch()
+        (source / '__main__.py').touch()
+
+        args = [str(source), '--exclude-pattern', r'.*\.py']
+        zipapp.main(args)
+        target = source.with_suffix('.pyz')
+        self.assertTrue(target.is_file())
+
+        with zipfile.ZipFile(target, 'r') as zf:
+            self.assertEqual(zf.namelist(),
+                [".DS_Store", "bin/", "bin/baz", "bin/qux"])
+
+    def test_cmdline_create_with_include_and_exclude_pattern(self):
+        source = self.tmpdir / 'source'
+        source.mkdir()
+        (source / '.DS_Store').touch()
+        (source / 'zed.py').touch()
+        (source / 'bin').mkdir()
+        (source / 'bin' / 'qux').touch()
+        (source / 'bin' / 'baz').touch()
+        (source / '__main__.py').touch()
 
+        args = [str(source), '--include-pattern', r'.*\.py', '--exclude-pattern', r'.*z.*']
+        zipapp.main(args)
+        target = source.with_suffix('.pyz')
+        self.assertTrue(target.is_file())
+        
+        with zipfile.ZipFile(target, 'r') as zf:
+            self.assertEqual(zf.namelist(),
+                ["__main__.py"])
 
 
 if __name__ == "__main__":