1 files changed, 148 insertions, 0 deletions

diff --git a/spec/cli/test_compilation.sh b/spec/cli/test_compilation.sh
new file mode 100755
index 0000000..4ddde5a
--- /dev/null
+++ b/spec/cli/test_compilation.sh
@@ -0,0 +1,148 @@
+#!/bin/bash
+# Test Compilation Functionality for YueScript CLI
+# Tests: File compilation, directory compilation, output options
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+source "$SCRIPT_DIR/cli_test_helper.sh"
+
+# Check binary
+check_yue_binary
+
+# Setup test environment
+setup_test_env
+TMP_DIR=$(get_test_tmp_dir)
+
+echo "========================================"
+echo "Testing Compilation Functionality"
+echo "========================================"
+echo ""
+
+# Test 1: Compile a simple file to stdout
+echo "Testing simple file compilation to stdout..."
+cat > "$TMP_DIR/simple.yue" << 'EOF'
+print "Hello, World!"
+EOF
+
+assert_output_contains "Compile simple file to stdout" "print" $YUE_BIN -p "$TMP_DIR/simple.yue"
+
+# Test 2: Compile a simple file to disk
+echo ""
+echo "Testing file compilation to disk..."
+cat > "$TMP_DIR/test1.yue" << 'EOF'
+x = 1 + 2
+print x
+EOF
+
+assert_success "Compile test1.yue to disk" $YUE_BIN "$TMP_DIR/test1.yue"
+assert_file_exists "Output file test1.lua should exist" "$TMP_DIR/test1.lua"
+
+# Test 3: Compile with -o option
+echo ""
+echo "Testing compilation with -o option..."
+cat > "$TMP_DIR/test2.yue" << 'EOF'
+x = 10
+print x
+EOF
+
+assert_success "Compile with -o option" $YUE_BIN -o "$TMP_DIR/output.lua" "$TMP_DIR/test2.yue"
+assert_file_exists "Custom output file should exist" "$TMP_DIR/output.lua"
+
+# Test 4: Compile directory with -t option (target directory)
+echo ""
+echo "Testing compilation with -t option..."
+mkdir -p "$TMP_DIR/src"
+mkdir -p "$TMP_DIR/build"
+cat > "$TMP_DIR/src/test3.yue" << 'EOF'
+print "test"
+EOF
+
+assert_success "Compile directory with -t option" $YUE_BIN "$TMP_DIR/src" -t "$TMP_DIR/build"
+assert_file_exists "Output should be in target directory" "$TMP_DIR/build/test3.lua"
+
+# Test 5: Compile directory recursively
+echo ""
+echo "Testing directory compilation..."
+mkdir -p "$TMP_DIR/project/src"
+mkdir -p "$TMP_DIR/project/build"
+
+cat > "$TMP_DIR/project/src/file1.yue" << 'EOF'
+print "file1"
+EOF
+
+cat > "$TMP_DIR/project/src/file2.yue" << 'EOF'
+print "file2"
+EOF
+
+assert_success "Compile entire directory" $YUE_BIN "$TMP_DIR/project/src" -t "$TMP_DIR/project/build"
+# Files are compiled directly in target directory, not preserving src subdir
+assert_file_exists "file1.lua should exist" "$TMP_DIR/project/build/file1.lua"
+assert_file_exists "file2.lua should exist" "$TMP_DIR/project/build/file2.lua"
+
+# Test 6: Compile with line numbers (-l)
+echo ""
+echo "Testing compilation with line numbers..."
+cat > "$TMP_DIR/test_line.yue" << 'EOF'
+print "line test"
+EOF
+
+assert_success "Compile with line numbers" $YUE_BIN -l "$TMP_DIR/test_line.yue" -o "$TMP_DIR/test_line.lua"
+assert_output_contains "Compiled file should have line comment" "yue" cat "$TMP_DIR/test_line.lua"
+
+# Test 7: Compile with spaces instead of tabs (-s)
+echo ""
+echo "Testing compilation with spaces..."
+cat > "$TMP_DIR/test_spaces.yue" << 'EOF'
+x = 1
+EOF
+
+assert_success "Compile with spaces option" $YUE_BIN -s "$TMP_DIR/test_spaces.yue" -o "$TMP_DIR/test_spaces.lua"
+
+# Test 8: Compile with minify (-m)
+echo ""
+echo "Testing compilation with minify..."
+cat > "$TMP_DIR/test_minify.yue" << 'EOF'
+-- this is a comment
+x = 1 + 2
+print x
+EOF
+
+assert_success "Compile with minify option" $YUE_BIN -m "$TMP_DIR/test_minify.yue" -o "$TMP_DIR/test_minify.lua"
+assert_file_exists "Minified file should exist" "$TMP_DIR/test_minify.lua"
+
+# Test 9: Stdin/stdout compilation
+echo ""
+echo "Testing stdin/stdout compilation..."
+echo 'print "stdin test"' | assert_output_contains "Compile from stdin" "print" $YUE_BIN -
+
+# Test 10: Glob variable dumping (-g)
+echo ""
+echo "Testing global variable dumping..."
+cat > "$TMP_DIR/test_globals.yue" << 'EOF'
+local x = 1
+print unknown_global
+EOF
+
+# -g dumps globals in format: NAME LINE COLUMN
+assert_output_contains "Dump global variables" "unknown_global" $YUE_BIN -g "$TMP_DIR/test_globals.yue"
+
+# Test 11: Benchmark compilation (-b)
+echo ""
+echo "Testing benchmark compilation..."
+cat > "$TMP_DIR/test_bench.yue" << 'EOF'
+print "benchmark"
+EOF
+
+assert_output_contains "Benchmark should show compile time" "Compile time:" $YUE_BIN -b "$TMP_DIR/test_bench.yue"
+
+# Test 12: Target version option
+echo ""
+echo "Testing target version option..."
+cat > "$TMP_DIR/test_target.yue" << 'EOF'
+print "target test"
+EOF
+
+assert_success "Compile with target 5.1" $YUE_BIN "$TMP_DIR/test_target.yue" -o "$TMP_DIR/test_target.lua" --target 5.1
+assert_file_exists "Target version compilation should succeed" "$TMP_DIR/test_target.lua"
+
+echo ""
+print_summary