Class TestSearchEngine

tests/test_search_engine.py:7–115

from tools.search_engine import search

class TestSearchEngine(unittest.TestCase):
    def setUp(self):
        # Capture stdout and stderr for testing
        self.stdout = StringIO()
        self.stderr = StringIO()
        self.old_stdout = sys.stdout
        self.old_stderr = sys.stderr
        sys.stdout = self.stdout
        sys.stderr = self.stderr

    def tearDown(self):
        # Restore stdout and stderr
        sys.stdout = self.old_stdout
        sys.stderr = self.old_stderr

    @patch('tools.search_engine.DDGS')
    def test_successful_search(self, mock_ddgs):
        # Mock search results
        mock_results = [
            {
                'href': 'http://example.com',
                'title': 'Example Title',
                'body': 'Example Body'
            },
            {
                'href': 'http://example2.com',
                'title': 'Example Title 2',
                'body': 'Example Body 2'
            }
        ]
        
        # Setup mock
        mock_ddgs_instance = MagicMock()
        mock_ddgs_instance.__enter__.return_value.text.return_value = mock_results
        mock_ddgs.return_value = mock_ddgs_instance

        # Run search
        search("test query", max_results=2)

        # Check debug output
        expected_debug = "DEBUG: Searching for query: test query (attempt 1/3)"
        self.assertIn(expected_debug, self.stderr.getvalue())
        self.assertIn("DEBUG: Found 2 results", self.stderr.getvalue())

        # Check search results output
        output = self.stdout.getvalue()
        self.assertIn("=== Result 1 ===", output)
        self.assertIn("URL: http://example.com", output)
        self.assertIn("Title: Example Title", output)
        self.assertIn("Snippet: Example Body", output)
        self.assertIn("=== Result 2 ===", output)
        self.assertIn("URL: http://example2.com", output)
        self.assertIn("Title: Example Title 2", output)
        self.assertIn("Snippet: Example Body 2", output)

        # Verify mock was called correctly
        mock_ddgs_instance.__enter__.return_value.text.assert_called_once_with(
            "test query",