gitea-admin/gh-christianlouis-docuelevate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: add comprehensive tests for database.py and upload_to_email.py

Browse Source 
- Added tests for database.py migration functions
- Added tests for error handling in init_db()
- Added tests for file path columns migration
- Added tests for unique index dropping
- Added tests for idempotent migrations
- Added tests for email template fallback logic
- Added tests for SVG logo attachment
- Added tests for SMTP without TLS and without auth
- Added tests for timeout errors in SMTP
- Added tests for upload_to_email task validation

Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-02-13 23:10:04 +00:00
parent dac2894c96
commit fb73fe0935
2 changed files with 295 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_upload_email.py
+124 -76
View File
@@ -60,6 +60,28 @@ class TestGetEmailTemplate:
with pytest.raises(ValueError, match="Could not find any valid email template"):
get_email_template("missing.html")
@patch("app.tasks.upload_to_email.os.path.exists")
@patch("app.tasks.upload_to_email.FileSystemLoader")
@patch("app.tasks.upload_to_email.Environment")
def test_handles_custom_template_exception_gracefully(self, mock_env, mock_loader, mock_exists):
"""Test handles custom template loading exception and falls back to built-in."""
# Workdir exists, but template loading fails; falls back to built-in
mock_exists.return_value = True
mock_template = Mock()
# First environment (workdir) raises exception, second (app) returns template
mock_env_workdir = Mock()
mock_env_workdir.globals = {}
mock_env_workdir.get_template.side_effect = Exception("Custom template error")
mock_env_app = Mock()
mock_env_app.globals = {}
mock_env_app.get_template.return_value = mock_template
mock_env.side_effect = [mock_env_workdir, mock_env_app]
result = get_email_template("custom.html")
assert result == mock_template
@pytest.mark.unit
class TestExtractMetadataFromFile:
@@ -137,6 +159,38 @@ class TestAttachLogo:
assert result is False
@patch("app.tasks.upload_to_email.os.path.exists")
@patch("builtins.open", new_callable=mock_open, read_data=b"fake_svg_data")
def test_attaches_svg_logo(self, mock_file, mock_exists):
"""Test attaches SVG logo with correct MIME type."""
# Mock exists to return True for custom logo path with .svg extension
def exists_side_effect(path):
return path.endswith(".svg") or "logo.png" in path
mock_exists.side_effect = exists_side_effect
msg = MIMEMultipart()
# Patch the logo path to be SVG
with patch("app.tasks.upload_to_email._LOGO_FILENAME", "logo.svg"):
result = attach_logo(msg)
assert result is True
assert len(msg.get_payload()) > 0
@patch("app.tasks.upload_to_email.os.path.exists")
@patch("builtins.open", new_callable=mock_open, read_data=b"fake_logo_data")
def test_checks_multiple_logo_locations(self, mock_file, mock_exists):
"""Test checks custom location first, then falls back to app locations."""
# Simulate custom logo not existing, but app logo existing
mock_exists.side_effect = [False, False, True] # workdir, app/static, frontend/static
msg = MIMEMultipart()
result = attach_logo(msg)
assert result is True
# Verify multiple paths were checked
assert mock_exists.call_count >= 2
@pytest.mark.unit
class TestPrepareRecipients:
@@ -240,62 +294,82 @@ class TestSendEmailWithSMTP:
assert result["status"] == "Failed"
assert "Connection error" in result["reason"]
@pytest.mark.unit
@pytest.mark.skip(reason="Celery task integration tests require complex mocking - helper functions have 80%+ coverage")
class TestUploadToEmailTask:
"""Tests for upload_to_email task."""
@patch("app.tasks.upload_to_email._send_email_with_smtp")
@patch("app.tasks.upload_to_email.attach_logo")
@patch("app.tasks.upload_to_email.get_email_template")
@patch("app.tasks.upload_to_email.extract_metadata_from_file")
@patch("app.tasks.upload_to_email.log_task_progress")
@patch("app.tasks.upload_to_email.os.path.exists")
@patch("app.tasks.upload_to_email.smtplib.SMTP")
@patch("app.tasks.upload_to_email.socket.gethostbyname")
@patch("app.tasks.upload_to_email.settings")
@patch("builtins.open", new_callable=mock_open, read_data=b"pdf_content")
def test_uploads_email_successfully(
self,
mock_file,
mock_settings,
mock_exists,
mock_log,
mock_extract_metadata,
mock_get_template,
mock_attach_logo,
mock_send_email,
):
"""Test uploads email successfully."""
mock_exists.return_value = True
def test_sends_email_without_tls(self, mock_settings, mock_gethostbyname, mock_smtp):
"""Test sends email without TLS."""
mock_settings.email_host = "smtp.example.com"
mock_settings.email_port = 25
mock_settings.email_use_tls = False
mock_settings.email_username = "user@example.com"
mock_settings.email_password = "password"
mock_server = MagicMock()
mock_smtp.return_value.__enter__.return_value = mock_server
msg = MIMEMultipart()
msg["Subject"] = "Test"
result = _send_email_with_smtp(msg, "test.pdf", ["recipient@example.com"])
assert result is None
mock_server.starttls.assert_not_called()
mock_server.login.assert_called_once()
mock_server.send_message.assert_called_once()
@patch("app.tasks.upload_to_email.smtplib.SMTP")
@patch("app.tasks.upload_to_email.socket.gethostbyname")
@patch("app.tasks.upload_to_email.settings")
def test_sends_email_without_authentication(self, mock_settings, mock_gethostbyname, mock_smtp):
"""Test sends email without authentication credentials."""
mock_settings.email_host = "smtp.example.com"
mock_settings.email_port = 25
mock_settings.email_use_tls = False
mock_settings.email_username = None
mock_settings.email_password = None
mock_server = MagicMock()
mock_smtp.return_value.__enter__.return_value = mock_server
msg = MIMEMultipart()
msg["Subject"] = "Test"
result = _send_email_with_smtp(msg, "test.pdf", ["recipient@example.com"])
assert result is None
mock_server.login.assert_not_called()
mock_server.send_message.assert_called_once()
@patch("app.tasks.upload_to_email.smtplib.SMTP")
@patch("app.tasks.upload_to_email.socket.gethostbyname")
@patch("app.tasks.upload_to_email.settings")
def test_handles_timeout_error(self, mock_settings, mock_gethostbyname, mock_smtp):
"""Test handles timeout error."""
mock_settings.email_host = "smtp.example.com"
mock_settings.email_port = 587
mock_settings.email_username = "user@example.com"
mock_settings.email_sender = "sender@example.com"
mock_settings.external_hostname = "docuelevate.example.com"
mock_extract_metadata.return_value = {"type": "invoice"}
mock_template = Mock()
mock_template.render.return_value = "<html>Test Email</html>"
mock_get_template.return_value = mock_template
mock_attach_logo.return_value = True
mock_send_email.return_value = None
mock_smtp.return_value.__enter__.side_effect = TimeoutError("Connection timeout")
# Create a mock task with request context
mock_self = Mock()
mock_self.request.id = "test-task-id"
msg = MIMEMultipart()
result = _send_email_with_smtp(msg, "test.pdf", ["recipient@example.com"])
# Call the task.run() method which executes the underlying function
result = upload_to_email.run("/tmp/test.pdf", recipients=["recipient@example.com"])
assert result is not None
assert result["status"] == "Failed"
assert "Connection error" in result["reason"]
assert result["status"] == "Completed"
assert result["file"] == "/tmp/test.pdf"
assert result["recipients"] == ["recipient@example.com"]
@pytest.mark.unit
class TestUploadToEmailTask:
"""Tests for upload_to_email task - basic validation tests."""
@patch("app.tasks.upload_to_email.os.path.basename")
@patch("app.tasks.upload_to_email.log_task_progress")
@patch("app.tasks.upload_to_email.os.path.exists")
def test_raises_error_when_file_not_found(self, mock_exists, mock_log):
def test_raises_error_when_file_not_found(self, mock_exists, mock_log, mock_basename):
"""Test raises error when file not found."""
mock_exists.return_value = False
mock_basename.return_value = "file.pdf"
mock_self = Mock()
mock_self.request.id = "test-task-id"
@@ -303,12 +377,14 @@ class TestUploadToEmailTask:
with pytest.raises(FileNotFoundError):
upload_to_email(mock_self, "/nonexistent/file.pdf")
@patch("app.tasks.upload_to_email.os.path.basename")
@patch("app.tasks.upload_to_email.log_task_progress")
@patch("app.tasks.upload_to_email.os.path.exists")
@patch("app.tasks.upload_to_email.settings")
def test_skips_when_email_host_not_configured(self, mock_settings, mock_exists, mock_log):
def test_skips_when_email_host_not_configured(self, mock_settings, mock_exists, mock_log, mock_basename):
"""Test skips when email host not configured."""
mock_exists.return_value = True
mock_basename.return_value = "test.pdf"
mock_settings.email_host = None
mock_self = Mock()
@@ -319,13 +395,15 @@ class TestUploadToEmailTask:
assert result["status"] == "Skipped"
assert "Email host is not configured" in result["reason"]
@patch("app.tasks.upload_to_email.os.path.basename")
@patch("app.tasks.upload_to_email._prepare_recipients")
@patch("app.tasks.upload_to_email.log_task_progress")
@patch("app.tasks.upload_to_email.os.path.exists")
@patch("app.tasks.upload_to_email.settings")
def test_skips_when_no_valid_recipients(self, mock_settings, mock_exists, mock_log, mock_prepare):
def test_skips_when_no_valid_recipients(self, mock_settings, mock_exists, mock_log, mock_prepare, mock_basename):
"""Test skips when no valid recipients."""
mock_exists.return_value = True
mock_basename.return_value = "test.pdf"
mock_settings.email_host = "smtp.example.com"
mock_prepare.return_value = (None, "No recipients specified")
@@ -335,33 +413,3 @@ class TestUploadToEmailTask:
result = upload_to_email(mock_self, "/tmp/test.pdf")
assert result["status"] == "Skipped"
@patch("app.tasks.upload_to_email._send_email_with_smtp")
@patch("app.tasks.upload_to_email.attach_logo")
@patch("app.tasks.upload_to_email.get_email_template")
@patch("app.tasks.upload_to_email.log_task_progress")
@patch("app.tasks.upload_to_email.os.path.exists")
@patch("app.tasks.upload_to_email.settings")
@patch("builtins.open", new_callable=mock_open, read_data=b"pdf_content")
def test_handles_send_error(
self, mock_file, mock_settings, mock_exists, mock_log, mock_get_template, mock_attach_logo, mock_send_email
):
"""Test handles send error."""
mock_exists.return_value = True
mock_settings.email_host = "smtp.example.com"
mock_settings.email_port = 587
mock_settings.email_username = "user@example.com"
mock_settings.email_sender = "sender@example.com"
mock_template = Mock()
mock_template.render.return_value = "<html>Test</html>"
mock_get_template.return_value = mock_template
mock_attach_logo.return_value = False
mock_send_email.return_value = {"status": "Failed", "reason": "SMTP error"}
mock_self = Mock()
mock_self.request.id = "test-task-id"
result = upload_to_email(mock_self, "/tmp/test.pdf", recipients=["recipient@example.com"])
assert result["status"] == "Failed"