hakorune/src/tests/mir_loopform_exit_phi.rs

/*!
 * Unit tests for LoopForm v2 exit PHI generation
 *
 * Tests the build_exit_phis() implementation in loopform_builder.rs
 * Focus: predecessor tracking and PHI input generation for break statements
 */

use crate::parser::NyashParser;
use crate::mir::{MirCompiler, MirVerifier};

#[test]
fn test_loopform_exit_phi_single_break() {
    // Enable LoopForm PHI v2 and MIR verification
    std::env::set_var("NYASH_LOOPFORM_PHI_V2", "1");
    std::env::set_var("NYASH_VM_VERIFY_MIR", "1");
    std::env::set_var("NYASH_LOOPFORM_DEBUG", "1");
    std::env::set_var("NYASH_PARSER_STAGE3", "1");
    std::env::set_var("NYASH_PARSER_ALLOW_SEMICOLON", "1");

    let src = r#"
static box TestExitPhi {
  test() {
    local i = 0
    loop(i < 10) {
      if i == 5 { break }
      i = i + 1
    }
    return i
  }
}
"#;

    println!("=== Test: Single break statement ===");

    // Parse
    let ast = NyashParser::parse_from_string(src)
        .expect("parse failed");

    // Compile
    let mut mc = MirCompiler::with_options(false);
    let cr = mc.compile(ast).expect("compile failed");

    // DEBUG: Dump MIR structure
    for (fname, func) in &cr.module.functions {
        eprintln!("=== Function: {} ===", fname);
        eprintln!("Entry block: {:?}", func.entry_block);
        eprintln!("Total blocks: {}", func.blocks.len());
        for (bid, block) in &func.blocks {
            eprintln!("  Block {:?}: {} instructions, successors={:?}",
                     bid, block.instructions.len(), block.successors);
            if *bid == crate::mir::BasicBlockId(10) {
                eprintln!("    BB10 instructions:");
                for inst in &block.instructions {
                    eprintln!("      {:?}", inst);
                }
            }
        }
    }

    // MIR verification
    let mut verifier = MirVerifier::new();
    if let Err(errors) = verifier.verify_module(&cr.module) {
        for err in &errors {
            eprintln!("❌ MIR verification error: {}", err);
        }
        panic!("❌ MIR verification failed with {} errors", errors.len());
    }
    println!("✅ MIR verification passed");
}

#[test]
fn test_loopform_exit_phi_multiple_breaks() {
    std::env::set_var("NYASH_LOOPFORM_PHI_V2", "1");
    std::env::set_var("NYASH_VM_VERIFY_MIR", "1");
    std::env::set_var("NYASH_LOOPFORM_DEBUG", "1");
    std::env::set_var("NYASH_PARSER_STAGE3", "1");
    std::env::set_var("NYASH_PARSER_ALLOW_SEMICOLON", "1");

    let src = r#"
static box TestMultiBreak {
  test() {
    local i = 0
    loop(i < 10) {
      if i == 3 { break }
      if i == 5 { break }
      i = i + 1
    }
    return i
  }
}
"#;

    println!("=== Test: Multiple break statements ===");

    let ast = NyashParser::parse_from_string(src).expect("parse failed");
    let mut mc = MirCompiler::with_options(false);
    let cr = mc.compile(ast).expect("compile failed");

    let mut verifier = MirVerifier::new();
    if let Err(errors) = verifier.verify_module(&cr.module) {
        for err in &errors {
            eprintln!("❌ MIR verification error: {}", err);
        }
        panic!("❌ MIR verification failed with {} errors", errors.len());
    }
    println!("✅ MIR verification passed");
}

#[test]
fn test_loopform_exit_phi_nested_if_break() {
    std::env::set_var("NYASH_LOOPFORM_PHI_V2", "1");
    std::env::set_var("NYASH_VM_VERIFY_MIR", "1");
    std::env::set_var("NYASH_LOOPFORM_DEBUG", "1");
    std::env::set_var("NYASH_PARSER_STAGE3", "1");
    std::env::set_var("NYASH_PARSER_ALLOW_SEMICOLON", "1");

    let src = r#"
static box TestNestedBreak {
  test() {
    local i = 0
    local found = 0
    loop(i < 10) {
      if i > 5 {
        if i == 7 {
          found = 1
          break
        }
      }
      i = i + 1
    }
    return found
  }
}
"#;

    println!("=== Test: Nested if with break ===");

    let ast = NyashParser::parse_from_string(src).expect("parse failed");
    let mut mc = MirCompiler::with_options(false);
    let cr = mc.compile(ast).expect("compile failed");

    let mut verifier = MirVerifier::new();
    if let Err(errors) = verifier.verify_module(&cr.module) {
        for err in &errors {
            eprintln!("❌ MIR verification error: {}", err);
        }
        panic!("❌ MIR verification failed with {} errors", errors.len());
    }
    println!("✅ MIR verification passed");
}

#[test]
fn test_loopform_exit_phi_multiple_vars() {
    std::env::set_var("NYASH_LOOPFORM_PHI_V2", "1");
    std::env::set_var("NYASH_VM_VERIFY_MIR", "1");
    std::env::set_var("NYASH_LOOPFORM_DEBUG", "1");
    std::env::set_var("NYASH_PARSER_STAGE3", "1");
    std::env::set_var("NYASH_PARSER_ALLOW_SEMICOLON", "1");

    let src = r#"
static box TestMultiVars {
  test() {
    local i = 0
    local sum = 0
    local product = 1
    loop(i < 10) {
      if sum > 20 { break }
      sum = sum + i
      product = product * 2
      i = i + 1
    }
    return sum
  }
}
"#;

    println!("=== Test: Multiple variables with break ===");

    let ast = NyashParser::parse_from_string(src).expect("parse failed");
    let mut mc = MirCompiler::with_options(false);
    let cr = mc.compile(ast).expect("compile failed");

    let mut verifier = MirVerifier::new();
    if let Err(errors) = verifier.verify_module(&cr.module) {
        for err in &errors {
            eprintln!("❌ MIR verification error: {}", err);
        }
        panic!("❌ MIR verification failed with {} errors", errors.len());
    }
    println!("✅ MIR verification passed");
}