diff options
| -rw-r--r-- | tests/test_functions.py | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/tests/test_functions.py b/tests/test_functions.py index d5d8f91..5ab1586 100644 --- a/tests/test_functions.py +++ b/tests/test_functions.py @@ -63,17 +63,29 @@ LIMIT 1""" class Test_Compact(Test_SQL): def test_compact1(self): - self.assertEqual(Tokens2Unicode(compact(self.sql, 'tests/files')), + 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') def test_compact2(self): - self.assertEqual(Tokens2Unicode(compact(self.sql2)), + stream = tokenize(self.sql2) + + result = compact(stream) + + self.assertEqual(Tokens2Unicode(result), 'SELECT child_entry,asdf AS inode,creation FROM links WHERE ' 'parent_dir==:parent_dir AND name==:name LIMIT 1') def test_compact3(self): - self.assertEqual(Tokens2Unicode(compact(self.sql3)), + stream = tokenize(self.sql3) + + result = compact(stream) + + self.assertEqual(Tokens2Unicode(result), 'SELECT 0 AS st_dev,0 AS st_uid,0 AS st_gid,dir_entries.type AS ' 'st_mode,dir_entries.inode AS st_ino,COUNT(links.child_entry)AS ' 'st_nlink,:creation AS st_ctime,dir_entries.access AS st_atime,' |