|
|
|
|
@@ -5,16 +5,16 @@ from unittest.mock import patch
|
|
|
|
|
import pytest
|
|
|
|
|
from sqlalchemy import select
|
|
|
|
|
|
|
|
|
|
from guarddog_nexus.harvester import harvest
|
|
|
|
|
from guarddog_nexus.models import Finding
|
|
|
|
|
from guarddog_nexus.core.harvester import harvest
|
|
|
|
|
from guarddog_nexus.db.models import Finding
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
|
|
|
async def test_harvest_new_package(db_session, guarddog_normalized_flagged):
|
|
|
|
|
with (
|
|
|
|
|
patch("guarddog_nexus.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.harvester.scan_package") as mock_scan,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.scan_package") as mock_scan,
|
|
|
|
|
):
|
|
|
|
|
mock_dl.return_value = "/tmp/test-package.tar.gz"
|
|
|
|
|
mock_sha.return_value = "abc123"
|
|
|
|
|
@@ -50,9 +50,9 @@ async def test_harvest_new_package(db_session, guarddog_normalized_flagged):
|
|
|
|
|
async def test_harvest_same_sha256_skips(db_session, guarddog_normalized_flagged):
|
|
|
|
|
"""Same SHA256 as existing scan → skip, don't re-scan."""
|
|
|
|
|
with (
|
|
|
|
|
patch("guarddog_nexus.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.harvester.scan_package") as mock_scan,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.scan_package") as mock_scan,
|
|
|
|
|
):
|
|
|
|
|
mock_dl.return_value = "/tmp/test.tar.gz"
|
|
|
|
|
mock_sha.return_value = "deadbeef"
|
|
|
|
|
@@ -86,9 +86,9 @@ async def test_harvest_same_sha256_skips(db_session, guarddog_normalized_flagged
|
|
|
|
|
async def test_harvest_different_sha256_scans_again(db_session, guarddog_normalized_flagged):
|
|
|
|
|
"""Same name/version, different SHA256 → new scan."""
|
|
|
|
|
with (
|
|
|
|
|
patch("guarddog_nexus.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.harvester.scan_package") as mock_scan,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.scan_package") as mock_scan,
|
|
|
|
|
):
|
|
|
|
|
mock_dl.return_value = "/tmp/test.tar.gz"
|
|
|
|
|
mock_scan.return_value = guarddog_normalized_flagged
|
|
|
|
|
@@ -123,9 +123,9 @@ async def test_harvest_different_sha256_scans_again(db_session, guarddog_normali
|
|
|
|
|
async def test_harvest_skips_active_scan_same_url(db_session, guarddog_normalized_flagged):
|
|
|
|
|
"""Concurrent webhooks for same URL: first proceeding, second skips as PENDING."""
|
|
|
|
|
with (
|
|
|
|
|
patch("guarddog_nexus.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.harvester.scan_package") as mock_scan,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.scan_package") as mock_scan,
|
|
|
|
|
):
|
|
|
|
|
mock_dl.return_value = "/tmp/test.tar.gz"
|
|
|
|
|
mock_sha.return_value = "aaa"
|
|
|
|
|
@@ -147,9 +147,9 @@ async def test_harvest_skips_active_scan_same_url(db_session, guarddog_normalize
|
|
|
|
|
async def test_harvest_same_url_sha256_dedup(db_session, guarddog_normalized_flagged):
|
|
|
|
|
"""Same URL twice: second run hits SHA256 dedup (first already completed)."""
|
|
|
|
|
with (
|
|
|
|
|
patch("guarddog_nexus.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.harvester.scan_package") as mock_scan,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.scan_package") as mock_scan,
|
|
|
|
|
):
|
|
|
|
|
mock_dl.return_value = "/tmp/test.tar.gz"
|
|
|
|
|
mock_sha.return_value = "ccc"
|
|
|
|
|
@@ -182,9 +182,9 @@ async def test_harvest_same_url_sha256_dedup(db_session, guarddog_normalized_fla
|
|
|
|
|
@pytest.mark.asyncio
|
|
|
|
|
async def test_harvest_clean_package(db_session, guarddog_normalized_clean):
|
|
|
|
|
with (
|
|
|
|
|
patch("guarddog_nexus.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.harvester.scan_package") as mock_scan,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.download_asset") as mock_dl,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.compute_sha256") as mock_sha,
|
|
|
|
|
patch("guarddog_nexus.core.harvester.scan_package") as mock_scan,
|
|
|
|
|
):
|
|
|
|
|
mock_dl.return_value = "/tmp/test.tar.gz"
|
|
|
|
|
mock_sha.return_value = "abc"
|
|
|
|
|
@@ -205,7 +205,7 @@ async def test_harvest_clean_package(db_session, guarddog_normalized_clean):
|
|
|
|
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
|
|
|
async def test_harvest_download_failure(db_session):
|
|
|
|
|
with patch("guarddog_nexus.harvester.download_asset") as mock_dl:
|
|
|
|
|
with patch("guarddog_nexus.core.harvester.download_asset") as mock_dl:
|
|
|
|
|
mock_dl.return_value = None
|
|
|
|
|
|
|
|
|
|
scan = await harvest(
|
|
|
|
|
|