Matt Van Horn
0a9ff16dfc
v3 rewrites the search engine from the ground up: - Intelligent pre-research: resolves X handles, GitHub repos, subreddits, TikTok hashtags, and YouTube channels before searching - GitHub person-mode: PR velocity, top repos by stars, release notes - GitHub project-mode: live star counts, README, releases, top issues - ELI5 mode: plain language synthesis, no jargon - 13+ sources: Reddit, X, YouTube, TikTok, Instagram, HN, Polymarket, GitHub, Threads, Pinterest, Perplexity, Bluesky, Web - Free Reddit comments via public JSON (no API key needed) - Fun judge v2: humor scoring baked into narrative - Cookie consent before browser scanning - 10,000 free ScrapeCreators calls - 1,012 tests Thank you to the community contributors whose issues and PRs shaped v3: @uppinote20 (#143), @zerone0x (#134, #136), @thinkun (#116), @thomasmktong (#124), @fanispoulinakisai-boop (#100), @pejmanjohn (#78), @zl190 (#115), @hnshah (#84, #85, #86) Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
# ruff: noqa: E402
|
|
import io
|
|
import sys
|
|
import unittest
|
|
from contextlib import redirect_stderr
|
|
from pathlib import Path
|
|
from unittest import mock
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "scripts"))
|
|
|
|
from lib import ui
|
|
|
|
|
|
class UiV3Tests(unittest.TestCase):
|
|
def test_show_diagnostic_banner_uses_v3_source_model(self):
|
|
diag = {
|
|
"available_sources": ["grounding", "youtube"],
|
|
"providers": {"google": True, "openai": False, "xai": False},
|
|
"x_backend": None,
|
|
"bird_installed": True,
|
|
"bird_authenticated": False,
|
|
"bird_username": None,
|
|
"native_web_backend": "brave",
|
|
}
|
|
with mock.patch.object(ui, "IS_TTY", False):
|
|
stderr = io.StringIO()
|
|
with redirect_stderr(stderr):
|
|
ui.show_diagnostic_banner(diag)
|
|
output = stderr.getvalue()
|
|
self.assertIn("Reddit", output)
|
|
self.assertIn("unavailable", output)
|
|
self.assertIn("Add AUTH_TOKEN/CT0 or XAI_API_KEY", output)
|
|
self.assertIn("brave API available", output)
|
|
|
|
def test_build_nux_message_mentions_v3_unlock_paths(self):
|
|
text = ui._build_nux_message(
|
|
{"available_sources": ["reddit", "youtube", "grounding"]}
|
|
)
|
|
self.assertIn("Reddit ✓, X ✗, YouTube ✓, Web ✓", text)
|
|
self.assertIn("works fine as-is", text)
|
|
self.assertIn("all free", text)
|
|
|
|
def test_show_complete_uses_actual_sources_for_source_restricted_runs(self):
|
|
with mock.patch.object(ui, "IS_TTY", False):
|
|
stderr = io.StringIO()
|
|
with redirect_stderr(stderr):
|
|
progress = ui.ProgressDisplay("test topic", show_banner=False)
|
|
progress.show_complete(
|
|
source_counts={"grounding": 2},
|
|
display_sources=["grounding"],
|
|
)
|
|
output = stderr.getvalue()
|
|
self.assertIn("Web: 2 results", output)
|
|
self.assertNotIn("Reddit:", output)
|
|
self.assertNotIn("X:", output)
|
|
|
|
def test_show_complete_supports_newer_sources(self):
|
|
with mock.patch.object(ui, "IS_TTY", False):
|
|
stderr = io.StringIO()
|
|
with redirect_stderr(stderr):
|
|
progress = ui.ProgressDisplay("test topic", show_banner=False)
|
|
progress.show_complete(
|
|
source_counts={
|
|
"bluesky": 3,
|
|
"truthsocial": 1,
|
|
"xiaohongshu": 4,
|
|
},
|
|
display_sources=["bluesky", "truthsocial", "xiaohongshu"],
|
|
)
|
|
output = stderr.getvalue()
|
|
self.assertIn("Bluesky: 3 posts", output)
|
|
self.assertIn("Truth Social: 1 post", output)
|
|
self.assertIn("Xiaohongshu: 4 posts", output)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|