Add the first test

[sqlconvert.git] / tests / test_tokens.py

diff --git a/tests/test_tokens.py b/tests/test_tokens.py
new file mode 100755 (executable)
index 0000000..c8e786f
--- /dev/null
+++ b/tests/test_tokens.py
@@ -0,0 +1,21 @@
+#! /usr/bin/env python
+
+
+import unittest
+from sqlparse import parse
+
+from mysql2sql.process_tokens import requote_names
+from mysql2sql.print_tokens import get_tokens_str
+from tests import main
+
+
+class TestTokens(unittest.TestCase):
+    def test_requote(self):
+        parsed = parse("select * from `T`")[0]
+        requote_names(parsed)
+        query = get_tokens_str(parsed)
+        self.assertEqual(query, 'SELECT * FROM "T";\n')
+
+
+if __name__ == "__main__":
+    main()