From dd92de7787777ad0c4e30d2b7543baf6f67f5b21 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Fri, 5 Jul 2024 17:32:20 -0300
Subject: [PATCH] Added basic tests for the Help class.

---
 aider/tests/test_help.py | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 aider/tests/test_help.py

diff --git a/aider/tests/test_help.py b/aider/tests/test_help.py
new file mode 100644
index 000000000..bc97daee2
--- /dev/null
+++ b/aider/tests/test_help.py
@@ -0,0 +1,38 @@
+import unittest
+from unittest.mock import patch, MagicMock
+from aider.help import Help
+
+class TestHelp(unittest.TestCase):
+    @patch('aider.help.get_index')
+    def setUp(self, mock_get_index):
+        self.mock_index = MagicMock()
+        mock_get_index.return_value = self.mock_index
+        self.help = Help()
+
+    def test_init(self):
+        self.assertIsNotNone(self.help.retriever)
+
+    def test_ask(self):
+        mock_node = MagicMock()
+        mock_node.text = "Test content"
+        mock_node.metadata = {"url": "https://example.com"}
+        self.help.retriever.retrieve.return_value = [mock_node]
+
+        result = self.help.ask("Test question")
+
+        self.assertIn("# Question: Test question", result)
+        self.assertIn('<doc from_url="https://example.com">', result)
+        self.assertIn("Test content", result)
+        self.assertIn("</doc>", result)
+
+    @patch('os.environ')
+    @patch('aider.help.Settings')
+    @patch('aider.help.HuggingFaceEmbedding')
+    def test_environment_and_settings(self, mock_hf, mock_settings, mock_environ):
+        Help()
+        mock_environ.__setitem__.assert_called_with("TOKENIZERS_PARALLELISM", "true")
+        self.assertEqual(mock_settings.embed_model, mock_hf.return_value)
+        mock_hf.assert_called_with(model_name="BAAI/bge-small-en-v1.5")
+
+if __name__ == '__main__':
+    unittest.main()