unit tests

Author: RoderickJDunn

tests/test_query.py

@@ -16,6 +16,7 @@ def normalize_spaces(s: str):
 class MockDialect(AbstractDialect):
     name = "MockDialect"
 
+    PLACEHOLDER_TABLE = None
     ROUNDS_ON_PREC_LOSS = False
 
     def quote(self, s: str) -> str:
@@ -50,6 +51,9 @@ def timestamp_value(self, t: datetime) -> str:
     def set_timezone_to_utc(self) -> str:
         return "set timezone 'UTC'"
 
+    def optimizer_hints(self, s: str):
+        return f"/*+ {s} */ " if s else ""
+
     def load_mixins(self):
         raise NotImplementedError()
 
@@ -189,6 +193,22 @@ def test_select_distinct(self):
         q = c.compile(t.select(this.b, distinct=True).select(distinct=False))
         self.assertEqual(q, "SELECT * FROM (SELECT DISTINCT b FROM a) tmp2")
 
+    def test_select_with_optimizer_hints(self):
+        c = Compiler(MockDatabase())
+        t = table("a")
+
+        q = c.compile(t.select(this.b, optimizer_hints='PARALLEL(a 16)'))
+        assert q == "SELECT /*+ PARALLEL(a 16) */ b FROM a"
+
+        q = c.compile(t.where(this.b > 10).select(this.b, optimizer_hints='PARALLEL(a 16)'))
+        self.assertEqual(q, "SELECT /*+ PARALLEL(a 16) */ b FROM a WHERE (b > 10)")
+
+        q = c.compile(t.limit(10).select(this.b, optimizer_hints='PARALLEL(a 16)'))
+        self.assertEqual(q, "SELECT /*+ PARALLEL(a 16) */ b FROM (SELECT * FROM a LIMIT 10) tmp1")
+
+        q = c.compile(t.select(this.a).group_by(this.b).agg(this.c).select(optimizer_hints='PARALLEL(a 16)'))
+        self.assertEqual(q, "SELECT /*+ PARALLEL(a 16) */ * FROM (SELECT b, c FROM (SELECT a FROM point) tmp1 GROUP BY 1) tmp2")
+
     def test_table_ops(self):
         c = Compiler(MockDatabase())
         a = table("a").select(this.x)