from database_setup import create_database
from models import db, User, Category

def test_chunk1():
    app = create_database()
    
    with app.app_context():
        print("🧪 Testing CHUNK 1: Foundation Models...")
        
        # Test User creation
        test_user = User(
            email='test@rateright.com.au',
            first_name='Test',
            last_name='User',
            role='worker',
            phone_number='0412345678',
            location='Sydney, NSW',
            abn_number='12345678901'
        )
        test_user.set_password('password123')
        db.session.add(test_user)
        
        # Test Category creation
        test_category = Category(
            name='Formworker',
            description='Formwork construction and setup',
            whs_risk_level='high',
            insurance_requirements='$20M public liability required'
        )
        db.session.add(test_category)
        
        db.session.commit()
        
        # Verify
        user_count = User.query.count()
        category_count = Category.query.count()
        
        print(f"✅ Users created: {user_count}")
        print(f"✅ Categories created: {category_count}")
        print("✅ CHUNK 1 completed successfully!")
        print("📦 Ready for CHUNK 2...")

if __name__ == '__main__':
    test_chunk1()