1
1
mirror of https://github.com/dbcli/pgcli.git synced 2024-10-06 02:07:53 +03:00

Test extract_tables with escaped table aliases

This commit is contained in:
Darik Gamble 2015-08-05 15:45:40 -04:00
parent d6e0a94b09
commit 21125e8820

View File

@ -14,10 +14,26 @@ def test_simple_select_single_table_schema_qualified():
tables = extract_tables('select * from abc.def')
assert tables == [('abc', 'def', None)]
def test_simple_select_single_table_double_quoted():
tables = extract_tables('select * from "Abc"')
assert tables == [(None, 'Abc', None)]
def test_simple_select_multiple_tables():
tables = extract_tables('select * from abc, def')
assert sorted(tables) == [(None, 'abc', None), (None, 'def', None)]
def test_simple_select_multiple_tables_double_quoted():
tables = extract_tables('select * from "Abc", "Def"')
assert tables == [(None, 'Abc', None), (None, 'Def', None)]
def test_simple_select_single_table_deouble_quoted_aliased():
tables = extract_tables('select * from "Abc" a')
assert tables == [(None, 'Abc', 'a')]
def test_simple_select_multiple_tables_deouble_quoted_aliased():
tables = extract_tables('select * from "Abc" a, "Def" d')
assert tables == [(None, 'Abc', 'a'), (None, 'Def', 'd')]
def test_simple_select_multiple_tables_schema_qualified():
tables = extract_tables('select * from abc.def, ghi.jkl')
assert sorted(tables) == [('abc', 'def', None), ('ghi', 'jkl', None)]