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()