tarfile: Add basic unittest for tarfile.TarFile.

Tests TarFile iteration and extracting file information.

Signed-off-by: Damien George <damien@micropython.org>

Author: dpgeorge

python-stdlib/tarfile/test.tar

@@ -0,0 +1,37 @@
+import tarfile
+import unittest
+
+
+test_tar_contents = (
+    ("a", "file", 2),
+    ("b", "file", 2),
+    ("dir/", "dir", 0),
+    ("dir/c", "file", 2),
+    ("dir/d", "file", 2),
+    ("tar.tar", "file", 10240),
+)
+
+test_sub_tar_contents = (
+    ("e", "file", 2),
+    ("f", "file", 2),
+)
+
+
+class TestTarFile(unittest.TestCase):
+    def check_contents(self, expected, tf):
+        for i, file in enumerate(tf):
+            name, type, size = expected[i]
+            self.assertEqual(file.name, name)
+            self.assertEqual(file.type, type)
+            self.assertEqual(file.size, size)
+
+    def test_iter(self):
+        tf = tarfile.TarFile("test.tar")
+        for _ in range(6):
+            self.assertIsInstance(next(tf), tarfile.TarInfo)
+        with self.assertRaises(StopIteration):
+            next(tf)
+
+    def test_contents(self):
+        tf = tarfile.TarFile("test.tar")
+        self.check_contents(test_tar_contents, tf)

python-stdlib/tarfile/test_tarfile.py

@@ -90,6 +90,7 @@ function ci_package_tests_run {
         python-stdlib/pathlib \
         python-stdlib/quopri \
         python-stdlib/shutil \
+        python-stdlib/tarfile \
         python-stdlib/tempfile \
         python-stdlib/time \
         python-stdlib/unittest/tests \