mockforge-vbr 0.3.100

//! Database migration system
//!
//! This module handles generating SQLite schema from entity definitions,
//! creating tables, indexes, foreign keys, and managing schema migrations.

use crate::entities::{Entity, EntityRegistry};
use crate::schema::{CascadeAction, ManyToManyDefinition, VbrSchemaDefinition};
use crate::{Error, Result};

/// Migration manager for database schema evolution
pub struct MigrationManager {
    /// Current migration version
    #[allow(dead_code)]
    version: u64,
}

impl MigrationManager {
    /// Create a new migration manager
    pub fn new() -> Self {
        Self { version: 0 }
    }

    /// Generate CREATE TABLE statement from an entity
    pub fn generate_create_table(&self, entity: &Entity) -> Result<String> {
        let schema = &entity.schema;
        let table_name = entity.table_name();

        let mut sql = format!("CREATE TABLE IF NOT EXISTS {} (\n", table_name);
        let mut columns = Vec::new();

        // Add columns from schema fields
        for field in &schema.base.fields {
            let column_def = self.field_to_column_definition(field, schema)?;
            columns.push(column_def);
        }

        // Add primary key constraint
        if !schema.primary_key.is_empty() {
            let pk_fields = schema.primary_key.join(", ");
            columns.push(format!("PRIMARY KEY ({})", pk_fields));
        }

        sql.push_str(&columns.join(",\n    "));
        sql.push_str("\n)");

        Ok(sql)
    }

    /// Generate foreign key constraints
    pub fn generate_foreign_keys(&self, entity: &Entity) -> Vec<String> {
        let mut fk_statements = Vec::new();
        let table_name = entity.table_name();

        for fk in &entity.schema.foreign_keys {
            let on_delete = cascade_action_to_sql(fk.on_delete);
            let on_update = cascade_action_to_sql(fk.on_update);

            let fk_name = format!("fk_{}_{}", table_name, fk.field);
            let statement = format!(
                "ALTER TABLE {} ADD CONSTRAINT {} FOREIGN KEY ({}) REFERENCES {}({}) ON DELETE {} ON UPDATE {}",
                table_name, fk_name, fk.field, fk.target_entity.to_lowercase() + "s", fk.target_field, on_delete, on_update
            );
            fk_statements.push(statement);
        }

        fk_statements
    }

    /// Generate index creation statements
    pub fn generate_indexes(&self, entity: &Entity) -> Vec<String> {
        let mut index_statements = Vec::new();
        let table_name = entity.table_name();

        for index in &entity.schema.indexes {
            let unique = if index.unique { "UNIQUE " } else { "" };
            let fields = index.fields.join(", ");
            let statement = format!(
                "CREATE {}INDEX IF NOT EXISTS {} ON {} ({})",
                unique, index.name, table_name, fields
            );
            index_statements.push(statement);
        }

        index_statements
    }

    /// Generate junction table creation statement for a many-to-many relationship
    pub fn generate_junction_table(&self, m2m: &ManyToManyDefinition) -> Result<String> {
        let junction_table = m2m
            .junction_table
            .as_ref()
            .ok_or_else(|| Error::generic("Junction table name is required".to_string()))?;

        // Get table names for entities (assuming pluralization)
        let table_a = m2m.entity_a.to_lowercase() + "s";
        let table_b = m2m.entity_b.to_lowercase() + "s";

        let on_delete_a = cascade_action_to_sql(m2m.on_delete_a);
        let on_delete_b = cascade_action_to_sql(m2m.on_delete_b);

        // Create junction table with composite primary key
        let sql = format!(
            "CREATE TABLE IF NOT EXISTS {} (
    {} TEXT NOT NULL,
    {} TEXT NOT NULL,
    PRIMARY KEY ({}, {}),
    FOREIGN KEY ({}) REFERENCES {}(id) ON DELETE {},
    FOREIGN KEY ({}) REFERENCES {}(id) ON DELETE {}
)",
            junction_table,
            m2m.entity_a_field,
            m2m.entity_b_field,
            m2m.entity_a_field,
            m2m.entity_b_field,
            m2m.entity_a_field,
            table_a,
            on_delete_a,
            m2m.entity_b_field,
            table_b,
            on_delete_b
        );

        Ok(sql)
    }

    /// Generate all junction tables for many-to-many relationships in the registry
    pub fn generate_all_junction_tables(
        &self,
        registry: &EntityRegistry,
    ) -> Result<Vec<(String, String)>> {
        let mut junction_tables = Vec::new();
        let mut processed = std::collections::HashSet::new();

        // Collect all many-to-many relationships
        for entity in registry.list() {
            if let Some(entity_def) = registry.get(&entity) {
                for m2m in &entity_def.schema.many_to_many {
                    // Create a canonical key to avoid duplicates
                    let (a, b) = if m2m.entity_a < m2m.entity_b {
                        (m2m.entity_a.clone(), m2m.entity_b.clone())
                    } else {
                        (m2m.entity_b.clone(), m2m.entity_a.clone())
                    };
                    let key = format!("{}:{}", a, b);

                    if !processed.contains(&key) {
                        processed.insert(key);
                        let table_name = m2m
                            .junction_table
                            .as_ref()
                            .ok_or_else(|| {
                                Error::generic("Junction table name is required".to_string())
                            })?
                            .clone();
                        let create_sql = self.generate_junction_table(m2m)?;
                        junction_tables.push((table_name, create_sql));
                    }
                }
            }
        }

        Ok(junction_tables)
    }

    /// Convert a field definition to SQL column definition
    fn field_to_column_definition(
        &self,
        field: &mockforge_data::FieldDefinition,
        schema: &VbrSchemaDefinition,
    ) -> Result<String> {
        let name = &field.name;
        let sql_type = rust_type_to_sql_type(&field.field_type)?;
        let mut parts = vec![format!("{} {}", name, sql_type)];

        // Add NOT NULL if required
        if field.required {
            parts.push("NOT NULL".to_string());
        }

        // Add default value if specified
        if let Some(default) = &field.default {
            let default_value = value_to_sql_default(default)?;
            parts.push(format!("DEFAULT {}", default_value));
        }

        // Check for auto-generation rules
        if let Some(rule) = schema.auto_generation.get(name) {
            match rule {
                crate::schema::AutoGenerationRule::AutoIncrement => {
                    parts.push("AUTOINCREMENT".to_string());
                }
                crate::schema::AutoGenerationRule::Uuid => {
                    // UUID will be generated in application code
                }
                crate::schema::AutoGenerationRule::Timestamp => {
                    parts.push("DEFAULT CURRENT_TIMESTAMP".to_string());
                }
                crate::schema::AutoGenerationRule::Date => {
                    parts.push("DEFAULT (date('now'))".to_string());
                }
                crate::schema::AutoGenerationRule::Custom(expr) => {
                    parts.push(format!("DEFAULT ({})", expr));
                }
                crate::schema::AutoGenerationRule::Pattern(_) => {
                    // Pattern-based IDs are generated in application code
                }
                crate::schema::AutoGenerationRule::Realistic { .. } => {
                    // Realistic IDs are generated in application code
                }
            }
        }

        Ok(parts.join(" "))
    }

    /// Migrate database schema for all entities
    pub async fn migrate(
        &self,
        entities: &[Entity],
        database: &mut dyn crate::database::VirtualDatabase,
    ) -> Result<()> {
        // Generate and execute CREATE TABLE statements
        for entity in entities {
            let create_table = self.generate_create_table(entity)?;
            database.create_table(&create_table).await?;

            // Create indexes
            for index_sql in self.generate_indexes(entity) {
                database.execute(&index_sql, &[]).await?;
            }

            // Note: Foreign keys are added after all tables are created
            // This will be handled in a separate pass
        }

        // Second pass: add foreign key constraints
        for entity in entities {
            for fk_sql in self.generate_foreign_keys(entity) {
                // Foreign keys might fail if tables don't exist yet, so we continue on error
                let _ = database.execute(&fk_sql, &[]).await;
            }
        }

        Ok(())
    }
}

impl Default for MigrationManager {
    fn default() -> Self {
        Self::new()
    }
}

/// Create a database table for an entity
///
/// This is a convenience function that creates a table, indexes, and foreign keys
/// for a single entity.
///
/// # Arguments
/// * `database` - The virtual database instance
/// * `entity` - The entity to create a table for
pub async fn create_table_for_entity(
    database: &dyn crate::database::VirtualDatabase,
    entity: &Entity,
) -> Result<()> {
    let manager = MigrationManager::new();

    // Create the table
    let create_table = manager.generate_create_table(entity)?;
    database.create_table(&create_table).await?;

    // Create indexes
    for index_sql in manager.generate_indexes(entity) {
        database.execute(&index_sql, &[]).await?;
    }

    // Create foreign keys (if target tables exist)
    for fk_sql in manager.generate_foreign_keys(entity) {
        // Foreign keys might fail if target tables don't exist yet
        // This is okay - they'll be created when the target entity is processed
        let _ = database.execute(&fk_sql, &[]).await;
    }

    Ok(())
}

/// Create all junction tables for many-to-many relationships
///
/// # Arguments
/// * `database` - The virtual database instance
/// * `registry` - The entity registry containing all entities
pub async fn create_junction_tables(
    database: &dyn crate::database::VirtualDatabase,
    registry: &EntityRegistry,
) -> Result<()> {
    let manager = MigrationManager::new();
    let junction_tables = manager.generate_all_junction_tables(registry)?;

    for (_table_name, create_sql) in junction_tables {
        database.create_table(&create_sql).await?;
    }

    Ok(())
}

/// Convert Rust type to SQL type
fn rust_type_to_sql_type(rust_type: &str) -> Result<&str> {
    match rust_type.to_lowercase().as_str() {
        "string" | "str" | "text" | "uuid" | "email" | "url" => Ok("TEXT"),
        "integer" | "int" | "i32" | "i64" => Ok("INTEGER"),
        "float" | "double" | "f32" | "f64" | "number" => Ok("REAL"),
        "boolean" | "bool" => Ok("INTEGER"), // SQLite uses INTEGER for booleans
        "date" | "datetime" | "timestamp" => Ok("TEXT"), // SQLite stores dates as TEXT
        _ => Ok("TEXT"),                     // Default to TEXT for unknown types
    }
}

/// Convert cascade action to SQL
fn cascade_action_to_sql(action: CascadeAction) -> &'static str {
    match action {
        CascadeAction::NoAction => "NO ACTION",
        CascadeAction::Cascade => "CASCADE",
        CascadeAction::SetNull => "SET NULL",
        CascadeAction::SetDefault => "SET DEFAULT",
        CascadeAction::Restrict => "RESTRICT",
    }
}

/// Convert serde_json::Value to SQL default value string
fn value_to_sql_default(value: &serde_json::Value) -> Result<String> {
    match value {
        serde_json::Value::String(s) => Ok(format!("'{}'", s.replace("'", "''"))),
        serde_json::Value::Number(n) => Ok(n.to_string()),
        serde_json::Value::Bool(b) => Ok(if *b { "1" } else { "0" }.to_string()),
        serde_json::Value::Null => Ok("NULL".to_string()),
        _ => Err(Error::generic("Unsupported default value type".to_string())),
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::database::{InMemoryDatabase, VirtualDatabase};
    use crate::schema::{
        AutoGenerationRule, CascadeAction, ForeignKeyDefinition, IndexDefinition,
        ManyToManyDefinition,
    };
    use mockforge_data::{FieldDefinition, SchemaDefinition};

    fn create_test_entity(name: &str) -> Entity {
        let base_schema = SchemaDefinition::new(name.to_string())
            .with_field(FieldDefinition::new("id".to_string(), "string".to_string()))
            .with_field(FieldDefinition::new("name".to_string(), "string".to_string()));

        let vbr_schema = VbrSchemaDefinition::new(base_schema);
        Entity::new(name.to_string(), vbr_schema)
    }

    // MigrationManager tests
    #[test]
    fn test_migration_manager_new() {
        let manager = MigrationManager::new();
        assert_eq!(manager.version, 0);
    }

    #[test]
    fn test_migration_manager_default() {
        let manager = MigrationManager::default();
        assert_eq!(manager.version, 0);
    }

    #[test]
    fn test_generate_create_table() {
        let manager = MigrationManager::new();
        let entity = create_test_entity("User");

        let result = manager.generate_create_table(&entity);
        assert!(result.is_ok());

        let sql = result.unwrap();
        assert!(sql.contains("CREATE TABLE IF NOT EXISTS users"));
        assert!(sql.contains("id TEXT"));
        assert!(sql.contains("name TEXT"));
        assert!(sql.contains("PRIMARY KEY (id)"));
    }

    #[test]
    fn test_generate_create_table_with_multiple_fields() {
        let manager = MigrationManager::new();

        let base_schema = SchemaDefinition::new("Product".to_string())
            .with_field(FieldDefinition::new("id".to_string(), "string".to_string()))
            .with_field(FieldDefinition::new("name".to_string(), "string".to_string()))
            .with_field(FieldDefinition::new("price".to_string(), "number".to_string()))
            .with_field(FieldDefinition::new("in_stock".to_string(), "boolean".to_string()));

        let vbr_schema = VbrSchemaDefinition::new(base_schema);
        let entity = Entity::new("Product".to_string(), vbr_schema);

        let sql = manager.generate_create_table(&entity).unwrap();
        assert!(sql.contains("name TEXT"));
        assert!(sql.contains("price REAL"));
        assert!(sql.contains("in_stock INTEGER"));
    }

    #[test]
    fn test_generate_foreign_keys() {
        let manager = MigrationManager::new();

        let base_schema = SchemaDefinition::new("Order".to_string())
            .with_field(FieldDefinition::new("id".to_string(), "string".to_string()));

        let mut vbr_schema = VbrSchemaDefinition::new(base_schema);
        vbr_schema.foreign_keys.push(ForeignKeyDefinition {
            field: "user_id".to_string(),
            target_entity: "User".to_string(),
            target_field: "id".to_string(),
            on_delete: CascadeAction::Cascade,
            on_update: CascadeAction::NoAction,
        });

        let entity = Entity::new("Order".to_string(), vbr_schema);

        let fk_statements = manager.generate_foreign_keys(&entity);
        assert_eq!(fk_statements.len(), 1);
        assert!(fk_statements[0].contains("FOREIGN KEY"));
        assert!(fk_statements[0].contains("user_id"));
        assert!(fk_statements[0].contains("CASCADE"));
    }

    #[test]
    fn test_generate_indexes() {
        let manager = MigrationManager::new();

        let base_schema = SchemaDefinition::new("User".to_string());
        let mut vbr_schema = VbrSchemaDefinition::new(base_schema);

        vbr_schema.indexes.push(IndexDefinition {
            name: "idx_email".to_string(),
            fields: vec!["email".to_string()],
            unique: true,
        });

        let entity = Entity::new("User".to_string(), vbr_schema);

        let index_statements = manager.generate_indexes(&entity);
        assert_eq!(index_statements.len(), 1);
        assert!(index_statements[0].contains("CREATE UNIQUE INDEX"));
        assert!(index_statements[0].contains("idx_email"));
        assert!(index_statements[0].contains("email"));
    }

    #[test]
    fn test_generate_indexes_non_unique() {
        let manager = MigrationManager::new();

        let base_schema = SchemaDefinition::new("Product".to_string());
        let mut vbr_schema = VbrSchemaDefinition::new(base_schema);

        vbr_schema.indexes.push(IndexDefinition {
            name: "idx_category".to_string(),
            fields: vec!["category".to_string()],
            unique: false,
        });

        let entity = Entity::new("Product".to_string(), vbr_schema);

        let index_statements = manager.generate_indexes(&entity);
        assert_eq!(index_statements.len(), 1);
        assert!(index_statements[0].contains("CREATE INDEX"));
        assert!(!index_statements[0].contains("UNIQUE"));
    }

    #[test]
    fn test_generate_junction_table() {
        let manager = MigrationManager::new();

        let m2m = ManyToManyDefinition::new("User".to_string(), "Role".to_string());

        let result = manager.generate_junction_table(&m2m);
        assert!(result.is_ok());

        let sql = result.unwrap();
        assert!(sql.contains("CREATE TABLE IF NOT EXISTS"));
        assert!(sql.contains("user_id"));
        assert!(sql.contains("role_id"));
        assert!(sql.contains("PRIMARY KEY"));
        assert!(sql.contains("FOREIGN KEY"));
    }

    #[test]
    fn test_generate_junction_table_custom() {
        let manager = MigrationManager::new();

        let m2m = ManyToManyDefinition::new("User".to_string(), "Role".to_string())
            .with_junction_table("user_role_mapping".to_string())
            .with_fields("usr_id".to_string(), "rol_id".to_string());

        let sql = manager.generate_junction_table(&m2m).unwrap();
        assert!(sql.contains("user_role_mapping"));
        assert!(sql.contains("usr_id"));
        assert!(sql.contains("rol_id"));
    }

    #[test]
    fn test_generate_all_junction_tables() {
        let manager = MigrationManager::new();
        let mut registry = EntityRegistry::new();

        // Create entities with M2M relationships
        let base_schema1 = SchemaDefinition::new("User".to_string());
        let mut vbr_schema1 = VbrSchemaDefinition::new(base_schema1);
        vbr_schema1
            .many_to_many
            .push(ManyToManyDefinition::new("User".to_string(), "Role".to_string()));
        let entity1 = Entity::new("User".to_string(), vbr_schema1);
        registry.register(entity1).unwrap();

        let base_schema2 = SchemaDefinition::new("Role".to_string());
        let vbr_schema2 = VbrSchemaDefinition::new(base_schema2);
        let entity2 = Entity::new("Role".to_string(), vbr_schema2);
        registry.register(entity2).unwrap();

        let result = manager.generate_all_junction_tables(&registry);
        assert!(result.is_ok());

        let junction_tables = result.unwrap();
        assert_eq!(junction_tables.len(), 1);
    }

    #[test]
    fn test_generate_all_junction_tables_no_duplicates() {
        let manager = MigrationManager::new();
        let mut registry = EntityRegistry::new();

        // Both entities define the same M2M relationship
        let base_schema1 = SchemaDefinition::new("User".to_string());
        let mut vbr_schema1 = VbrSchemaDefinition::new(base_schema1);
        vbr_schema1
            .many_to_many
            .push(ManyToManyDefinition::new("User".to_string(), "Role".to_string()));
        let entity1 = Entity::new("User".to_string(), vbr_schema1);
        registry.register(entity1).unwrap();

        let base_schema2 = SchemaDefinition::new("Role".to_string());
        let mut vbr_schema2 = VbrSchemaDefinition::new(base_schema2);
        vbr_schema2
            .many_to_many
            .push(ManyToManyDefinition::new("Role".to_string(), "User".to_string()));
        let entity2 = Entity::new("Role".to_string(), vbr_schema2);
        registry.register(entity2).unwrap();

        let junction_tables = manager.generate_all_junction_tables(&registry).unwrap();
        // Should only create one junction table, not two
        assert_eq!(junction_tables.len(), 1);
    }

    #[tokio::test]
    async fn test_migrate() {
        let manager = MigrationManager::new();
        let mut database = InMemoryDatabase::new().await.unwrap();
        database.initialize().await.unwrap();

        let entity = create_test_entity("User");
        let entities = vec![entity];

        let result = manager.migrate(&entities, &mut database).await;
        assert!(result.is_ok());

        // Verify table was created
        assert!(database.table_exists("users").await.unwrap());
    }

    #[tokio::test]
    async fn test_create_table_for_entity() {
        let mut database = InMemoryDatabase::new().await.unwrap();
        database.initialize().await.unwrap();

        let entity = create_test_entity("Product");

        let result = create_table_for_entity(&database, &entity).await;
        assert!(result.is_ok());

        // Verify table was created
        assert!(database.table_exists("products").await.unwrap());
    }

    #[tokio::test]
    async fn test_create_junction_tables() {
        let mut database = InMemoryDatabase::new().await.unwrap();
        database.initialize().await.unwrap();

        let mut registry = EntityRegistry::new();

        // Create entities with M2M relationship
        let base_schema1 = SchemaDefinition::new("User".to_string());
        let mut vbr_schema1 = VbrSchemaDefinition::new(base_schema1);
        vbr_schema1
            .many_to_many
            .push(ManyToManyDefinition::new("User".to_string(), "Group".to_string()));
        let entity1 = Entity::new("User".to_string(), vbr_schema1);
        registry.register(entity1).unwrap();

        let result = create_junction_tables(&database, &registry).await;
        assert!(result.is_ok());
    }

    // Helper function tests
    #[test]
    fn test_rust_type_to_sql_type() {
        assert_eq!(rust_type_to_sql_type("string").unwrap(), "TEXT");
        assert_eq!(rust_type_to_sql_type("String").unwrap(), "TEXT");
        assert_eq!(rust_type_to_sql_type("integer").unwrap(), "INTEGER");
        assert_eq!(rust_type_to_sql_type("int").unwrap(), "INTEGER");
        assert_eq!(rust_type_to_sql_type("i32").unwrap(), "INTEGER");
        assert_eq!(rust_type_to_sql_type("float").unwrap(), "REAL");
        assert_eq!(rust_type_to_sql_type("f64").unwrap(), "REAL");
        assert_eq!(rust_type_to_sql_type("boolean").unwrap(), "INTEGER");
        assert_eq!(rust_type_to_sql_type("bool").unwrap(), "INTEGER");
        assert_eq!(rust_type_to_sql_type("date").unwrap(), "TEXT");
        assert_eq!(rust_type_to_sql_type("datetime").unwrap(), "TEXT");
        assert_eq!(rust_type_to_sql_type("unknown_type").unwrap(), "TEXT");
    }

    #[test]
    fn test_cascade_action_to_sql() {
        assert_eq!(cascade_action_to_sql(CascadeAction::NoAction), "NO ACTION");
        assert_eq!(cascade_action_to_sql(CascadeAction::Cascade), "CASCADE");
        assert_eq!(cascade_action_to_sql(CascadeAction::SetNull), "SET NULL");
        assert_eq!(cascade_action_to_sql(CascadeAction::SetDefault), "SET DEFAULT");
        assert_eq!(cascade_action_to_sql(CascadeAction::Restrict), "RESTRICT");
    }

    #[test]
    fn test_value_to_sql_default() {
        // String
        let result = value_to_sql_default(&serde_json::json!("test")).unwrap();
        assert_eq!(result, "'test'");

        // String with quotes
        let result = value_to_sql_default(&serde_json::json!("it's")).unwrap();
        assert_eq!(result, "'it''s'");

        // Number
        let result = value_to_sql_default(&serde_json::json!(42)).unwrap();
        assert_eq!(result, "42");

        // Boolean true
        let result = value_to_sql_default(&serde_json::json!(true)).unwrap();
        assert_eq!(result, "1");

        // Boolean false
        let result = value_to_sql_default(&serde_json::json!(false)).unwrap();
        assert_eq!(result, "0");

        // Null
        let result = value_to_sql_default(&serde_json::json!(null)).unwrap();
        assert_eq!(result, "NULL");

        // Unsupported type (array)
        let result = value_to_sql_default(&serde_json::json!([]));
        assert!(result.is_err());
    }

    #[test]
    fn test_field_to_column_definition_with_auto_generation() {
        let manager = MigrationManager::new();

        let base_schema = SchemaDefinition::new("User".to_string())
            .with_field(FieldDefinition::new("id".to_string(), "string".to_string()));

        let mut vbr_schema = VbrSchemaDefinition::new(base_schema);
        vbr_schema.auto_generation.insert("id".to_string(), AutoGenerationRule::Uuid);

        // Access field from base schema
        let field = &vbr_schema.base.fields[0];
        let result = manager.field_to_column_definition(field, &vbr_schema);
        assert!(result.is_ok());

        let column_def = result.unwrap();
        assert!(column_def.contains("id"));
        assert!(column_def.contains("TEXT"));
    }

    #[test]
    fn test_field_to_column_definition_with_default() {
        let manager = MigrationManager::new();

        let mut field = FieldDefinition::new("status".to_string(), "string".to_string());
        field.default = Some(serde_json::json!("active"));
        let base_schema = SchemaDefinition::new("User".to_string()).with_field(field);

        let vbr_schema = VbrSchemaDefinition::new(base_schema);
        let field_ref = &vbr_schema.base.fields[0];

        let column_def = manager.field_to_column_definition(field_ref, &vbr_schema).unwrap();
        assert!(column_def.contains("DEFAULT 'active'"));
    }

    #[test]
    fn test_field_to_column_definition_required() {
        let manager = MigrationManager::new();

        let base_schema = SchemaDefinition::new("User".to_string())
            .with_field(FieldDefinition::new("email".to_string(), "string".to_string()));

        let vbr_schema = VbrSchemaDefinition::new(base_schema);
        let field = &vbr_schema.base.fields[0];

        let column_def = manager.field_to_column_definition(field, &vbr_schema).unwrap();
        assert!(column_def.contains("NOT NULL"));
    }

    #[test]
    fn test_generate_create_table_with_composite_primary_key() {
        let manager = MigrationManager::new();

        let base_schema = SchemaDefinition::new("UserRole".to_string());
        let vbr_schema = VbrSchemaDefinition::new(base_schema)
            .with_primary_key(vec!["user_id".to_string(), "role_id".to_string()]);

        let entity = Entity::new("UserRole".to_string(), vbr_schema);

        let sql = manager.generate_create_table(&entity).unwrap();
        assert!(sql.contains("PRIMARY KEY (user_id, role_id)"));
    }

    #[test]
    fn test_generate_create_table_with_auto_increment() {
        let manager = MigrationManager::new();

        let base_schema = SchemaDefinition::new("Counter".to_string())
            .with_field(FieldDefinition::new("id".to_string(), "integer".to_string()));

        let mut vbr_schema = VbrSchemaDefinition::new(base_schema);
        vbr_schema
            .auto_generation
            .insert("id".to_string(), AutoGenerationRule::AutoIncrement);

        let entity = Entity::new("Counter".to_string(), vbr_schema);

        let sql = manager.generate_create_table(&entity).unwrap();
        assert!(sql.contains("AUTOINCREMENT"));
    }
}