fix test

Author: hanouticelina

tests/test_cli.py

@@ -269,22 +269,24 @@ def test_verify_success(self, runner: CliRunner) -> None:
 
             with (
                 patch("huggingface_hub.cli.cache.scan_cache_dir", return_value=hf_cache_info),
-                patch(
-                    "huggingface_hub.cli.cache.list_repo_tree",
-                    return_value=[SimpleNamespace(path=file_name, blob_id="unused", lfs={"sha256": blob_identifier})],
-                ) as list_tree_mock,
+                patch("huggingface_hub.cli.cache.get_hf_api") as get_api_mock,
             ):
+                api = get_api_mock.return_value
+                api.list_repo_tree.return_value = [
+                    SimpleNamespace(path=file_name, blob_id="unused", lfs={"sha256": blob_identifier})
+                ]
                 result = runner.invoke(app, ["cache", "verify", repo.cache_id])
 
         assert result.exit_code == 0
         assert "Verified 1 file(s) across 1 revision(s)" in result.stdout
-        list_tree_mock.assert_called_once()
-        kwargs = list_tree_mock.call_args.kwargs
+        get_api_mock.assert_called_once()
+        api = get_api_mock.return_value
+        api.list_repo_tree.assert_called_once()
+        kwargs = api.list_repo_tree.call_args.kwargs
         assert kwargs["repo_id"] == repo_id
         assert kwargs["recursive"] is True
         assert kwargs["revision"] == commit_hash
         assert kwargs["repo_type"] == "model"
-        assert kwargs["token"] is None
 
     def test_verify_reports_mismatch(self, runner: CliRunner) -> None:
         commit_hash = "2" * 40
@@ -339,11 +341,12 @@ def test_verify_reports_mismatch(self, runner: CliRunner) -> None:
 
             with (
                 patch("huggingface_hub.cli.cache.scan_cache_dir", return_value=hf_cache_info),
-                patch(
-                    "huggingface_hub.cli.cache.list_repo_tree",
-                    return_value=[SimpleNamespace(path=file_name, blob_id="unused", lfs={"sha256": "c" * 64})],
-                ),
+                patch("huggingface_hub.cli.cache.get_hf_api") as get_api_mock,
             ):
+                api = get_api_mock.return_value
+                api.list_repo_tree.return_value = [
+                    SimpleNamespace(path=file_name, blob_id="unused", lfs={"sha256": "c" * 64})
+                ]
                 result = runner.invoke(app, ["cache", "verify", repo.cache_id])
 
         assert result.exit_code == 1
@@ -404,16 +407,17 @@ def test_verify_reports_missing_local_file(self, runner: CliRunner) -> None:
 
             with (
                 patch("huggingface_hub.cli.cache.scan_cache_dir", return_value=hf_cache_info),
-                patch(
-                    "huggingface_hub.cli.cache.list_repo_tree",
-                    return_value=[
-                        SimpleNamespace(path=file_name, blob_id="unused", lfs=None),
-                        SimpleNamespace(
-                            path="missing.txt", blob_id="deadbeefdeadbeefdeadbeefdeadbeefdeadbeef", lfs=None
-                        ),
-                    ],
-                ),
+                patch("huggingface_hub.cli.cache.get_hf_api") as get_api_mock,
             ):
+                api = get_api_mock.return_value
+                api.list_repo_tree.return_value = [
+                    SimpleNamespace(path=file_name, blob_id="unused", lfs=None),
+                    SimpleNamespace(
+                        path="missing.txt",
+                        blob_id="blobid",
+                        lfs=None,
+                    ),
+                ]
                 result = runner.invoke(app, ["cache", "verify", repo.cache_id])
 
         assert result.exit_code == 1