test_builder.py

from builders import docs, hiring, outreach, foundational, persistence
import os

def test_docs_builder():
    print("Testing Docs Builder...")
    plan = docs.generate_business_plan("BuilderCorp", "SaaS", "Automation", "SMEs")
    assert "BuilderCorp" in plan
    assert "Market Analysis" in plan
    sop = docs.generate_sop("Hiring", ["Post Job", "Screen", "Interview"])
    assert "SOP: Hiring" in sop
    assert "1. Post Job" in sop
    print("PASSED")

def test_growth_builder():
    print("Testing Growth Builder...")
    intro = outreach.draft_investor_intro("Alice", "BuilderCorp", "10k MRR", "$1M")
    assert "Alice" in intro
    assert "10k MRR" in intro
    calendar = outreach.generate_content_calendar("AI", "LinkedIn")
    assert "Content Calendar" in calendar
    print("PASSED")

def test_persistence():
    print("Testing Persistence...")
    if os.path.exists("data/ventures.json"):
        os.remove("data/ventures.json")
    
    res = persistence.save_venture_profile("V1", "Tech", "VisionX", "MissionY", ["V1", "V2"])
    assert "saved" in res
    profile = persistence.get_venture_profile("V1")
    assert profile["industry"] == "Tech"
    print("PASSED")

if __name__ == "__main__":
    test_docs_builder()
    test_growth_builder()
    test_persistence()
    print("\nAll Business Builder tests passed!")