Create test_generator.py

Author: BobTheBuidler

tests/test_generator.py

@@ -0,0 +1,75 @@
+"""Tests for the Sphinx documentation generator."""
+
+import os
+import pytest
+from vyper_sphinx_docs.generator import SphinxGenerator
+from vyper_sphinx_docs.parser import VyperParser
+
+
+def test_sphinx_generation(contracts_dir, output_dir):
+    """Test complete Sphinx documentation generation."""
+    # Parse contracts
+    parser = VyperParser(str(contracts_dir))
+    contracts = parser.parse_contracts()
+    
+    # Generate documentation
+    generator = SphinxGenerator(str(output_dir))
+    generator.generate(contracts)
+    
+    # Check generated files
+    docs_dir = output_dir / "docs"
+    assert docs_dir.exists()
+    
+    # Check conf.py
+    conf_py = docs_dir / "conf.py"
+    assert conf_py.exists()
+    conf_content = conf_py.read_text()
+    assert "sphinx.ext.autodoc" in conf_content
+    assert "sphinx_rtd_theme" in conf_content
+    
+    # Check index.rst
+    index_rst = docs_dir / "index.rst"
+    assert index_rst.exists()
+    index_content = index_rst.read_text()
+    assert "Vyper Smart Contracts Documentation" in index_content
+    assert "token" in index_content
+    assert "nested_token" in index_content
+    
+    # Check contract documentation
+    token_rst = docs_dir / "token.rst"
+    assert token_rst.exists()
+    token_content = token_rst.read_text()
+    assert "ERC20 Token Implementation" in token_content
+    assert "transfer" in token_content
+    assert "balance_of" in token_content
+
+
+def test_contract_rst_generation(contracts_dir, output_dir):
+    """Test detailed RST file generation for contracts."""
+    parser = VyperParser(str(contracts_dir))
+    contracts = parser.parse_contracts()
+    
+    generator = SphinxGenerator(str(output_dir))
+    generator.generate(contracts)
+    
+    # Check token.rst content
+    token_rst = output_dir / "docs" / "token.rst"
+    content = token_rst.read_text()
+    
+    # Check sections
+    assert "token" in content
+    assert "Functions" in content
+    
+    # Check function documentation
+    assert ".. py:function:: transfer" in content
+    assert "to: address" in content
+    assert "amount: uint256" in content
+    assert "-> bool" in content
+    
+    assert ".. py:function:: balance_of" in content
+    assert "account: address" in content
+    assert "-> uint256" in content
+    
+    # Check docstrings
+    assert "Transfer tokens to a specified address" in content
+    assert "Get the token balance of an account" in content