"""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"