1 files changed, 189 insertions, 0 deletions

diff --git a/spec/cli/test_reserve_comments.sh b/spec/cli/test_reserve_comments.sh
new file mode 100755
index 0000000..6c0c1a7
--- /dev/null
+++ b/spec/cli/test_reserve_comments.sh
@@ -0,0 +1,189 @@
+#!/bin/bash
+# Test Reserve Comments Functionality for YueScript CLI
+# Tests: -c, --reserve-comments option
+
+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 Reserve Comments (-c) Option"
+echo "========================================"
+echo ""
+
+# Test 1: Reserve top-level comments
+echo "Testing top-level comments preservation..."
+cat > "$TMP_DIR/top_level.yue" << 'EOF'
+-- Top level comment
+x = 1
+-- Another comment
+y = 2
+EOF
+
+assert_output_contains "Reserve top-level comments" "Top level comment" $YUE_BIN -c -p "$TMP_DIR/top_level.yue"
+assert_output_contains "Reserve second comment" "Another comment" $YUE_BIN -c -p "$TMP_DIR/top_level.yue"
+
+# Test 2: Without -c option, comments should not appear
+echo ""
+echo "Testing comments are removed without -c option..."
+assert_output_not_contains "Comments should be removed without -c" "Top level comment" $YUE_BIN -p "$TMP_DIR/top_level.yue"
+
+# Test 3: Reserve comments in table
+echo ""
+echo "Testing comments in tables..."
+cat > "$TMP_DIR/table_comments.yue" << 'EOF'
+t = {
+    -- First value comment
+    1,
+    -- Second value comment
+    2
+}
+EOF
+
+assert_output_contains "Table comments should be preserved" "First value comment" $YUE_BIN -c -p "$TMP_DIR/table_comments.yue"
+assert_output_contains "Table second comment preserved" "Second value comment" $YUE_BIN -c -p "$TMP_DIR/table_comments.yue"
+
+# Test 4: Reserve comments in if statement
+echo ""
+echo "Testing comments in if statements..."
+cat > "$TMP_DIR/if_comments.yue" << 'EOF'
+if true
+    -- Inside if block
+    print "test"
+EOF
+
+assert_output_contains "If block comments should be preserved" "Inside if block" $YUE_BIN -c -p "$TMP_DIR/if_comments.yue"
+
+# Test 5: Reserve comments in function
+echo ""
+echo "Testing comments in functions..."
+cat > "$TMP_DIR/func_comments.yue" << 'EOF'
+func = =>
+    -- Inside function
+    print "hello"
+EOF
+
+assert_output_contains "Function comments should be preserved" "Inside function" $YUE_BIN -c -p "$TMP_DIR/func_comments.yue"
+
+# Test 6: Reserve comments with empty lines
+echo ""
+echo "Testing empty lines preservation..."
+cat > "$TMP_DIR/empty_lines.yue" << 'EOF'
+-- First comment
+x = 1
+-- Second comment
+EOF
+
+OUTPUT_WITHOUT_C=$($YUE_BIN -p "$TMP_DIR/empty_lines.yue")
+OUTPUT_WITH_C=$($YUE_BIN -c -p "$TMP_DIR/empty_lines.yue")
+if [ $? -eq 0 ]; then
+    # Count newlines - with -c should have more (or equal) lines due to comment preservation
+    LINES_WITHOUT_C=$(echo "$OUTPUT_WITHOUT_C" | wc -l)
+    LINES_WITH_C=$(echo "$OUTPUT_WITH_C" | wc -l)
+    if [ $LINES_WITH_C -ge $LINES_WITHOUT_C ]; then
+        echo -e "${GREEN}✓${NC} Empty lines and comments should be preserved"
+        TESTS_PASSED=$((TESTS_PASSED + 1))
+    else
+        echo -e "${RED}✗${NC} Empty lines and comments should be preserved"
+        echo -e "  ${YELLOW}Lines without -c: $LINES_WITHOUT_C, with -c: $LINES_WITH_C${NC}"
+        TESTS_FAILED=$((TESTS_FAILED + 1))
+    fi
+    TESTS_RUN=$((TESTS_RUN + 1))
+else
+    echo -e "${RED}✗${NC} Empty lines test failed"
+    TESTS_FAILED=$((TESTS_FAILED + 1))
+    TESTS_RUN=$((TESTS_RUN + 1))
+fi
+
+# Test 7: Reserve comments in table with TableBlock syntax
+echo ""
+echo "Testing comments in TableBlock..."
+cat > "$TMP_DIR/tableblock_comments.yue" << 'EOF'
+tbl = {
+    -- Key comment
+    key: "value"
+    -- Another key
+    another: 123
+}
+EOF
+
+assert_output_contains "TableBlock key comment preserved" "Key comment" $YUE_BIN -c -p "$TMP_DIR/tableblock_comments.yue"
+assert_output_contains "TableBlock second comment preserved" "Another key" $YUE_BIN -c -p "$TMP_DIR/tableblock_comments.yue"
+
+# Test 8: Reserve comments - long form option
+echo ""
+echo "Testing --reserve-comments long form option..."
+assert_output_contains "Long form option should preserve comments" "First value comment" $YUE_BIN --reserve-comments -p "$TMP_DIR/table_comments.yue"
+
+# Test 9: Compile to file with reserve comments
+echo ""
+echo "Testing compilation to file with comments..."
+cat > "$TMP_DIR/file_comment.yue" << 'EOF'
+-- This is a test
+value = 42
+EOF
+
+assert_success "Compile with -c to file" $YUE_BIN -c "$TMP_DIR/file_comment.yue" -o "$TMP_DIR/file_comment.lua"
+assert_file_exists "Output file should exist" "$TMP_DIR/file_comment.lua"
+assert_output_contains "Compiled file should contain comments" "This is a test" cat "$TMP_DIR/file_comment.lua"
+
+# Test 10: Reserve comments with multiple statements
+echo ""
+echo "Testing comments with multiple statements..."
+cat > "$TMP_DIR/multi_stmt.yue" << 'EOF'
+-- Assign x
+x = 1
+-- Assign y
+y = 2
+-- Assign z
+z = 3
+EOF
+
+OUTPUT=$($YUE_BIN -c -p "$TMP_DIR/multi_stmt.yue")
+if [ $? -eq 0 ]; then
+    if echo "$OUTPUT" | grep -q "Assign x" && echo "$OUTPUT" | grep -q "Assign y" && echo "$OUTPUT" | grep -q "Assign z"; then
+        echo -e "${GREEN}✓${NC} All comments should be preserved"
+        TESTS_PASSED=$((TESTS_PASSED + 1))
+    else
+        echo -e "${RED}✗${NC} All comments should be preserved"
+        echo -e "  ${YELLOW}Output: $OUTPUT${NC}"
+        TESTS_FAILED=$((TESTS_FAILED + 1))
+    fi
+    TESTS_RUN=$((TESTS_RUN + 1))
+else
+    echo -e "${RED}✗${NC} Multiple statements test failed"
+    TESTS_FAILED=$((TESTS_FAILED + 1))
+    TESTS_RUN=$((TESTS_RUN + 1))
+fi
+
+# Test 11: Comments in while loop
+echo ""
+echo "Testing comments in while loop..."
+cat > "$TMP_DIR/while_comments.yue" << 'EOF'
+while true
+    -- Loop body comment
+    print "looping"
+    break
+EOF
+
+assert_output_contains "While loop comments preserved" "Loop body comment" $YUE_BIN -c -p "$TMP_DIR/while_comments.yue"
+
+# Test 12: Comments in for loop
+echo ""
+echo "Testing comments in for loop..."
+cat > "$TMP_DIR/for_comments.yue" << 'EOF'
+for i = 1, 3
+    -- For loop comment
+    print i
+EOF
+
+assert_output_contains "For loop comments preserved" "For loop comment" $YUE_BIN -c -p "$TMP_DIR/for_comments.yue"
+
+echo ""
+print_summary