Test that prep_download returns the right File object

1 files changed, 19 insertions, 5 deletions

diff --git a/tests/test_dl.py b/tests/test_dl.py
index 297c59d..6d0b484 100644
--- a/tests/test_dl.py
+++ b/tests/test_dl.py
@@ -16,8 +16,11 @@
 # along with Wikiget. If not, see <https://www.gnu.org/licenses/>.
 
 from pathlib import Path
+from unittest.mock import MagicMock, patch
 
 import pytest
+from mwclient import Site
+from mwclient.image import Image
 
 from wikiget.dl import prep_download, process_download
 from wikiget.file import File
@@ -25,13 +28,24 @@ from wikiget.wikiget import parse_args
 
 
 class TestPrepDownload:
-    # TODO: don't hit the actual API when doing tests
-    @pytest.mark.skip(reason="skip tests that query a live API")
-    def test_prep_download(self) -> None:
-        """The prep_download function should create a file object."""
+    @patch("wikiget.dl.query_api")
+    @patch("wikiget.dl.connect_to_site")
+    def test_prep_download(
+        self, mock_connect_to_site: MagicMock, mock_query_api: MagicMock
+    ) -> None:
+        """The prep_download function should create the expected file object."""
+        mock_site = MagicMock(Site)
+        mock_image = MagicMock(Image)
+
+        mock_connect_to_site.return_value = mock_site
+        mock_query_api.return_value = mock_image
+
+        expected_file = File(name="Example.jpg")
+        expected_file.image = mock_image
+
         args = parse_args(["File:Example.jpg"])
         file = prep_download(args.FILE, args)
-        assert file is not None
+        assert file == expected_file
 
     def test_prep_download_with_existing_file(self, tmp_path: Path) -> None:
         """