Fix a situation when there are no attachments

Author: lcd1232

tempmail/models.py

@@ -1,6 +1,7 @@
 """Data models for the Temp Mail API."""
 
 import enum
+import typing
 from dataclasses import dataclass
 from datetime import datetime
 from typing import List, Dict, Any
@@ -88,6 +89,7 @@ class EmailMessage:
 
     @classmethod
     def from_json(cls, data: Dict[str, Any]) -> "EmailMessage":
+        attachments: typing.List[Dict[str, Any]] = data["attachments"] or []
         return cls(
             id=data["id"],
             from_addr=data["from"],
@@ -99,7 +101,7 @@ def from_json(cls, data: Dict[str, Any]) -> "EmailMessage":
             ),
             cc=data["cc"],
             body_html=data["body_html"],
-            attachments=[Attachment.from_json(v) for v in data["attachments"]],
+            attachments=[Attachment.from_json(v) for v in attachments],
         )
 
 

tests/test_client.py

@@ -155,6 +155,46 @@ def test_list_email_messages_success(self, mock_request):
             ],
         )
 
+    @patch("tempmail.client.requests.Session.request")
+    def test_list_email_messages_no_attachments(self, mock_request):
+        mock_response = Mock()
+        mock_response.status_code = 200
+        mock_response.json.return_value = {
+            "messages": [
+                {
+                    "id": "msg1",
+                    "from": "[email protected]",
+                    "to": "[email protected]",
+                    "cc": ["[email protected]"],
+                    "subject": "Test Subject",
+                    "body_text": "Test body",
+                    "body_html": "<p>Test body</p>",
+                    "created_at": "2023-01-01T00:00:00Z",
+                    "attachments": None,
+                }
+            ]
+        }
+        mock_response.headers = self._rate_limit_headers
+        mock_request.return_value = mock_response
+
+        client = TempMailClient("test-api-key")
+        messages: typing.List[EmailMessage] = client.list_email_messages("[email protected]")
+
+        assert len(messages) == 1
+        assert messages[0] == EmailMessage(
+            id="msg1",
+            from_addr="[email protected]",
+            to_addr="[email protected]",
+            cc=["[email protected]"],
+            subject="Test Subject",
+            body_text="Test body",
+            body_html="<p>Test body</p>",
+            created_at=datetime.datetime(
+                2023, 1, 1, 0, 0, tzinfo=datetime.timezone.utc
+            ),
+            attachments=[],
+        )
+
     @patch("tempmail.client.requests.Session.request")
     def test_list_email_messages_empty(self, mock_request):
         mock_response = Mock()