|
|
""" |
|
|
Tests for the EmailParser class. |
|
|
Run with: pytest tests/test_parser.py -v |
|
|
""" |
|
|
|
|
|
import pytest |
|
|
import sys |
|
|
from pathlib import Path |
|
|
|
|
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent / "src")) |
|
|
|
|
|
|
|
|
class TestEmailParserImport: |
|
|
"""Test that EmailParser can be imported.""" |
|
|
|
|
|
def test_import_parser(self): |
|
|
"""Test that parser module can be imported.""" |
|
|
try: |
|
|
from data.parser import EmailParser |
|
|
assert EmailParser is not None |
|
|
except ImportError as e: |
|
|
pytest.skip(f"Could not import EmailParser: {e}") |
|
|
|
|
|
|
|
|
class TestEmailParserMethods: |
|
|
"""Test EmailParser methods with mock data.""" |
|
|
|
|
|
def test_clean_text_removes_urls(self): |
|
|
"""Test URL removal in _clean_text.""" |
|
|
from data.parser import EmailParser |
|
|
|
|
|
|
|
|
class MockParser(EmailParser): |
|
|
def __init__(self): |
|
|
|
|
|
pass |
|
|
|
|
|
parser = MockParser() |
|
|
|
|
|
text = "Visit https://example.com for more info" |
|
|
result = parser._clean_text(text) |
|
|
assert "https://example.com" not in result |
|
|
|
|
|
def test_clean_text_normalizes_whitespace(self): |
|
|
"""Test whitespace normalization.""" |
|
|
from data.parser import EmailParser |
|
|
|
|
|
class MockParser(EmailParser): |
|
|
def __init__(self): |
|
|
pass |
|
|
|
|
|
parser = MockParser() |
|
|
|
|
|
text = "Hello World\n\nTest" |
|
|
result = parser._clean_text(text) |
|
|
assert " " not in result |
|
|
|
|
|
def test_decode_header_none(self): |
|
|
"""Test handling of None header.""" |
|
|
from data.parser import EmailParser |
|
|
|
|
|
class MockParser(EmailParser): |
|
|
def __init__(self): |
|
|
pass |
|
|
|
|
|
parser = MockParser() |
|
|
|
|
|
result = parser._decode_header(None) |
|
|
assert result == "" |
|
|
|
|
|
def test_decode_header_plain_string(self): |
|
|
"""Test decoding plain string header.""" |
|
|
from data.parser import EmailParser |
|
|
|
|
|
class MockParser(EmailParser): |
|
|
def __init__(self): |
|
|
pass |
|
|
|
|
|
parser = MockParser() |
|
|
|
|
|
result = parser._decode_header("Simple Subject") |
|
|
assert result == "Simple Subject" |
|
|
|
|
|
|
|
|
class TestEmailParserIntegration: |
|
|
"""Integration tests that require actual files.""" |
|
|
|
|
|
def test_parser_file_not_found(self): |
|
|
"""Test that missing file raises error.""" |
|
|
from data.parser import EmailParser |
|
|
|
|
|
with pytest.raises(FileNotFoundError): |
|
|
EmailParser(Path("/nonexistent/path/file.mbox")) |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
pytest.main([__file__, "-v"]) |
|
|
|
