100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
"""EXEC SQL 内 DB2 schema 限定表名（SCHEMA.TABLE）剥离测试。
|
||
|
||
gixsql 会把 `FROM SALARYDB.EMP-MASTER` 原样归一化为 `FROM SALARYDB.EMP_MASTER`
|
||
注入 GIXSQL SQL 字符串；SQLite 无 schema 对象，OPEN CURSOR 报
|
||
'no such table: SALARYDB.EMP_MASTER' → SQLCODE≠0 → EMP-OPEN 错误分支 → ABEND，
|
||
主循环 2000MAJSOR 永不执行（SHA02MNC 覆盖率 66.7% 根因）。
|
||
|
||
修复：gixpp 前把 EXEC SQL 块内 FROM/INTO/UPDATE/JOIN 后的 SCHEMA.TABLE
|
||
剥掉限定符（通用规则，处理任意 schema 名，无程序硬编码）。
|
||
"""
|
||
|
||
import sys, os
|
||
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
|
||
from runners.gixsql_runner import (
|
||
_strip_schema_qualifiers,
|
||
_normalize_schema_qualifiers,
|
||
)
|
||
|
||
|
||
def test_from_schema_table_stripped():
|
||
src = (
|
||
" EXEC SQL\n"
|
||
" DECLARE C2 CURSOR FOR\n"
|
||
" SELECT EMP-ID, EMP-NAME\n"
|
||
" FROM SALARYDB.EMP-MASTER\n"
|
||
" ORDER BY EMP-ID\n"
|
||
" END-EXEC.\n"
|
||
)
|
||
out = _strip_schema_qualifiers(src)
|
||
assert "FROM SALARYDB.EMP-MASTER" not in out
|
||
assert "FROM EMP-MASTER" in out
|
||
|
||
|
||
def test_from_three_part_qualified_stripped():
|
||
src = "EXEC SQL SELECT A FROM DB2INST1.PAYROLL.TIMESHEET END-EXEC."
|
||
out = _strip_schema_qualifiers(src)
|
||
assert "FROM TIMESHEET" in out
|
||
assert "DB2INST1.PAYROLL" not in out
|
||
|
||
|
||
def test_update_schema_table_stripped():
|
||
src = "EXEC SQL UPDATE SALARYDB.EMP-MASTER SET DEPT-CODE = '01' WHERE EMP-ID = :X END-EXEC."
|
||
out = _strip_schema_qualifiers(src)
|
||
assert "UPDATE EMP-MASTER" in out
|
||
|
||
|
||
def test_into_schema_table_stripped():
|
||
src = "EXEC SQL INSERT INTO SALARYDB.EMP-MASTER (EMP-ID) VALUES (:X) END-EXEC."
|
||
out = _strip_schema_qualifiers(src)
|
||
assert "INSERT INTO EMP-MASTER" in out
|
||
|
||
|
||
def test_join_schema_table_stripped():
|
||
src = ("EXEC SQL SELECT A.ID FROM A "
|
||
"JOIN SALARYDB.EMP-MASTER B ON A.ID = B.EMP-ID END-EXEC.")
|
||
out = _strip_schema_qualifiers(src)
|
||
assert "JOIN EMP-MASTER B" in out
|
||
|
||
|
||
def test_column_alias_qualifier_untouched():
|
||
"""列限定（E.EMP-ID / B.DEPT-CODE，非表名位置）不得剥离。"""
|
||
src = ("EXEC SQL SELECT E.EMP-ID FROM EMP-MASTER E "
|
||
"WHERE E.DEPT-CODE = '01' END-EXEC.")
|
||
out = _strip_schema_qualifiers(src)
|
||
assert "SELECT E.EMP-ID" in out
|
||
assert "E.DEPT-CODE" in out
|
||
assert "FROM EMP-MASTER E" in out
|
||
|
||
|
||
def test_hostvar_into_untouched():
|
||
"""SELECT ... INTO :DBV-X（宿主变量，非 SCHEMA.TABLE）不得剥离。"""
|
||
src = "EXEC SQL SELECT EMP-NAME INTO :DBV-EMP-NAME FROM EMP-MASTER END-EXEC."
|
||
out = _strip_schema_qualifiers(src)
|
||
assert "INTO :DBV-EMP-NAME" in out
|
||
|
||
|
||
def test_unqualified_table_untouched():
|
||
src = "EXEC SQL SELECT EMP-ID FROM EMP-MASTER ORDER BY EMP-ID END-EXEC."
|
||
assert _strip_schema_qualifiers(src) == src
|
||
|
||
|
||
def test_wrapper_only_touches_exec_sql_blocks():
|
||
text = " MOVE 'FROM A.B' TO WS-X.\n"
|
||
assert _normalize_schema_qualifiers(text) == text
|
||
|
||
|
||
def test_wrapper_strips_inside_exec_sql_only():
|
||
text = (
|
||
" MOVE 'X' TO WS-A.\n"
|
||
" EXEC SQL\n"
|
||
" SELECT EMP-ID\n"
|
||
" FROM SALARYDB.EMP-MASTER\n"
|
||
" END-EXEC.\n"
|
||
" MOVE 'FROM C.D' TO WS-B.\n"
|
||
)
|
||
out = _normalize_schema_qualifiers(text)
|
||
assert "FROM SALARYDB.EMP-MASTER" not in out
|
||
assert "FROM EMP-MASTER" in out
|
||
assert "MOVE 'FROM C.D' TO WS-B" in out
|