96 lines
4.0 KiB
Python
96 lines
4.0 KiB
Python
"""Tests for pz_parser phase 3 — mod attribution."""
|
|
from __future__ import annotations
|
|
|
|
import pathlib
|
|
import sys
|
|
import unittest
|
|
|
|
sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[1]))
|
|
|
|
import pz_parser # noqa: E402
|
|
|
|
FIXTURE_DIR = pathlib.Path(__file__).resolve().parent / "fixtures"
|
|
|
|
|
|
def fixture(name: str) -> pathlib.Path:
|
|
return FIXTURE_DIR / name
|
|
|
|
|
|
class AttributionBucketTests(unittest.TestCase):
|
|
"""Three confidence buckets: direct (high), inferred (medium),
|
|
unattributed (low)."""
|
|
|
|
def test_direct_attribution_when_lua_marker_on_entry(self) -> None:
|
|
entries = pz_parser.parse_file(fixture("fixture_lua_attributed.txt"))
|
|
records = pz_parser.classify_entries(entries, source_file="la.txt")
|
|
self.assertEqual(len(records), 1)
|
|
rec = records[0]
|
|
self.assertEqual(rec.attribution, "direct")
|
|
self.assertEqual(rec.confidence, "high")
|
|
# mod_id is normalised: lowercase, no spaces / apostrophes / hyphens.
|
|
self.assertEqual(rec.mod_id, "testmodalpha")
|
|
self.assertEqual(rec.mod_name, "Test Mod Alpha")
|
|
|
|
def test_inferred_attribution_within_lookback_window(self) -> None:
|
|
entries = pz_parser.parse_file(fixture("fixture_inferred.txt"))
|
|
records = pz_parser.classify_entries(entries, source_file="in.txt")
|
|
self.assertEqual(len(records), 1)
|
|
rec = records[0]
|
|
self.assertEqual(rec.attribution, "inferred")
|
|
self.assertEqual(rec.confidence, "medium")
|
|
self.assertEqual(rec.mod_id, "spongiesclothing")
|
|
|
|
def test_unattributed_when_no_marker_and_not_lua_shaped(self) -> None:
|
|
entries = pz_parser.parse_file(fixture("fixture_unattributed.txt"))
|
|
records = pz_parser.classify_entries(entries, source_file="ua.txt")
|
|
self.assertEqual(len(records), 1)
|
|
rec = records[0]
|
|
self.assertEqual(rec.attribution, "unattributed")
|
|
self.assertEqual(rec.confidence, "low")
|
|
self.assertEqual(rec.mod_id, "__unattributed__")
|
|
|
|
|
|
class LookbackBoundaryTests(unittest.TestCase):
|
|
"""Phase 3 — 40-line inferred-attribution window boundary."""
|
|
|
|
def test_lua_marker_beyond_lookback_does_not_attribute(self) -> None:
|
|
# Fixture places the Lua((MOD:...)) >40 lines before the ERROR.
|
|
entries = pz_parser.parse_file(fixture("fixture_lookback_boundary.txt"))
|
|
records = pz_parser.classify_entries(entries, source_file="lb.txt")
|
|
self.assertEqual(len(records), 1)
|
|
rec = records[0]
|
|
# The Lua-shaped ERROR is far enough back to be unattributed.
|
|
self.assertEqual(rec.attribution, "unattributed")
|
|
self.assertEqual(rec.mod_id, "__unattributed__")
|
|
|
|
def test_non_lua_shaped_body_rejects_inferred_attribution(self) -> None:
|
|
# Recent Lua((MOD:Spongies Clothing)) emitted, but the ERROR body
|
|
# ("Disk full while writing chunk data") isn't Lua-shaped.
|
|
entries = pz_parser.parse_file(fixture("fixture_non_lua_no_inferred.txt"))
|
|
records = pz_parser.classify_entries(entries, source_file="nl.txt")
|
|
self.assertEqual(len(records), 1)
|
|
rec = records[0]
|
|
self.assertEqual(rec.attribution, "unattributed")
|
|
|
|
|
|
class NeededByTests(unittest.TestCase):
|
|
"""Phase 3 — direct attribution via "needed by <mod>" hint."""
|
|
|
|
def test_needed_by_extracts_dependent_mod(self) -> None:
|
|
entries = pz_parser.parse_file(fixture("fixture_require_failed.txt"))
|
|
records = pz_parser.classify_entries(entries, source_file="rf.txt")
|
|
self.assertEqual(len(records), 1)
|
|
rec = records[0]
|
|
# "needed by Test Mod Alpha" should set the mod to Test Mod Alpha
|
|
# (preferred over the require("...") side which would mention
|
|
# DependencyMod). Either way we want direct/high.
|
|
self.assertEqual(rec.attribution, "direct")
|
|
self.assertEqual(rec.confidence, "high")
|
|
# The "needed by" branch is checked before the require() branch in
|
|
# the priority order; mod_id should reflect Test Mod Alpha.
|
|
self.assertEqual(rec.mod_id, "testmodalpha")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|