Add recipients and content assertions to tests

Author: raunakab

backend/tests/daily/connectors/imap/models.py

@@ -1,15 +1,24 @@
 from pydantic import BaseModel
 
 from onyx.connectors.models import Document
+from tests.daily.connectors.utils import to_text_sections
 
 
 class EmailDoc(BaseModel):
     subject: str
+    recipients: set[str]
+    body: str
 
     @classmethod
     def from_doc(cls, document: Document) -> "EmailDoc":
-        assert document.title  # Acceptable since this will only be used in tests.
+        # Acceptable to perform assertions since this class is only used in tests.
+        assert document.title
+        assert document.external_access
+
+        body = " ".join(to_text_sections(iterator=iter(document.sections)))
 
         return cls(
             subject=document.title,
+            recipients=document.external_access.external_user_emails,
+            body=body,
         )

backend/tests/daily/connectors/imap/test_imap_connector.py

@@ -49,9 +49,13 @@ def imap_connector() -> ImapConnector:
         [
             EmailDoc(
                 subject="Testing",
+                recipients=set(["admin@onyx-test.com"]),
+                body="Hello, testing.",
             ),
             EmailDoc(
                 subject="Hello world",
+                recipients=set(["admin@onyx-test.com", "r@rabh.io"]),
+                body='Hello world, this is an email that contains multiple "To" recipients.',
             ),
         ]
     ],