|
1 | 1 | import json |
| 2 | +import logging |
2 | 3 | import pathlib |
3 | 4 | from unittest.mock import Mock, patch |
4 | 5 |
|
| 6 | +import pytest |
| 7 | +import requests.exceptions |
5 | 8 | from packaging.requirements import Requirement |
6 | 9 | from packaging.version import Version |
| 10 | +from resolvelib.resolvers import ResolverException |
7 | 11 |
|
8 | 12 | from fromager import bootstrapper, requirements_file |
9 | 13 | from fromager.context import WorkContext |
@@ -293,3 +297,175 @@ def test_build_from_source_returns_dataclass(tmp_context: WorkContext) -> None: |
293 | 297 | assert result.sdist_root_dir == mock_sdist_root |
294 | 298 | assert result.build_env is not None |
295 | 299 | assert result.source_type == SourceType.SDIST |
| 300 | + |
| 301 | + |
| 302 | +class TestDownloadWheelFromCacheErrorHandling: |
| 303 | + """Tests for _download_wheel_from_cache exception differentiation.""" |
| 304 | + |
| 305 | + def test_resolver_exception_returns_none( |
| 306 | + self, tmp_context: WorkContext, caplog: pytest.LogCaptureFixture |
| 307 | + ) -> None: |
| 308 | + """ResolverException (wheel not found) returns (None, None) and logs info.""" |
| 309 | + bt = bootstrapper.Bootstrapper(tmp_context) |
| 310 | + bt.cache_wheel_server_url = "https://cache.example.com/simple" |
| 311 | + |
| 312 | + with patch( |
| 313 | + "fromager.resolver.resolve", |
| 314 | + side_effect=ResolverException("no matching version"), |
| 315 | + ): |
| 316 | + result = bt._download_wheel_from_cache( |
| 317 | + req=Requirement("test-package"), |
| 318 | + resolved_version=Version("1.0.0"), |
| 319 | + ) |
| 320 | + |
| 321 | + assert result == (None, None) |
| 322 | + log_record = next( |
| 323 | + r for r in caplog.records if "did not find wheel for" in r.message |
| 324 | + ) |
| 325 | + assert log_record.levelno == logging.INFO |
| 326 | + |
| 327 | + @pytest.mark.parametrize( |
| 328 | + "exc_class,exc_msg", |
| 329 | + [ |
| 330 | + (requests.exceptions.ConnectionError, "DNS failure"), |
| 331 | + (requests.exceptions.Timeout, "timed out"), |
| 332 | + (requests.exceptions.HTTPError, "401 Unauthorized"), |
| 333 | + ], |
| 334 | + ) |
| 335 | + def test_request_exception_returns_none_with_warning( |
| 336 | + self, |
| 337 | + tmp_context: WorkContext, |
| 338 | + caplog: pytest.LogCaptureFixture, |
| 339 | + exc_class: type[Exception], |
| 340 | + exc_msg: str, |
| 341 | + ) -> None: |
| 342 | + """RequestException subtypes return (None, None) and log warning.""" |
| 343 | + bt = bootstrapper.Bootstrapper(tmp_context) |
| 344 | + bt.cache_wheel_server_url = "https://cache.example.com/simple" |
| 345 | + |
| 346 | + with patch( |
| 347 | + "fromager.resolver.resolve", |
| 348 | + side_effect=exc_class(exc_msg), |
| 349 | + ): |
| 350 | + result = bt._download_wheel_from_cache( |
| 351 | + req=Requirement("test-package"), |
| 352 | + resolved_version=Version("1.0.0"), |
| 353 | + ) |
| 354 | + |
| 355 | + assert result == (None, None) |
| 356 | + log_record = next( |
| 357 | + r |
| 358 | + for r in caplog.records |
| 359 | + if "network error checking wheel cache" in r.message |
| 360 | + ) |
| 361 | + assert log_record.levelno == logging.WARNING |
| 362 | + |
| 363 | + def test_unexpected_exception_returns_none_with_warning( |
| 364 | + self, tmp_context: WorkContext, caplog: pytest.LogCaptureFixture |
| 365 | + ) -> None: |
| 366 | + """Unexpected exceptions (e.g. ValueError, OSError) return (None, None) and log warning.""" |
| 367 | + bt = bootstrapper.Bootstrapper(tmp_context) |
| 368 | + bt.cache_wheel_server_url = "https://cache.example.com/simple" |
| 369 | + |
| 370 | + with patch( |
| 371 | + "fromager.resolver.resolve", |
| 372 | + side_effect=ValueError("unexpected parsing error"), |
| 373 | + ): |
| 374 | + result = bt._download_wheel_from_cache( |
| 375 | + req=Requirement("test-package"), |
| 376 | + resolved_version=Version("1.0.0"), |
| 377 | + ) |
| 378 | + |
| 379 | + assert result == (None, None) |
| 380 | + log_record = next( |
| 381 | + r |
| 382 | + for r in caplog.records |
| 383 | + if "unexpected error checking wheel cache" in r.message |
| 384 | + ) |
| 385 | + assert log_record.levelno == logging.WARNING |
| 386 | + |
| 387 | + def test_download_wheel_network_error_returns_none_with_warning( |
| 388 | + self, tmp_context: WorkContext, caplog: pytest.LogCaptureFixture |
| 389 | + ) -> None: |
| 390 | + """RequestException from download_wheel (after resolve succeeds) is caught.""" |
| 391 | + bt = bootstrapper.Bootstrapper(tmp_context) |
| 392 | + bt.cache_wheel_server_url = "https://cache.example.com/simple" |
| 393 | + |
| 394 | + with ( |
| 395 | + patch( |
| 396 | + "fromager.resolver.resolve", |
| 397 | + return_value=( |
| 398 | + "https://cache.example.com/simple/test-package/test_package-1.0.0-py3-none-any.whl", |
| 399 | + "1.0.0", |
| 400 | + ), |
| 401 | + ), |
| 402 | + patch( |
| 403 | + "fromager.bootstrapper.wheels.extract_info_from_wheel_file", |
| 404 | + return_value=("test_package", "1.0.0", None, None), |
| 405 | + ), |
| 406 | + patch( |
| 407 | + "fromager.bootstrapper.wheels.download_wheel", |
| 408 | + side_effect=requests.exceptions.ConnectionError("connection reset"), |
| 409 | + ), |
| 410 | + ): |
| 411 | + result = bt._download_wheel_from_cache( |
| 412 | + req=Requirement("test-package"), |
| 413 | + resolved_version=Version("1.0.0"), |
| 414 | + ) |
| 415 | + |
| 416 | + assert result == (None, None) |
| 417 | + log_record = next( |
| 418 | + r |
| 419 | + for r in caplog.records |
| 420 | + if "network error checking wheel cache" in r.message |
| 421 | + ) |
| 422 | + assert log_record.levelno == logging.WARNING |
| 423 | + |
| 424 | + def test_download_wheel_unexpected_error_returns_none_with_warning( |
| 425 | + self, tmp_context: WorkContext, caplog: pytest.LogCaptureFixture |
| 426 | + ) -> None: |
| 427 | + """Unexpected exception from download_wheel (after resolve succeeds) is caught.""" |
| 428 | + bt = bootstrapper.Bootstrapper(tmp_context) |
| 429 | + bt.cache_wheel_server_url = "https://cache.example.com/simple" |
| 430 | + |
| 431 | + with ( |
| 432 | + patch( |
| 433 | + "fromager.resolver.resolve", |
| 434 | + return_value=( |
| 435 | + "https://cache.example.com/simple/test-package/test_package-1.0.0-py3-none-any.whl", |
| 436 | + "1.0.0", |
| 437 | + ), |
| 438 | + ), |
| 439 | + patch( |
| 440 | + "fromager.bootstrapper.wheels.extract_info_from_wheel_file", |
| 441 | + return_value=("test_package", "1.0.0", None, None), |
| 442 | + ), |
| 443 | + patch( |
| 444 | + "fromager.bootstrapper.wheels.download_wheel", |
| 445 | + side_effect=OSError("disk full"), |
| 446 | + ), |
| 447 | + ): |
| 448 | + result = bt._download_wheel_from_cache( |
| 449 | + req=Requirement("test-package"), |
| 450 | + resolved_version=Version("1.0.0"), |
| 451 | + ) |
| 452 | + |
| 453 | + assert result == (None, None) |
| 454 | + log_record = next( |
| 455 | + r |
| 456 | + for r in caplog.records |
| 457 | + if "unexpected error checking wheel cache" in r.message |
| 458 | + ) |
| 459 | + assert log_record.levelno == logging.WARNING |
| 460 | + |
| 461 | + def test_no_cache_url_returns_none(self, tmp_context: WorkContext) -> None: |
| 462 | + """When no cache URL is configured, returns (None, None) immediately.""" |
| 463 | + bt = bootstrapper.Bootstrapper(tmp_context) |
| 464 | + bt.cache_wheel_server_url = "" |
| 465 | + |
| 466 | + result = bt._download_wheel_from_cache( |
| 467 | + req=Requirement("test-package"), |
| 468 | + resolved_version=Version("1.0.0"), |
| 469 | + ) |
| 470 | + |
| 471 | + assert result == (None, None) |
0 commit comments