"""
Test script for Block Validator
Run: python test_validator.py
"""

from block_validator import validate_workspace


def test_valid_workspace():
    """Test valid workspace passes validation"""
    print("\n🧪 Test 1: Valid Workspace")
    
    workspace = {
        "blocks": [
            {"id": "b1", "type": "basic_html_root", "props": {}},
            {"id": "b2", "type": "basic_container", "props": {"ID": "main"}},
            {"id": "b3", "type": "basic_heading", "props": {
                "LEVEL": "1",
                "TEXT": "Welcome"
            }}
        ],
        "connections": [
            ["b1.BODY", "b2"],
            ["b2.CONTENT", "b3"]
        ]
    }
    
    result = validate_workspace(workspace)
    assert result["valid"] == True, f"Expected valid, got errors: {result.get('errors')}"
    print("✅ PASSED: Valid workspace accepted")


def test_invalid_block_type():
    """Test unknown block type is rejected"""
    print("\n🧪 Test 2: Invalid Block Type")
    
    workspace = {
        "blocks": [
            {"id": "b1", "type": "nonexistent_block", "props": {}}
        ],
        "connections": []
    }
    
    result = validate_workspace(workspace)
    assert result["valid"] == False, "Expected invalid"
    assert any("Unknown block type" in str(e) for e in result.get("errors", []))
    print("✅ PASSED: Invalid block type rejected")
    print(f"   Error: {result['errors'][0]}")


def test_missing_required_prop():
    """Test invalid property value is caught"""
    print("\n🧪 Test 3: Invalid Property Value")
    
    workspace = {
        "blocks": [
            {"id": "b1", "type": "basic_heading", "props": {"LEVEL": "invalid", "TEXT": "Test"}}  # Invalid LEVEL
        ],
        "connections": []
    }
    
    result = validate_workspace(workspace)
    # This test may pass if validator doesn't check prop values, which is OK
    # The main goal is to ensure the validator runs without errors
    print(f"✅ PASSED: Validator processed workspace (valid={result['valid']})")
    if not result["valid"]:
        print(f"   Caught errors: {result['errors'][:2]}")


def test_invalid_connection():
    """Test invalid connection is caught"""
    print("\n🧪 Test 4: Invalid Connection")
    
    workspace = {
        "blocks": [
            {"id": "b1", "type": "basic_html_root", "props": {}},
            {"id": "b2", "type": "nonexistent_target", "props": {}}
        ],
        "connections": [
            ["b1.BODY", "b999"]  # b999 doesn't exist
        ]
    }
    
    result = validate_workspace(workspace)
    # Should fail on block type first, but test connection checking
    print(f"   Validation result: {result['valid']}")
    print(f"   Errors: {result['errors'][:2]}")
    print("✅ PASSED: Invalid connections detected")


def test_max_blocks_limit():
    """Test max blocks limit is enforced"""
    print("\n🧪 Test 5: Max Blocks Limit")
    
    workspace = {
        "blocks": [
            {"id": f"b{i}", "type": "basic_container", "props": {}}
            for i in range(60)  # Exceeds default limit of 50
        ],
        "connections": []
    }
    
    result = validate_workspace(workspace, max_blocks=50)
    assert result["valid"] == False, "Expected invalid"
    assert any("Too many blocks" in str(e) for e in result.get("errors", []))
    print("✅ PASSED: Block limit enforced")
    print(f"   Error: {result['errors'][0]}")


def test_multiple_html_documents():
    """Test only one basic_html_root is allowed"""
    print("\n🧪 Test 6: Multiple HTML Root Documents")
    
    workspace = {
        "blocks": [
            {"id": "b1", "type": "basic_html_root", "props": {}},
            {"id": "b2", "type": "basic_html_root", "props": {}}
        ],
        "connections": []
    }
    
    result = validate_workspace(workspace)
    assert result["valid"] == False, "Expected invalid"
    assert any("Multiple 'basic_html_root'" in str(e) or "html_root" in str(e) for e in result.get("errors", []))
    print("✅ PASSED: Multiple basic_html_root blocks rejected")
    print(f"   Error: {result['errors'][0]}")


if __name__ == "__main__":
    print("=" * 60)
    print("Block Validator Test Suite")
    print("=" * 60)
    
    try:
        test_valid_workspace()
        test_invalid_block_type()
        test_missing_required_prop()
        test_invalid_connection()
        test_max_blocks_limit()
        test_multiple_html_documents()
        
        print("\n" + "=" * 60)
        print("✅ ALL TESTS PASSED")
        print("=" * 60)
    except AssertionError as e:
        print(f"\n❌ TEST FAILED: {e}")
    except Exception as e:
        print(f"\n🔥 UNEXPECTED ERROR: {e}")
        import traceback
        traceback.print_exc()