"""The tests for the notify smtp platform.""" import re import unittest from homeassistant.components.smtp.notify import MailNotificationService from tests.async_mock import patch from tests.common import get_test_home_assistant class MockSMTP(MailNotificationService): """Test SMTP object that doesn't need a working server.""" def _send_email(self, msg): """Just return string for testing.""" return msg.as_string() class TestNotifySmtp(unittest.TestCase): """Test the smtp notify.""" def setUp(self): # pylint: disable=invalid-name """Set up things to be run when tests are started.""" self.hass = get_test_home_assistant() self.mailer = MockSMTP( "localhost", 25, 5, "test@test.com", 1, "testuser", "testpass", ["recip1@example.com", "testrecip@test.com"], "Home Assistant", 0, ) self.addCleanup(self.tear_down_cleanup) def tear_down_cleanup(self): """Stop down everything that was started.""" self.hass.stop() @patch("email.utils.make_msgid", return_value="") def test_text_email(self, mock_make_msgid): """Test build of default text email behavior.""" msg = self.mailer.send_message("Test msg") expected = ( '^Content-Type: text/plain; charset="us-ascii"\n' "MIME-Version: 1.0\n" "Content-Transfer-Encoding: 7bit\n" "Subject: Home Assistant\n" "To: recip1@example.com,testrecip@test.com\n" "From: Home Assistant \n" "X-Mailer: Home Assistant\n" "Date: [^\n]+\n" "Message-Id: <[^@]+@[^>]+>\n" "\n" "Test msg$" ) assert re.search(expected, msg) @patch("email.utils.make_msgid", return_value="") def test_mixed_email(self, mock_make_msgid): """Test build of mixed text email behavior.""" msg = self.mailer.send_message("Test msg", data={"images": ["test.jpg"]}) assert "Content-Type: multipart/related" in msg @patch("email.utils.make_msgid", return_value="") def test_html_email(self, mock_make_msgid): """Test build of html email behavior.""" html = """

Intruder alert at apartment!!

""" msg = self.mailer.send_message( "Test msg", data={"html": html, "images": ["test.jpg"]} ) assert "Content-Type: multipart/related" in msg