Merge pull request #420 from adamtheturtle/debug

Add annotations to mention

Author: adamtheturtle

src/sphinx_notion/__init__.py

@@ -81,6 +81,7 @@
 from ultimate_notion.obj_api.core import ObjectRef, UserRef
 from ultimate_notion.obj_api.enums import BGColor, CodeLang, Color
 from ultimate_notion.obj_api.objects import (
+    Annotations,
     DateRange,
     MentionDatabase,
     MentionDate,
@@ -518,7 +519,10 @@ def _(node: _MentionUserNode) -> Text:
     """
     user_id = node.attributes["user_id"]
     user_ref = UserRef(id=user_id)
-    mention_obj = MentionUser.build_mention_from(user=user_ref)
+    mention_obj = MentionUser.build_mention_from(
+        user=user_ref,
+        style=Annotations(),
+    )
     return Text.wrap_obj_ref(obj_refs=[mention_obj])
 
 
@@ -530,7 +534,10 @@ def _(node: _MentionPageNode) -> Text:
     """
     page_id = node.attributes["page_id"]
     page_obj_ref = ObjectRef(id=page_id)
-    mention_obj = MentionPage.build_mention_from(page=page_obj_ref)
+    mention_obj = MentionPage.build_mention_from(
+        page=page_obj_ref,
+        style=Annotations(),
+    )
     return Text.wrap_obj_ref(obj_refs=[mention_obj])
 
 
@@ -542,7 +549,10 @@ def _(node: _MentionDatabaseNode) -> Text:
     """
     database_id = node.attributes["database_id"]
     database_obj_ref = ObjectRef(id=database_id)
-    mention_obj = MentionDatabase.build_mention_from(db=database_obj_ref)
+    mention_obj = MentionDatabase.build_mention_from(
+        db=database_obj_ref,
+        style=Annotations(),
+    )
     return Text.wrap_obj_ref(obj_refs=[mention_obj])
 
 
@@ -555,7 +565,10 @@ def _(node: _MentionDateNode) -> Text:
     parsed_date = node.attributes["date"]
     date_range = DateRange.build(dt_spec=parsed_date)
 
-    mention_obj = MentionDate.build_mention_from(date_range=date_range)
+    mention_obj = MentionDate.build_mention_from(
+        date_range=date_range,
+        style=Annotations(),
+    )
     return Text.wrap_obj_ref(obj_refs=[mention_obj])
 
 

tests/test_integration.py

@@ -58,6 +58,7 @@
 from ultimate_notion.obj_api.core import ObjectRef, UserRef
 from ultimate_notion.obj_api.enums import BGColor, CodeLang, Color
 from ultimate_notion.obj_api.objects import (
+    Annotations,
     DateRange,
     MentionDatabase,
     MentionDate,
@@ -3336,7 +3337,11 @@ def test_notion_mention_user(
     """
 
     user_ref = UserRef(id=UUID(hex=test_user_id))
-    mention_obj = MentionUser.build_mention_from(user=user_ref)
+    mention_obj = MentionUser.build_mention_from(
+        user=user_ref,
+        # We require annotations else the equivalence check later will fail.
+        style=Annotations(),
+    )
     expected_blocks = [
         UnoParagraph(
             text=text(text="Hello ")
@@ -3368,7 +3373,10 @@ def test_notion_mention_page(
     """
 
     page_obj_ref = ObjectRef(id=UUID(hex=test_page_id))
-    mention_obj = MentionPage.build_mention_from(page=page_obj_ref)
+    mention_obj = MentionPage.build_mention_from(
+        page=page_obj_ref,
+        style=Annotations(),
+    )
     expected_blocks = [
         UnoParagraph(
             text=text(text="See ")
@@ -3400,7 +3408,10 @@ def test_notion_mention_database(
     """
 
     database_obj_ref = ObjectRef(id=UUID(hex=test_database_id))
-    mention_obj = MentionDatabase.build_mention_from(db=database_obj_ref)
+    mention_obj = MentionDatabase.build_mention_from(
+        db=database_obj_ref,
+        style=Annotations(),
+    )
     expected_blocks = [
         UnoParagraph(
             text=text(text="Check the ")
@@ -3433,7 +3444,10 @@ def test_notion_mention_date(
 
     parsed_date = dt.date.fromisoformat(test_date)
     date_range = DateRange.build(dt_spec=parsed_date)
-    mention_obj = MentionDate.build_mention_from(date_range=date_range)
+    mention_obj = MentionDate.build_mention_from(
+        date_range=date_range,
+        style=Annotations(),
+    )
     expected_blocks = [
         UnoParagraph(
             text=text(text="The meeting is on ")