Implement imap_content event for imap integration (#90242)

2023-03-28 21:02:43 +02:00 · 2023-03-28 21:02:43 +02:00 · 24d0d15f38
commit 24d0d15f38
parent d21433b6af
5 changed files with 671 additions and 9 deletions
--- a/tests/components/imap/const.py
+++ b/tests/components/imap/const.py
@ -0,0 +1,139 @@
+"""Constants for tests imap integration."""
+
+TEST_MESSAGE = (
+    b"Return-Path: <john.doe@example.com>\r\nDelivered-To: notify@example.com\r\n"
+    b"Received: from beta.example.com\r\n\tby beta with LMTP\r\n\t"
+    b"id eLp2M/GcHWQTLxQAho4UZQ\r\n\t(envelope-from <john.doe@example.com>)\r\n\t"
+    b"for <notify@example.com>; Fri, 24 Mar 2023 13:52:01 +0100\r\n"
+    b"Received: from localhost (localhost [127.0.0.1])\r\n\t"
+    b"by beta.example.com (Postfix) with ESMTP id D0FFA61425\r\n\t"
+    b"for <notify@example.com>; Fri, 24 Mar 2023 13:52:01 +0100 (CET)\r\n"
+    b"Date: Fri, 24 Mar 2023 13:52:00 +0100\r\n"
+    b"MIME-Version: 1.0\r\n"
+    b"To: notify@example.com\r\n"
+    b"From: John Doe <john.doe@example.com>\r\n"
+    b"Subject: Test subject\r\n"
+)
+
+TEST_CONTENT_TEXT_BARE = b"\r\n" b"Test body\r\n" b"\r\n"
+
+TEST_CONTENT_BINARY = (
+    b"Content-Type: application/binary\r\n"
+    b"Content-Transfer-Encoding: base64\r\n"
+    b"\r\n"
+    b"VGVzdCBib2R5\r\n"
+)
+
+TEST_CONTENT_TEXT_PLAIN = (
+    b"Content-Type: text/plain; charset=UTF-8; format=flowed\r\n"
+    b"Content-Transfer-Encoding: 7bit\r\n\r\nTest body\r\n\r\n"
+)
+
+TEST_CONTENT_TEXT_OTHER = (
+    b"Content-Type: text/other; charset=UTF-8\r\n"
+    b"Content-Transfer-Encoding: 7bit\r\n\r\nTest body\r\n\r\n"
+)
+
+TEST_CONTENT_HTML = (
+    b"Content-Type: text/html; charset=UTF-8\r\n"
+    b"Content-Transfer-Encoding: 7bit\r\n"
+    b"\r\n"
+    b"<html>\r\n"
+    b"  <head>\r\n"
+    b'    <meta http-equiv="content-type" content="text/html; charset=UTF-8">\r\n'
+    b"  </head>\r\n"
+    b"  <body>\r\n"
+    b"    <p>Test body<br>\r\n"
+    b"    </p>\r\n"
+    b"  </body>\r\n"
+    b"</html>\r\n"
+    b"\r\n"
+)
+
+TEST_CONTENT_MULTIPART = (
+    b"\r\nThis is a multi-part message in MIME format.\r\n"
+    + b"--------------McwBciN2C0o3rWeF1tmFo2oI\r\n"
+    + TEST_CONTENT_TEXT_PLAIN
+    + b"--------------McwBciN2C0o3rWeF1tmFo2oI\r\n"
+    + TEST_CONTENT_HTML
+    + b"--------------McwBciN2C0o3rWeF1tmFo2oI--\r\n"
+)
+
+EMPTY_SEARCH_RESPONSE = ("OK", [b"", b"Search completed (0.0001 + 0.000 secs)."])
+BAD_RESPONSE = ("BAD", [b"", b"Unexpected error"])
+
+TEST_SEARCH_RESPONSE = ("OK", [b"1", b"Search completed (0.0001 + 0.000 secs)."])
+
+TEST_FETCH_RESPONSE_TEXT_BARE = (
+    "OK",
+    [
+        b"1 FETCH (BODY[] {"
+        + str(len(TEST_MESSAGE + TEST_CONTENT_TEXT_BARE)).encode("utf-8")
+        + b"}",
+        bytearray(TEST_MESSAGE + TEST_CONTENT_TEXT_BARE),
+        b")",
+        b"Fetch completed (0.0001 + 0.000 secs).",
+    ],
+)
+
+TEST_FETCH_RESPONSE_TEXT_PLAIN = (
+    "OK",
+    [
+        b"1 FETCH (BODY[] {"
+        + str(len(TEST_MESSAGE + TEST_CONTENT_TEXT_PLAIN)).encode("utf-8")
+        + b"}",
+        bytearray(TEST_MESSAGE + TEST_CONTENT_TEXT_PLAIN),
+        b")",
+        b"Fetch completed (0.0001 + 0.000 secs).",
+    ],
+)
+
+TEST_FETCH_RESPONSE_TEXT_OTHER = (
+    "OK",
+    [
+        b"1 FETCH (BODY[] {"
+        + str(len(TEST_MESSAGE + TEST_CONTENT_TEXT_OTHER)).encode("utf-8")
+        + b"}",
+        bytearray(TEST_MESSAGE + TEST_CONTENT_TEXT_OTHER),
+        b")",
+        b"Fetch completed (0.0001 + 0.000 secs).",
+    ],
+)
+
+TEST_FETCH_RESPONSE_BINARY = (
+    "OK",
+    [
+        b"1 FETCH (BODY[] {"
+        + str(len(TEST_MESSAGE + TEST_CONTENT_BINARY)).encode("utf-8")
+        + b"}",
+        bytearray(TEST_MESSAGE + TEST_CONTENT_BINARY),
+        b")",
+        b"Fetch completed (0.0001 + 0.000 secs).",
+    ],
+)
+
+TEST_FETCH_RESPONSE_HTML = (
+    "OK",
+    [
+        b"1 FETCH (BODY[] {"
+        + str(len(TEST_MESSAGE + TEST_CONTENT_HTML)).encode("utf-8")
+        + b"}",
+        bytearray(TEST_MESSAGE + TEST_CONTENT_HTML),
+        b")",
+        b"Fetch completed (0.0001 + 0.000 secs).",
+    ],
+)
+
+TEST_FETCH_RESPONSE_MULTIPART = (
+    "OK",
+    [
+        b"1 FETCH (BODY[] {"
+        + str(len(TEST_MESSAGE + TEST_CONTENT_MULTIPART)).encode("utf-8")
+        + b"}",
+        bytearray(TEST_MESSAGE + TEST_CONTENT_MULTIPART),
+        b")",
+        b"Fetch completed (0.0001 + 0.000 secs).",
+    ],
+)
+
+RESPONSE_BAD = ("BAD", [])