test: mock network calls in test_retrieve

tests/test_api.py

@@ -1,6 +1,9 @@
+import io
+import tarfile
 import tempfile
 from pyiri2016.api import update
 from unittest import TestCase
+from unittest.mock import patch, ANY
 from simple_settings import LazySettings
 from parameterized import parameterized
 import pathlib
@@ -9,25 +12,41 @@
 
 
 class TestApiUpdate(TestCase):
-    @classmethod
-    def setUpClass(cls):
-        cls.temporary_directory = tempfile.TemporaryDirectory()
-
-    @classmethod
-    def tearDownClass(cls):
-        cls.temporary_directory.cleanup()
-
-    def _count_files(self, directory: str) -> int:
-        return len([fname for fname in pathlib.Path(directory).iterdir() if fname.is_file()])
+    def _make_tarball(self, directory: str, filename: str) -> str:
+        """Create a minimal tarball fixture with one inner file."""
+        tarball_path = str(pathlib.Path(directory) / filename)
+        inner_content = b"fake fortran source"
+        buf = io.BytesIO(inner_content)
+        buf.seek(0)
+        with tarfile.open(tarball_path, "w") as tar:
+            info = tarfile.TarInfo(name="inner_file.f")
+            info.size = len(inner_content)
+            tar.addfile(info, buf)
+        return tarball_path
 
     @parameterized.expand(
         [
-            (SETTINGS.FORTRAN_CODE_URL, SETTINGS.FORTRAN_CODE_COMPRESSED_FILE),
-            (SETTINGS.COMMON_FILES_URL, SETTINGS.COMMON_FILES_COMPRESSED_FILE),
-            (SETTINGS.INDICES_URL, SETTINGS.INDICES_FILES[0]),
-            (SETTINGS.INDICES_URL, SETTINGS.INDICES_FILES[1]),
+            (SETTINGS.FORTRAN_CODE_URL, SETTINGS.FORTRAN_CODE_COMPRESSED_FILE, True),
+            (SETTINGS.COMMON_FILES_URL, SETTINGS.COMMON_FILES_COMPRESSED_FILE, True),
+            (SETTINGS.INDICES_URL, SETTINGS.INDICES_FILES[0], False),
+            (SETTINGS.INDICES_URL, SETTINGS.INDICES_FILES[1], False),
         ]
     )
-    def test_retrieve(self, url: str, filename: str):
-        update.retrieve(url, filename, directory=self.temporary_directory.name)
-        self.assertGreater(self._count_files(self.temporary_directory.name), 0)
+    def test_retrieve(self, url: str, filename: str, is_tar: bool):
+        with tempfile.TemporaryDirectory() as tmpdir:
+            if is_tar:
+                fake_path = self._make_tarball(tmpdir, filename)
+            else:
+                fake_path = str(pathlib.Path(tmpdir) / filename)
+                pathlib.Path(fake_path).write_bytes(b"fake index data")
+
+            with patch("pyiri2016.api.update.wget.download", return_value=fake_path) as mock_download:
+                update.retrieve(url, filename, directory=tmpdir)
+                mock_download.assert_called_once_with(
+                    f"{url}/{filename}", out=tmpdir, bar=ANY
+                )
+
+            file_count = len(
+                [f for f in pathlib.Path(tmpdir).iterdir() if f.is_file()]
+            )
+            self.assertGreater(file_count, 0)