From 21125e8820b2de7fdfb11f62b3a2c5bdf12adb92 Mon Sep 17 00:00:00 2001 From: Darik Gamble Date: Wed, 5 Aug 2015 15:45:40 -0400 Subject: [PATCH] Test extract_tables with escaped table aliases --- tests/test_parseutils.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/test_parseutils.py b/tests/test_parseutils.py index 37e4aca2..09b8ea76 100644 --- a/tests/test_parseutils.py +++ b/tests/test_parseutils.py @@ -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)]