test(m4): cover the new security guards + pin e2e to exact MITRE v19 counts
- 5 new pytest covering paths the code-reviewer flagged as un-asserted:
* `test_seed_refuses_file_url` — `file://` scheme rejected before I/O
(was the SSRF-to-local-FS vector).
* `test_seed_refuses_disallowed_https_host` — non-allowlisted HTTPS
host rejected with `MitreSourceForbidden`.
* `test_seed_refuses_custom_url_without_sha` — end-to-end guard that
`seed_mitre(source=<custom URL>, expected_sha256=None,
allow_unverified=False)` raises `MitreSeedError`.
* `test_dotted_id_fallback_resolves_orphan_subtechnique` — STIX bundle
without `relationship[subtechnique-of]` still attaches T1059.001 to
T1059 via the dotted-id convention.
* `test_seed_clears_version_when_source_is_not_default` — seed from a
local path leaves `settings.mitre_version` NULL (no stale pin).
- Existing `test_checksum_mismatch_aborts` reworked to monkey-patch
`_ensure_host_allowed` so `file://` can drive the test past the allowlist
gate (was relying on file:// being accepted before CR1).
- Removed unused `uuid` import.
- e2e: assertions on `tactics_upserted`/`techniques_upserted`/
`subtechniques_upserted` switched from `>= 14/180/400` thresholds to
`=== 15/222/475` exact counts pinned to MITRE Enterprise v19.0 + 0
orphans. Catches parser regressions that would silently include revoked
rows. Bump alongside MITRE_VERSION when re-pinning.
- e2e: `Math.random()` → `crypto.randomUUID().slice(0, 8)` for unique
test-run emails (collision-safe across parallel CI workers).
DoD: 58 pytest pass (was 53), 34 Playwright pass.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -8,7 +8,6 @@ from __future__ import annotations
|
||||
|
||||
import json
|
||||
import secrets
|
||||
import uuid
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
@@ -245,12 +244,17 @@ def test_seed_persists_setting(app, fixture_bundle_path):
|
||||
assert status["version"] is None # only set when source == MITRE_DEFAULT_URL
|
||||
|
||||
|
||||
def test_checksum_mismatch_aborts(tmp_path):
|
||||
"""A wrong sha256 triggers MitreChecksumMismatch and skips DB writes."""
|
||||
def test_checksum_mismatch_aborts(tmp_path, monkeypatch):
|
||||
"""A wrong sha256 triggers MitreChecksumMismatch and skips DB writes.
|
||||
|
||||
We monkey-patch the allowlist to accept `file://` for the duration of the
|
||||
test — file:// is rejected in production by `_ensure_host_allowed` (cf.
|
||||
`test_seed_refuses_file_url`), but we need to drive `_download` past that
|
||||
gate to exercise the sha256 path.
|
||||
"""
|
||||
path = tmp_path / "tiny.json"
|
||||
path.write_text(json.dumps(MINIMAL_BUNDLE))
|
||||
# Force the URL path so download() is invoked. We mock by passing a file:// URL.
|
||||
# Simpler: call _download() directly with a bogus hash.
|
||||
monkeypatch.setattr(mitre_svc, "_ensure_host_allowed", lambda _: None)
|
||||
bogus = "0" * 64
|
||||
with pytest.raises(mitre_svc.MitreChecksumMismatch):
|
||||
mitre_svc._download(
|
||||
@@ -387,3 +391,63 @@ def test_matrix_endpoint_requires_auth(app, fixture_bundle_path):
|
||||
mitre_svc.seed_mitre(source=fixture_bundle_path, expected_sha256=None)
|
||||
with app.test_client() as c:
|
||||
assert c.get("/api/v1/mitre/matrix").status_code == 401
|
||||
|
||||
|
||||
# === Security guards ==========================================================
|
||||
|
||||
|
||||
def test_seed_refuses_file_url(tmp_path):
|
||||
"""file:// (or any scheme outside the allowlist) is rejected — protects
|
||||
against a privileged operator pivoting the in-container fetch to local
|
||||
filesystem reads via the URL path."""
|
||||
path = tmp_path / "bundle.json"
|
||||
path.write_text(json.dumps(MINIMAL_BUNDLE))
|
||||
with pytest.raises(mitre_svc.MitreSourceForbidden):
|
||||
mitre_svc._download(f"file://{path}", tmp_path / "out.json")
|
||||
|
||||
|
||||
def test_seed_refuses_disallowed_https_host(tmp_path):
|
||||
"""An HTTPS URL outside MITRE_ALLOWED_HOSTS is rejected before any I/O.
|
||||
Closes the SSRF surface (cloud metadata, internal mirrors)."""
|
||||
with pytest.raises(mitre_svc.MitreSourceForbidden):
|
||||
mitre_svc._download("https://attacker.example/bundle.json", tmp_path / "out.json")
|
||||
|
||||
|
||||
def test_seed_refuses_custom_url_without_sha(tmp_path):
|
||||
"""End-to-end refusal: even an allowlisted custom URL needs a sha or an
|
||||
explicit allow_unverified=True."""
|
||||
# Use the default URL with a different sha to simulate "custom" semantics
|
||||
# without actually hitting the network: pass a different MITRE_DEFAULT_URL.
|
||||
# The cleanest expression is to call seed_mitre with the same URL but no sha
|
||||
# — but the default URL gets the default sha auto-set; we need to bypass.
|
||||
with pytest.raises(mitre_svc.MitreSeedError):
|
||||
mitre_svc.seed_mitre(
|
||||
source="https://raw.githubusercontent.com/some-other-path/bundle.json",
|
||||
expected_sha256=None,
|
||||
allow_unverified=False,
|
||||
)
|
||||
|
||||
|
||||
def test_dotted_id_fallback_resolves_orphan_subtechnique(app, tmp_path):
|
||||
"""When the STIX `subtechnique-of` relationship is missing, the parser
|
||||
must fall back to the dotted convention (T1003.001 → T1003)."""
|
||||
bundle = json.loads(json.dumps(MINIMAL_BUNDLE)) # deep copy
|
||||
# Strip the relationship object so the parent_stix_id lookup fails.
|
||||
bundle["objects"] = [o for o in bundle["objects"] if o.get("type") != "relationship"]
|
||||
bundle_path = tmp_path / "no-rel.json"
|
||||
bundle_path.write_text(json.dumps(bundle))
|
||||
|
||||
result = mitre_svc.seed_mitre(source=bundle_path, expected_sha256=None)
|
||||
# The fallback resolves T1059.001 → T1059 via the dotted-id pattern,
|
||||
# so the subtechnique is still attached (no orphan).
|
||||
assert result.subtechniques_upserted == 1
|
||||
assert result.subtechniques_skipped_orphan == 0
|
||||
|
||||
|
||||
def test_seed_clears_version_when_source_is_not_default(app, fixture_bundle_path):
|
||||
"""A custom source must NULL `mitre_version` so /mitre/status doesn't lie
|
||||
about a stale upstream pin."""
|
||||
# First seed from the default URL would set version=19.0; here we seed from
|
||||
# a local file path, which should write version=None.
|
||||
mitre_svc.seed_mitre(source=fixture_bundle_path, expected_sha256=None)
|
||||
assert mitre_svc.read_status()["version"] is None
|
||||
|
||||
Reference in New Issue
Block a user