Files
ds-chat/backend/tests/test_smtp_settings.py
ksmithandClaude Sonnet 5 691f597597 Fix SMTP sending forcing implicit TLS regardless of port
cfg.use_tls was passed straight through as aiosmtplib's use_tls kwarg,
which means implicit TLS -- encrypted from the first byte, port 465's
convention. Port 587, what most providers (including the one that
surfaced this: DreamHost) document as their primary submission port,
needs STARTTLS instead -- a plaintext connection that upgrades in-band.
Forcing implicit TLS against a STARTTLS-only port breaks the handshake
outright: [SSL: WRONG_VERSION_NUMBER], a client TLS ClientHello sent to a
server still expecting a plaintext SMTP greeting.

The "Use TLS" checkbox still means "encrypt this connection" -- the fix
infers which of the two negotiation modes to use from the port (465 ->
implicit, everything else -> STARTTLS), matching the convention every
mail client uses. start_tls is passed as an explicit requirement rather
than left to aiosmtplib's opportunistic default, so a server that turns
out not to support STARTTLS fails loudly instead of silently sending in
plaintext despite the admin asking for encryption.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
2026-08-15 23:09:39 -06:00

260 lines
8.2 KiB
Python

import uuid
from sqlalchemy import select
from app.crypto import decrypt
from app.models import SmtpSettings, User
from tests.conftest import register_and_login
async def _get_settings_row(db_session) -> SmtpSettings:
result = await db_session.execute(select(SmtpSettings))
return result.scalar_one()
async def _make_admin(db_session, user_id: str) -> None:
user = await db_session.get(User, uuid.UUID(user_id))
user.is_site_admin = True
await db_session.commit()
async def test_smtp_settings_require_admin(client, db_session):
await register_and_login(client, db_session, username="alice")
resp = await client.get("/api/admin/settings/smtp")
assert resp.status_code == 403
resp = await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 587,
"from_address": "noreply@example.com",
},
)
assert resp.status_code == 403
async def test_smtp_settings_get_before_configured(client, db_session):
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
resp = await client.get("/api/admin/settings/smtp")
assert resp.status_code == 200
assert resp.json() is None
async def test_smtp_settings_update_and_password_never_returned(client, db_session):
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
resp = await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 587,
"username": "bot",
"password": "super-secret",
"from_address": "noreply@example.com",
"use_tls": True,
},
)
assert resp.status_code == 200, resp.text
body = resp.json()
assert "password" not in body
assert body["has_password"] is True
assert body["host"] == "smtp.example.com"
get_resp = await client.get("/api/admin/settings/smtp")
assert get_resp.status_code == 200
assert "password" not in get_resp.json()
assert get_resp.json()["has_password"] is True
async def test_smtp_settings_password_encrypted_at_rest(client, db_session):
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 587,
"password": "super-secret",
"from_address": "noreply@example.com",
},
)
row = await _get_settings_row(db_session)
assert row.password_encrypted != "super-secret"
assert decrypt(row.password_encrypted) == "super-secret"
async def test_smtp_settings_blank_password_keeps_existing(client, db_session):
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 587,
"password": "first-password",
"from_address": "noreply@example.com",
},
)
encrypted_before = (await _get_settings_row(db_session)).password_encrypted
resp = await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 2525,
"from_address": "noreply@example.com",
},
)
assert resp.status_code == 200
assert resp.json()["port"] == 2525
assert resp.json()["has_password"] is True
row = await _get_settings_row(db_session)
assert row.password_encrypted == encrypted_before
async def test_send_test_email_not_configured(client, db_session):
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
resp = await client.post("/api/admin/settings/smtp/test")
assert resp.status_code == 400
async def test_send_test_email_success(client, db_session, monkeypatch):
calls = []
async def fake_send(message, **kwargs):
calls.append(kwargs)
monkeypatch.setattr("app.services.email_service.aiosmtplib.send", fake_send)
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 587,
"from_address": "noreply@example.com",
},
)
resp = await client.post("/api/admin/settings/smtp/test")
assert resp.status_code == 204
assert len(calls) == 1
assert calls[0]["hostname"] == "smtp.example.com"
async def test_send_test_email_port_587_uses_starttls_not_implicit_tls(client, db_session, monkeypatch):
# Regression test: port 587 (what most providers, e.g. DreamHost,
# document as their primary submission port) needs STARTTLS -- a
# plaintext connection that upgrades in-band -- not implicit TLS
# (encrypted from the first byte, port 465's convention). Passing
# cfg.use_tls straight through as aiosmtplib's `use_tls` kwarg forces
# implicit TLS regardless of port, which breaks the handshake outright
# against a STARTTLS-only server ([SSL: WRONG_VERSION_NUMBER]).
calls = []
async def fake_send(message, **kwargs):
calls.append(kwargs)
monkeypatch.setattr("app.services.email_service.aiosmtplib.send", fake_send)
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 587,
"from_address": "noreply@example.com",
"use_tls": True,
},
)
resp = await client.post("/api/admin/settings/smtp/test")
assert resp.status_code == 204
assert calls[0]["use_tls"] is False
assert calls[0]["start_tls"] is True
async def test_send_test_email_port_465_uses_implicit_tls(client, db_session, monkeypatch):
calls = []
async def fake_send(message, **kwargs):
calls.append(kwargs)
monkeypatch.setattr("app.services.email_service.aiosmtplib.send", fake_send)
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 465,
"from_address": "noreply@example.com",
"use_tls": True,
},
)
resp = await client.post("/api/admin/settings/smtp/test")
assert resp.status_code == 204
assert calls[0]["use_tls"] is True
assert calls[0]["start_tls"] is False
async def test_send_test_email_tls_disabled_uses_neither_mode(client, db_session, monkeypatch):
calls = []
async def fake_send(message, **kwargs):
calls.append(kwargs)
monkeypatch.setattr("app.services.email_service.aiosmtplib.send", fake_send)
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 25,
"from_address": "noreply@example.com",
"use_tls": False,
},
)
resp = await client.post("/api/admin/settings/smtp/test")
assert resp.status_code == 204
assert calls[0]["use_tls"] is False
assert calls[0]["start_tls"] is False
async def test_send_test_email_surfaces_failure(client, db_session, monkeypatch):
async def fake_send(message, **kwargs):
raise ConnectionRefusedError("boom")
monkeypatch.setattr("app.services.email_service.aiosmtplib.send", fake_send)
admin = await register_and_login(client, db_session, username="admin1")
await _make_admin(db_session, admin["id"])
await client.put(
"/api/admin/settings/smtp",
json={
"host": "smtp.example.com",
"port": 587,
"from_address": "noreply@example.com",
},
)
resp = await client.post("/api/admin/settings/smtp/test")
assert resp.status_code == 502