diff options
| -rw-r--r-- | tests/test_functions.py | 25 |
1 files changed, 21 insertions, 4 deletions
diff --git a/tests/test_functions.py b/tests/test_functions.py index 5ab1586..ff8625a 100644 --- a/tests/test_functions.py +++ b/tests/test_functions.py @@ -15,7 +15,7 @@ from sqlparse.filters import compact from sqlparse.functions import getcolumns, getlimit, IsType -class Test_SQL(TestCase): +class Test_IncludeStatement(TestCase): sql = """-- type: script -- return: integer @@ -25,6 +25,24 @@ class Test_SQL(TestCase): VALUES(:inode) LIMIT 1""" + def test_includeStatement(self): + stream = compact(tokenize(self.sql), 'tests/files') + + result = Tokens2Unicode(stream) + + self.assertEqual(result, + 'INSERT INTO dir_entries(type)VALUES(:type);INSERT INTO ' + 'directories(inode)VALUES(:inode)LIMIT 1') + + +class Test_SQL(TestCase): + sql = """-- type: script + -- return: integer + + INSERT INTO directories(inode) + VALUES(:inode) + LIMIT 1""" + sql2 = """SELECT child_entry,asdf AS inode, creation FROM links WHERE parent_dir == :parent_dir AND name == :name @@ -63,13 +81,12 @@ LIMIT 1""" class Test_Compact(Test_SQL): def test_compact1(self): - stream = compact(tokenize(self.sql), 'tests/files') + stream = compact(tokenize(self.sql)) result = Tokens2Unicode(stream) self.assertEqual(result, - 'INSERT INTO dir_entries(type)VALUES(:type);INSERT INTO ' - 'directories(inode)VALUES(:inode)LIMIT 1') + 'INSERT INTO directories(inode)VALUES(:inode)LIMIT 1') def test_compact2(self): stream = tokenize(self.sql2) |