Harden mail connector network boundaries
This commit is contained in:
@@ -1,16 +1,44 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import unittest
|
||||
from unittest.mock import patch
|
||||
|
||||
from govoplan_core.security.outbound_http import OutboundHttpBlocked
|
||||
from govoplan_mail.backend.config import SmtpConfig
|
||||
from govoplan_mail.backend.sending.smtp import (
|
||||
SmtpConfigurationError,
|
||||
_open_smtp,
|
||||
_prepare_smtp_send,
|
||||
_smtp_send_result,
|
||||
)
|
||||
|
||||
|
||||
class SmtpSendHelperTests(unittest.TestCase):
|
||||
def test_real_smtp_connections_honor_deployment_egress_policy(self):
|
||||
config = SmtpConfig(host="smtp.internal", port=587, security="starttls")
|
||||
with patch(
|
||||
"govoplan_mail.backend.sending.smtp.validate_outbound_host",
|
||||
side_effect=OutboundHttpBlocked("private network blocked"),
|
||||
), self.assertRaisesRegex(SmtpConfigurationError, "private network blocked"):
|
||||
_open_smtp(config)
|
||||
|
||||
def test_smtp_revalidates_and_pins_at_connection_time(self):
|
||||
config = SmtpConfig(host="smtp.example.test", port=587, security="starttls")
|
||||
public = [(2, 1, 6, "", ("93.184.216.34", 587))]
|
||||
private = [(2, 1, 6, "", ("127.0.0.1", 587))]
|
||||
with patch.dict(
|
||||
"os.environ",
|
||||
{"APP_ENV": "production", "GOVOPLAN_CONNECTOR_ALLOW_PRIVATE_NETWORKS": "false"},
|
||||
), patch(
|
||||
"govoplan_core.security.outbound_http.socket.getaddrinfo",
|
||||
side_effect=(public, private),
|
||||
), patch("govoplan_core.security.outbound_http.socket.socket") as socket_factory, self.assertRaisesRegex(
|
||||
OutboundHttpBlocked,
|
||||
"non-public network",
|
||||
):
|
||||
_open_smtp(config)
|
||||
socket_factory.assert_not_called()
|
||||
|
||||
def test_prepare_smtp_send_validates_envelope(self):
|
||||
config = SmtpConfig(host="smtp.example.org", port=587, security="starttls")
|
||||
with self.assertRaisesRegex(SmtpConfigurationError, "envelope sender"):
|
||||
@@ -45,4 +73,3 @@ class SmtpSendHelperTests(unittest.TestCase):
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user