"""Snapshot-style tests for the daily digest formatter. These hit format_digest with hand-rolled GlobalDigest objects so we exercise every branch (quiet day / bottom trigger / actionable Trump / divergence / empty all-around) without standing up a DB. The actual build_global_digest SQL is exercised by integration tests / manual /digest in the bot. The point of these tests is to lock the user-facing wording — if a future change accidentally turns "BTC bottom triggers: 3/3 firing" into a less clear phrasing, CI catches it. """ from app.services.telegram_digest import ( GlobalDigest, MacroBlock, KolBlock, TrumpBlock, format_digest, ) def _empty_blocks(): return ( MacroBlock(available=False, composite=None, regime=None, ahr999=None, fear_greed=None, bottom_signal_firing=False, bottom_votes=None, funding_signal_firing=False), KolBlock(posts_24h=0, bullish=0, bearish=0, neutral=0, divergences_24h=0, sample_divergence=None), TrumpBlock(posts_24h=0, actionable=[]), ) def test_quiet_day_renders_compact_and_mentions_no_activity(): m, k, t = _empty_blocks() m.available = True m.regime = "NEUTRAL" m.ahr999 = 0.62 m.fear_greed = 38 g = GlobalDigest(date_label="May 26 · Mon", macro=m, kol=k, trump=t, headline_emoji="⚪", headline_text="Quiet day, nothing to act on.") out = format_digest(g) assert "Trump Alpha · Daily Brief" in out assert "⚪" in out assert "Quiet day" in out assert "neutral zone" in out assert "AHR999=0.62" in out assert "No KOL posts" in out assert "No posts in last 24h" in out # No per-user Pro line for free users assert "Your status" not in out # Always ends with the unsubscribe hint assert "/digest off" in out def test_bottom_trigger_day_promotes_macro_section_and_keeps_short(): m, k, t = _empty_blocks() m.available = True m.bottom_signal_firing = True m.bottom_votes = 3 m.ahr999 = 0.41 m.fear_greed = 14 g = GlobalDigest(date_label="May 26 · Mon", macro=m, kol=k, trump=t, headline_emoji="🟢", headline_text="BTC bottom signal firing. Worth a look.") out = format_digest(g) assert "🟢" in out assert "BTC bottom signal firing" in out # Macro section shows the trigger detail assert "3/3 firing" in out assert "AHR999=0.41" in out def test_trump_actionable_list_is_capped_at_three_with_summary_overflow(): m, k, t = _empty_blocks() m.available = True m.regime = "NEUTRAL" t.posts_24h = 6 t.actionable = [ ("BTC", "LONG", 92), ("SOL", "LONG", 81), ("ETH", "SHORT", 76), ("DOGE", "LONG", 71), # 4th — should be summarised ] g = GlobalDigest(date_label="May 26 · Mon", macro=m, kol=k, trump=t, headline_emoji="🟢", headline_text="Trump posted 4 actionable crypto signals today.") out = format_digest(g) assert "BTC · LONG · conf 92" in out assert "SOL · LONG · conf 81" in out assert "ETH · SHORT · conf 76" in out # The fourth must roll into the "and N more" line, not be listed verbatim assert "DOGE" not in out assert "1 more" in out def test_divergence_warning_surfaces_sample_line(): m, k, t = _empty_blocks() m.available = True m.regime = "NEUTRAL" k.posts_24h = 14 k.bullish = 9 k.bearish = 1 k.divergences_24h = 1 k.sample_divergence = "ArthurHayes publicly bullish BTC, on-chain decreased" g = GlobalDigest(date_label="May 26 · Mon", macro=m, kol=k, trump=t, headline_emoji="🟡", headline_text="1 KOL talks-vs-trades divergence today.") out = format_digest(g) assert "⚠️ ArthurHayes" in out assert "9 bullish" in out def test_pro_user_state_appended_for_wallet_bound(): m, k, t = _empty_blocks() m.available = True m.regime = "NEUTRAL" g = GlobalDigest(date_label="May 26 · Mon", macro=m, kol=k, trump=t, headline_emoji="⚪", headline_text="Quiet day, nothing to act on.") out = format_digest(g, user_state={ "wallet": "0xabc", "auto_trade": True, "trades_today": 2, "open_pnl_summary": "1 open (+12.4 USD)", }) assert "Your status" in out assert "auto-trade ON" in out assert "2 trades today" in out assert "1 open (+12.4 USD)" in out def test_pro_state_suppressed_for_free_user(): m, k, t = _empty_blocks() g = GlobalDigest(date_label="May 26 · Mon", macro=m, kol=k, trump=t, headline_emoji="⚪", headline_text="Quiet day, nothing to act on.") out_free = format_digest(g, user_state={}) out_no_wallet = format_digest(g, user_state={"wallet": None}) assert "Your status" not in out_free assert "Your status" not in out_no_wallet def test_html_safe_against_ampersand_in_label(): """The 'F&G' label must always emit as F&G so Telegram's HTML parser doesn't reject the message.""" m, k, t = _empty_blocks() m.available = True m.regime = "NEUTRAL" m.ahr999 = 0.62 m.fear_greed = 38 g = GlobalDigest(date_label="May 26 · Mon", macro=m, kol=k, trump=t, headline_emoji="⚪", headline_text="Quiet day, nothing to act on.") out = format_digest(g) assert "F&G=38" in out assert "F&G=38" not in out def test_total_length_well_under_telegram_cap(): """Even the busiest realistic day must fit in one Telegram message (4096 chars).""" m, k, t = _empty_blocks() m.available = True m.bottom_signal_firing = True m.bottom_votes = 3 m.ahr999 = 0.41 m.fear_greed = 14 k.posts_24h = 30 k.bullish = 20 k.bearish = 8 k.divergences_24h = 4 k.sample_divergence = "Hayes publicly bullish ETH, on-chain decreased" t.posts_24h = 12 t.actionable = [("BTC", "LONG", 95), ("SOL", "LONG", 88), ("ETH", "SHORT", 80)] g = GlobalDigest(date_label="May 26 · Mon", macro=m, kol=k, trump=t, headline_emoji="🟢", headline_text="Lots happening today.") out = format_digest(g, user_state={ "wallet": "0xabc", "auto_trade": True, "trades_today": 4, "open_pnl_summary": "3 open (+125.3 USD)", }) assert len(out) < 4096