Skip to content

fix(tests): Mock HF in integration tests #259

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
matthewgrossman wants to merge 4 commits into
base: main
Choose a base branch
from
Draft

fix(tests): Mock HF in integration tests #259

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
e4df4da
fix(tests): Mock HF in integration tests
matthewgrossman Jun 10, 2026
50602f5
add fixtures
matthewgrossman Jun 10, 2026
62a4252
fix(tests): add parallelism conftest and fixture download script
matthewgrossman Jun 10, 2026
2c2f641
fix(tests): remove parallelism fixtures, keep only auth test mock
matthewgrossman Jun 10, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 17 additions & 1 deletion services/core/models/tests/integration/test_models_with_auth.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@

from contextlib import contextmanager
from typing import Generator
from unittest.mock import patch
from unittest.mock import Mock, patch

import pytest
from nemo_platform import NeMoPlatform, PermissionDeniedError
Expand Down Expand Up @@ -1292,6 +1292,22 @@ class TestTrustRemoteCodePermission:
trust_remote_code=True requires models.trust-remote-code.set.
"""

@pytest.fixture(autouse=True)
def _mock_hf_storage(self):
"""Prevent real HuggingFace API calls during fileset creation.

These tests verify authorization logic, not HF connectivity.
Mocking the HfApi avoids rate-limit failures in CI.
"""
with patch("nmp.core.files.app.backends.huggingface.HfApi") as mock_cls:
mock_api = Mock()
mock_repo_info = Mock()
mock_repo_info.sha = "abc123mocked"
mock_repo_info.siblings = [] # skip file metadata check
mock_api.repo_info.return_value = mock_repo_info
mock_cls.return_value = mock_api
yield

def test_create_model_trust_remote_code_true_has_permission_succeeds(self, sdk: NeMoPlatform):
"""Create with trust_remote_code=True succeeds when principal has models.trust-remote-code.set (repo not on allow list)."""
workspace = short_unique_name("trc-has")
Expand Down
Loading