| import unittest | |
| from app.agents.browser_search import build_search_url, choose_search_url, detect_search_engine | |
| class BrowserSearchTests(unittest.TestCase): | |
| def test_build_search_url_defaults_to_bing(self): | |
| url = build_search_url("qual idade do trump?") | |
| self.assertEqual(url, "https://www.bing.com/search?q=qual+idade+do+trump%3F") | |
| def test_choose_search_url_rotates_when_current_engine_is_blocked(self): | |
| url = choose_search_url( | |
| "qual idade do trump?", | |
| visited_urls=["https://www.bing.com/search?q=qual+idade+do+trump%3F"], | |
| current_url="https://www.bing.com/search?q=qual+idade+do+trump%3F", | |
| blocked=True, | |
| ) | |
| self.assertEqual(url, "https://en.wikipedia.org/w/index.php?search=qual+idade+do+trump%3F") | |
| def test_detect_search_engine_handles_known_urls(self): | |
| self.assertEqual(detect_search_engine("https://www.bing.com/search?q=teste"), "bing") | |
| self.assertEqual( | |
| detect_search_engine("https://en.wikipedia.org/w/index.php?search=donald+trump"), | |
| "wikipedia", | |
| ) | |
| self.assertEqual( | |
| detect_search_engine("https://html.duckduckgo.com/html/?q=donald+trump"), | |
| "duckduckgo", | |
| ) | |
| if __name__ == "__main__": | |
| unittest.main() | |