test_website.py

import os
from modules import scaffold, components, styling

def test_scaffold():
    print("Testing Scaffolding...")
    res = scaffold.init_project("TestSite", "Tailwind CDN")
    assert "initialized" in res
    assert os.path.exists("web_projects/TestSite/index.html")
    print("PASSED")

def test_components():
    print("Testing Component Gen...")
    hero = components.generate_component("hero", details={"title": "Hello World"})
    assert "Hello World" in hero
    assert "bg-white" in hero
    print("PASSED")

def test_styling():
    print("Testing Styling Gen...")
    script = styling.generate_theme_script("#ff0000", "#00ff00")
    assert "#ff0000" in script
    assert "tailwind.config" in script
    print("PASSED")

if __name__ == "__main__":
    test_scaffold()
    test_components()
    test_styling()
    print("\nAll Website Builder MCP tests passed!")