Benjamin-eecs/openrsi-commit-runtime-assets

1#!/bin/bash2set -uxo pipefail3SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"4cd /workspace5git config --global --add safe.directory /workspace6mkdir -p /logs/verifier7git checkout dd5ee369ed28f84bc74810baa0877085224f8525 -- test/test_unions.py || true8git apply --verbose --reject - <<'EOF_R2E_TEST_PATCH'9diff --git a/test/test_unions.py b/test/test_unions.py10index f0506be..c4ee03c 10064411--- a/test/test_unions.py12+++ b/test/test_unions.py13@@ -1,6 +1,23 @@14 from sql_metadata import Parser15 16 17+def test_union_column_aliases():18+    # https://github.com/macbre/sql-metadata/issues/40119+    # When UNION combines queries with the same alias,20+    # columns_aliases should aggregate all source columns21+    query = """22+    select a.A as M23+    from tab1 a24+    union all25+    select b.B as M26+    from tab2 b27+    """28+    parser = Parser(query)29+    assert parser.columns_aliases == {"M": ["tab1.A", "tab2.B"]}30+    assert parser.columns == ["tab1.A", "tab2.B"]31+    assert parser.tables == ["tab1", "tab2"]32+33+34 def test_union():35     query = """36     SELECT37 38EOF_R2E_TEST_PATCH39R2E_APPLY_RC=$?40if [ "$R2E_APPLY_RC" -ne 0 ]; then41  echo "0.000000" > /logs/verifier/reward.txt42  printf '%s' '{"reward": 0.0, "resolved": false, "parse_status": "test_patch_apply_failed"}' > /logs/verifier/reward-details.json43  echo "R2E: test_patch failed to apply (rc=$R2E_APPLY_RC) — failing closed" >&244  exit 045fi46( python -m pytest -v test/test_unions.py ) > /logs/verifier/test_output.log 2>&147TEST_EXIT_CODE=$?48cat /logs/verifier/test_output.log49python3 "$SCRIPT_DIR/verifier.py" --log /logs/verifier/test_output.log --f2p "$SCRIPT_DIR/f2p.json" --p2p "$SCRIPT_DIR/p2p.json" --test-cmds 'python -m pytest -v test/test_unions.py' --exit-code "$TEST_EXIT_CODE" --out-dir /logs/verifier || { [ "$TEST_EXIT_CODE" -eq 0 ] && echo "1.0" > /logs/verifier/reward.txt || echo "0.0" > /logs/verifier/reward.txt; }50git checkout dd5ee369ed28f84bc74810baa0877085224f8525 -- test/test_unions.py || true51exit 052