fix: allow threads and pinterest search sources
This commit is contained in:
committed by
Trevin Chow
parent
6b40d2c46f
commit
7bda02169d
@@ -79,6 +79,8 @@ MOCK_AVAILABLE_SOURCES = [
|
|||||||
"xiaohongshu",
|
"xiaohongshu",
|
||||||
"github",
|
"github",
|
||||||
"perplexity",
|
"perplexity",
|
||||||
|
"threads",
|
||||||
|
"pinterest",
|
||||||
"xquik",
|
"xquik",
|
||||||
"digg",
|
"digg",
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -71,6 +71,12 @@ class CliV3Tests(unittest.TestCase):
|
|||||||
cli.parse_search_flag("web, reddit, hn, web"),
|
cli.parse_search_flag("web, reddit, hn, web"),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def test_parse_search_flag_accepts_optional_social_sources(self):
|
||||||
|
self.assertEqual(
|
||||||
|
["threads", "pinterest"],
|
||||||
|
cli.parse_search_flag("threads, pinterest"),
|
||||||
|
)
|
||||||
|
|
||||||
def test_parse_search_flag_rejects_invalid_or_empty_inputs(self):
|
def test_parse_search_flag_rejects_invalid_or_empty_inputs(self):
|
||||||
with self.assertRaises(SystemExit):
|
with self.assertRaises(SystemExit):
|
||||||
cli.parse_search_flag("unknown")
|
cli.parse_search_flag("unknown")
|
||||||
|
|||||||
Reference in New Issue
Block a user