From 1caf9a623a965f9fd984832438e9789148a60db4 Mon Sep 17 00:00:00 2001 From: 7000pctAUTO Date: Tue, 3 Feb 2026 10:39:15 +0000 Subject: [PATCH] fix: resolve CI import and type mismatch issues --- tests/test_models.py | 44 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 tests/test_models.py diff --git a/tests/test_models.py b/tests/test_models.py new file mode 100644 index 0000000..a6556de --- /dev/null +++ b/tests/test_models.py @@ -0,0 +1,44 @@ +"""Tests for core models.""" + +import pytest + +from src.core.models import Issue, IssueCategory, ScanResult, SeverityLevel + + +class TestModels: + """Tests for data models.""" + + def test_issue_creation(self): + """Test Issue dataclass creation.""" + issue = Issue( + category=IssueCategory.SECURITY, + severity=SeverityLevel.HIGH, + file_path="test.py", + line_number=10, + message="Security issue found", + ) + assert issue.category == IssueCategory.SECURITY + assert issue.severity == SeverityLevel.HIGH + assert issue.file_path == "test.py" + assert issue.line_number == 10 + assert issue.message == "Security issue found" + + def test_scan_result_creation(self): + """Test ScanResult dataclass creation.""" + result = ScanResult( + issues=[], + warnings=[], + files_scanned=5, + scan_time=1.5, + ) + assert result.issues == [] + assert result.warnings == [] + assert result.files_scanned == 5 + assert result.scan_time == 1.5 + + def test_severity_level_ordering(self): + """Test SeverityLevel enum values.""" + assert SeverityLevel.LOW.value == "low" + assert SeverityLevel.MEDIUM.value == "medium" + assert SeverityLevel.HIGH.value == "high" + assert SeverityLevel.CRITICAL.value == "critical"