max replies allowed is 5

Author: patrickZWY

manual_testing/mastodon_preview_test.py

@@ -0,0 +1,25 @@
+import os, sys 
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
+
+from social_posters.mastodon import PosterMastodon
+from mastodon_manual_test import post_exceed_500_chars_limit_with_adoption_link
+
+poster = PosterMastodon.__new__(PosterMastodon)
+
+pet = post_exceed_500_chars_limit_with_adoption_link()
+post = poster.format_post(pet)
+
+main_caption, replies = poster._format_caption_thread(post)
+
+print("\n" + "=" * 60)
+print("MAIN POST")
+print("=" * 60)
+print(main_caption)
+print(f"\nLength: {len(main_caption)}")
+
+for i, reply in enumerate(replies, start=1):
+    print("\n" + "=" * 60)
+    print(f"REPLY {i}")
+    print("=" * 60)
+    print(reply)
+    print(f"\nLength: {len(reply)}")

social_posters/mastodon.py

@@ -11,7 +11,7 @@
 THREAD_SUFFIX = "\n\nMore details below ⬇️"
 MASTODON_CHARACTER_LIMIT = 500
 TRUNCATION_SUFFIX = "..."
-
+MAX_REPLIES = 5
 
 class PosterMastodon(SocialPoster):
     def __init__(self):
@@ -126,12 +126,21 @@ def _split_reply_chunks(self, text: str) -> list[str]:
         chunks = []
         remaining = text.strip()
 
-        while remaining:
+        while remaining and len(chunks) < MAX_REPLIES:
             chunk, remaining = self._safe_truncate(
                 remaining,
                 MASTODON_CHARACTER_LIMIT
             )
             chunks.append(chunk)
+        
+        if remaining and chunks:
+            last_chunk = chunks[-1]
+
+            cutoff = MASTODON_CHARACTER_LIMIT - len(TRUNCATION_SUFFIX)
+
+            last_chunk, _ = self._safe_truncate(last_chunk, cutoff)
+
+            chunks[-1] = f"{last_chunk}{TRUNCATION_SUFFIX}"
 
         return chunks
 

tests/test_mastodon.py

@@ -1,5 +1,5 @@
 from abstractions import Post
-from social_posters.mastodon import PosterMastodon, MASTODON_CHARACTER_LIMIT
+from social_posters.mastodon import PosterMastodon, MASTODON_CHARACTER_LIMIT, MAX_REPLIES
 
 
 class TestMastodonCaption:
@@ -16,6 +16,43 @@ def reconstruct_text(self, main_caption: str, replies: list[str]) -> str:
         )
 
         return " ".join([main_without_suffix] + replies).strip()
+
+    def test_reply_count_is_capped(self):
+        post = Post(text="hello " * 5000)
+
+        _, replies = self.poster._format_caption_thread(post)
+
+        assert len(replies) <= MAX_REPLIES
+
+    def test_last_reply_has_truncation_suffix_when_capped(self):
+        post = Post(text="hello " * 5000)
+
+        _, replies = self.poster._format_caption_thread(post)
+
+        assert len(replies) == MAX_REPLIES
+        assert replies[-1].endswith("...")
+        assert len(replies[-1]) <= MASTODON_CHARACTER_LIMIT
+    
+    def test_last_reply_has_no_truncation_suffix_when_not_capped(self):
+        post = Post(text="hello " * 300)
+
+        _, replies = self.poster._format_caption_thread(post)
+
+        assert replies
+        assert len(replies) < MAX_REPLIES
+        assert not replies[-1].endswith("...")
+
+    def test_capped_thread_does_not_preserve_all_original_text(self):
+        original_text = "hello " * 5000
+        post = Post(text=original_text)
+
+        main_caption, replies = self.poster._format_caption_thread(post)
+
+        reconstructed = self.reconstruct_text(main_caption, replies)
+
+        assert len(replies) == MAX_REPLIES
+        assert reconstructed != original_text
+        assert replies[-1].endswith("...")
 
     def test_thread_preserves_original_text_content(self):
         original_text = " ".join(f"word{i}" for i in range(300))