add pytests

tests/test_llms_txt.py

@@ -0,0 +1,82 @@
+"""Test the sphinx_llms_txt extension."""
+
+from sphinx_llms_txt import LLMSFullManager, setup
+
+
+def test_version():
+    """Test that the version is defined."""
+    from sphinx_llms_txt import __version__
+
+    assert __version__
+
+
+def test_setup_returns_valid_dict():
+    """Test that the setup function returns a valid dict."""
+
+    # Mock a Sphinx app
+    class MockApp:
+        def __init__(self):
+            self.config_values = {}
+            self.connections = {}
+
+        def add_config_value(self, name, default, rebuild):
+            self.config_values[name] = (default, rebuild)
+
+        def connect(self, event, handler):
+            self.connections[event] = handler
+
+    app = MockApp()
+    result = setup(app)
+
+    # Check that result is a dict
+    assert isinstance(result, dict)
+    assert "version" in result
+    assert "parallel_read_safe" in result
+    assert "parallel_write_safe" in result
+
+
+def test_llms_full_manager_initialization():
+    """Test initialization of LLMSFullManager."""
+    manager = LLMSFullManager()
+    assert manager.page_titles == {}
+    assert manager.config == {}
+    assert manager.master_doc is None
+    assert manager.env is None
+
+
+def test_manager_page_title_update():
+    """Test updating page titles."""
+    manager = LLMSFullManager()
+    manager.update_page_title("doc1", "Title 1")
+    manager.update_page_title("doc2", "Title 2")
+
+    assert manager.page_titles["doc1"] == "Title 1"
+    assert manager.page_titles["doc2"] == "Title 2"
+
+
+def test_set_config():
+    """Test setting configuration."""
+    manager = LLMSFullManager()
+    config = {
+        "llms_txt_filename": "custom.txt",
+        "llms_txt_verbose": True,
+    }
+    manager.set_config(config)
+    assert manager.config == config
+
+
+def test_set_master_doc():
+    """Test setting master doc."""
+    manager = LLMSFullManager()
+    manager.set_master_doc("index")
+    assert manager.master_doc == "index"
+
+
+def test_empty_page_order():
+    """Test get_page_order returns empty list when env or master_doc not set."""
+    manager = LLMSFullManager()
+    assert manager.get_page_order() == []
+
+    # Set only master_doc, but not env
+    manager.set_master_doc("index")
+    assert manager.get_page_order() == []