From c68f5c9255eaccffa6df8129810eb9f7c8f247de Mon Sep 17 00:00:00 2001 From: tduhamel42 Date: Fri, 14 Nov 2025 15:45:23 +0100 Subject: [PATCH] fix: Import FoundBy from correct module in tests Fixed Pydantic validation error by importing FoundBy from modules.base instead of models.finding_schema. The ModuleFinding class in base.py uses its own FoundBy definition, and Pydantic requires the exact same class instance for validation to pass. This resolves the validation errors: - test_cargo_fuzzer.py::test_create_finding_from_crash - test_atheris_fuzzer.py::test_create_crash_finding --- backend/tests/unit/test_modules/test_atheris_fuzzer.py | 2 +- backend/tests/unit/test_modules/test_cargo_fuzzer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/tests/unit/test_modules/test_atheris_fuzzer.py b/backend/tests/unit/test_modules/test_atheris_fuzzer.py index 0efcc76..8873aa6 100644 --- a/backend/tests/unit/test_modules/test_atheris_fuzzer.py +++ b/backend/tests/unit/test_modules/test_atheris_fuzzer.py @@ -159,7 +159,7 @@ class TestAtherisFuzzerFindingGeneration: async def test_create_crash_finding(self, atheris_fuzzer): """Test crash finding creation""" - from models.finding_schema import FoundBy + from modules.base import FoundBy finding = atheris_fuzzer.create_finding( rule_id="atheris-crash", diff --git a/backend/tests/unit/test_modules/test_cargo_fuzzer.py b/backend/tests/unit/test_modules/test_cargo_fuzzer.py index 5b79ad7..ef20b77 100644 --- a/backend/tests/unit/test_modules/test_cargo_fuzzer.py +++ b/backend/tests/unit/test_modules/test_cargo_fuzzer.py @@ -161,7 +161,7 @@ class TestCargoFuzzerFindingGeneration: async def test_create_finding_from_crash(self, cargo_fuzzer): """Test finding creation""" - from models.finding_schema import FoundBy + from modules.base import FoundBy finding = cargo_fuzzer.create_finding( rule_id="cargo-fuzz-crash",