chore: import upstream snapshot with attribution

python/tests/core/email/__init__.py

@@ -0,0 +1,13 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========

python/tests/core/email/test_client.py

@@ -0,0 +1,102 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from unittest.mock import AsyncMock, MagicMock
+
+import pytest
+
+from mirage.accessor.email import EmailAccessor
+from mirage.core.email._client import list_folders, list_message_uids
+from mirage.resource.email.config import EmailConfig
+
+
+@pytest.fixture
+def config():
+    return EmailConfig(
+        imap_host="imap.test.com",
+        smtp_host="smtp.test.com",
+        username="user@test.com",
+        password="pass",
+    )
+
+
+@pytest.fixture
+def accessor(config):
+    return EmailAccessor(config)
+
+
+@pytest.mark.asyncio
+async def test_list_folders(accessor):
+    mock_imap = AsyncMock()
+    mock_response = MagicMock()
+    mock_response.lines = [
+        b'(\\HasNoChildren) "/" "INBOX"',
+        b'(\\HasNoChildren) "/" "Sent"',
+        b'(\\HasNoChildren) "/" "Drafts"',
+    ]
+    mock_imap.list.return_value = mock_response
+    accessor._imap = mock_imap
+    accessor._imap.protocol = True
+
+    folders = await list_folders(accessor)
+    assert "INBOX" in folders
+    assert "Sent" in folders
+    assert "Drafts" in folders
+
+
+@pytest.mark.asyncio
+async def test_list_message_uids(accessor):
+    mock_imap = AsyncMock()
+    mock_select_response = MagicMock()
+    mock_select_response.result = "OK"
+    mock_imap.select.return_value = mock_select_response
+
+    mock_search_response = MagicMock()
+    mock_search_response.result = "OK"
+    mock_search_response.lines = [b"1 2 3 4 5"]
+    mock_imap.search.return_value = mock_search_response
+
+    mock_fetch_response = MagicMock()
+    mock_fetch_response.lines = [
+        b"1 FETCH (UID 101)",
+        b"2 FETCH (UID 102)",
+        b"3 FETCH (UID 103)",
+        b"4 FETCH (UID 104)",
+        b"5 FETCH (UID 105)",
+        b"FETCH completed",
+    ]
+    mock_imap.fetch.return_value = mock_fetch_response
+    accessor._imap = mock_imap
+    accessor._imap.protocol = True
+
+    uids = await list_message_uids(accessor, "INBOX")
+    assert uids == ["101", "102", "103", "104", "105"]
+
+
+@pytest.mark.asyncio
+async def test_list_message_uids_empty(accessor):
+    mock_imap = AsyncMock()
+    mock_select_response = MagicMock()
+    mock_select_response.result = "OK"
+    mock_imap.select.return_value = mock_select_response
+
+    mock_search_response = MagicMock()
+    mock_search_response.result = "OK"
+    mock_search_response.lines = [b""]
+    mock_imap.search.return_value = mock_search_response
+    accessor._imap = mock_imap
+    accessor._imap.protocol = True
+
+    uids = await list_message_uids(accessor, "INBOX")
+    assert uids == []

python/tests/core/email/test_parse.py

@@ -0,0 +1,99 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+
+from mirage.core.email._parse import parse_rfc822
+
+
+def test_parse_simple_text_email():
+    raw = (b"From: Alice <alice@example.com>\r\n"
+           b"To: Bob <bob@example.com>\r\n"
+           b"Subject: Hello\r\n"
+           b"Date: Mon, 14 Apr 2026 10:30:00 +0000\r\n"
+           b"Message-ID: <abc123@example.com>\r\n"
+           b"\r\n"
+           b"Hello, world!")
+    result = parse_rfc822(raw)
+    assert result["from"] == {"name": "Alice", "email": "alice@example.com"}
+    assert result["to"] == [{"name": "Bob", "email": "bob@example.com"}]
+    assert result["subject"] == "Hello"
+    assert result["body_text"] == "Hello, world!"
+    assert result["message_id"] == "<abc123@example.com>"
+    assert result["has_attachments"] is False
+
+
+def test_parse_multipart_email():
+    msg = MIMEMultipart()
+    msg["From"] = "Alice <alice@example.com>"
+    msg["To"] = "Bob <bob@example.com>"
+    msg["Subject"] = "With HTML"
+    msg["Message-ID"] = "<def456@example.com>"
+    msg.attach(MIMEText("Plain text body", "plain"))
+    msg.attach(MIMEText("<p>HTML body</p>", "html"))
+    raw = msg.as_bytes()
+
+    result = parse_rfc822(raw)
+    assert result["body_text"] == "Plain text body"
+    assert result["body_html"] == "<p>HTML body</p>"
+    assert result["has_attachments"] is False
+
+
+def test_parse_headers_only():
+    raw = (b"From: Alice <alice@example.com>\r\n"
+           b"Subject: Test\r\n"
+           b"\r\n"
+           b"Body text here")
+    result = parse_rfc822(raw, headers_only=True)
+    assert result["subject"] == "Test"
+    assert result["body_text"] == ""
+
+
+def test_parse_reply_headers():
+    raw = (b"From: Bob <bob@example.com>\r\n"
+           b"To: Alice <alice@example.com>\r\n"
+           b"Subject: Re: Hello\r\n"
+           b"In-Reply-To: <abc123@example.com>\r\n"
+           b"References: <abc123@example.com> <def456@example.com>\r\n"
+           b"\r\n"
+           b"Thanks!")
+    result = parse_rfc822(raw)
+    assert result["in_reply_to"] == "<abc123@example.com>"
+    assert result["references"] == [
+        "<abc123@example.com>", "<def456@example.com>"
+    ]
+
+
+def test_parse_address_without_name():
+    raw = (b"From: alice@example.com\r\n"
+           b"To: bob@example.com\r\n"
+           b"Subject: Test\r\n"
+           b"\r\n"
+           b"Body")
+    result = parse_rfc822(raw)
+    assert result["from"]["email"] == "alice@example.com"
+    assert result["from"]["name"] == ""
+
+
+def test_parse_cc():
+    raw = (b"From: alice@example.com\r\n"
+           b"To: bob@example.com\r\n"
+           b"Cc: Charlie <charlie@example.com>, dave@example.com\r\n"
+           b"Subject: Test\r\n"
+           b"\r\n"
+           b"Body")
+    result = parse_rfc822(raw)
+    assert len(result["cc"]) == 2
+    assert result["cc"][0]["email"] == "charlie@example.com"

python/tests/core/email/test_send.py

@@ -0,0 +1,92 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from mirage.core.email.send import forward_message, reply_message, send_message
+from mirage.resource.email.config import EmailConfig
+
+
+@pytest.fixture
+def config():
+    return EmailConfig(
+        imap_host="imap.test.com",
+        smtp_host="smtp.test.com",
+        smtp_port=587,
+        username="user@test.com",
+        password="pass",
+    )
+
+
+@pytest.mark.asyncio
+async def test_send_message(config):
+    with patch("mirage.core.email.send.aiosmtplib.send",
+               new_callable=AsyncMock) as mock_send:
+        result = await send_message(config, "bob@example.com", "Hello",
+                                    "Hi there")
+        assert result["status"] == "sent"
+        assert result["to"] == "bob@example.com"
+        assert result["subject"] == "Hello"
+        mock_send.assert_called_once()
+
+
+@pytest.mark.asyncio
+async def test_reply_message(config):
+    original = {
+        "from": {
+            "name": "Alice",
+            "email": "alice@example.com"
+        },
+        "to": [{
+            "name": "",
+            "email": "user@test.com"
+        }],
+        "cc": [],
+        "subject": "Hello",
+        "message_id": "<abc123@example.com>",
+        "references": [],
+        "body_text": "Original text",
+        "date": "Mon, 14 Apr 2026 10:30:00 +0000",
+    }
+    with patch("mirage.core.email.send.aiosmtplib.send",
+               new_callable=AsyncMock) as mock_send:
+        result = await reply_message(config, original, "Thanks!")
+        assert result["status"] == "sent"
+        assert result["to"] == "alice@example.com"
+        assert result["subject"] == "Re: Hello"
+        mock_send.assert_called_once()
+        sent_msg = mock_send.call_args[0][0]
+        assert sent_msg["In-Reply-To"] == "<abc123@example.com>"
+
+
+@pytest.mark.asyncio
+async def test_forward_message(config):
+    original = {
+        "from": {
+            "name": "Alice",
+            "email": "alice@example.com"
+        },
+        "subject": "Hello",
+        "body_text": "Original text",
+        "date": "Mon, 14 Apr 2026 10:30:00 +0000",
+    }
+    with patch("mirage.core.email.send.aiosmtplib.send",
+               new_callable=AsyncMock) as mock_send:
+        result = await forward_message(config, original, "charlie@example.com")
+        assert result["status"] == "sent"
+        assert result["to"] == "charlie@example.com"
+        assert result["subject"] == "Fwd: Hello"
+        mock_send.assert_called_once()