test_schema_generation.py

#!/usr/bin/env python3
"""
Test script to validate schema generation before full migration
"""

from couchdb_converter.couchdb_schema_analyzer import CouchDBSchemaAnalyzer
from couchdb_converter.couchdb2mysql import CouchDBSQLBuilder
import json

def test_schema_generation():
    print("Testing CouchDB Schema Generation")
    print("=" * 50)
    
    # Initialize analyzer
    analyzer = CouchDBSchemaAnalyzer('http://admin:root@localhost:5984', 'nosql2sql_test')
    
    # Analyze documents
    print("Step 1: Analyzing documents...")
    if not analyzer.analyze_documents():
        print("❌ Document analysis failed")
        return
    
    print("✅ Document analysis successful")
    
    # Generate schema recommendations
    print("\nStep 2: Generating schema recommendations...")
    schema = analyzer.generate_schema_recommendations()
    
    print(f"✅ Generated schema with {len(schema['tables'])} tables:")
    for table_name in schema['tables'].keys():
        print(f"  - {table_name}")
    
    # Test SQL generation
    print("\nStep 3: Testing SQL generation...")
    try:
        sql_builder = CouchDBSQLBuilder(schema)
        create_statements = sql_builder.generate_all_create_statements()
        
        print(f"✅ Generated {len(create_statements)} CREATE statements")
        
        print("\nStep 4: Showing generated SQL:")
        for i, sql in enumerate(create_statements):
            table_name = sql.split()[5].strip('`')
            print(f"\n--- Table {i+1}: {table_name} ---")
            print(sql)
            
    except Exception as e:
        print(f"❌ SQL generation failed: {e}")
        import traceback
        traceback.print_exc()
        return
    
    print("\n✅ All tests passed! Schema generation is working correctly.")

if __name__ == "__main__":
    test_schema_generation()