tests: add basic sql tests

Author: sirupsen

tests/test_diff_tables.py

@@ -6,7 +6,7 @@
 from xdiff.database import connect_to_uri
 from xdiff.diff_tables import TableDiffer, TableSegment
 
-from .common import str_to_checksum, TEST_MYSQL_CONN_STRING
+from .common import TEST_MYSQL_CONN_STRING, str_to_checksum
 
 
 class TestDiffTables(unittest.TestCase):

tests/test_sql.py

@@ -0,0 +1,31 @@
+import unittest
+
+from xdiff.database import connect_to_uri
+from xdiff.sql import Compiler, Select, TableName
+
+from .common import TEST_MYSQL_CONN_STRING
+
+
+class TestSQL(unittest.TestCase):
+ def setUp(self):
+ self.mysql = connect_to_uri(TEST_MYSQL_CONN_STRING)
+ self.compiler = Compiler(self.mysql)
+
+ def test_compile_string(self):
+ self.assertEqual("SELECT 1", self.compiler.compile("SELECT 1"))
+
+ def test_compile_int(self):
+ self.assertEqual("1", self.compiler.compile(1))
+
+ def test_compile_table_name(self):
+ self.assertEqual("`marine_mammals.walrus`", self.compiler.compile(
+ TableName(("marine_mammals", "walrus"))))
+
+ def test_compile_select(self):
+ expected_sql = "SELECT name FROM `marine_mammals.walrus`"
+ self.assertEqual(expected_sql, self.compiler.compile(
+ Select(
+ ["name"],
+ TableName(("marine_mammals", "walrus")),
+ ))
+ )