ILIKE operator translation (#35)

* feat: Translating ILIKE and NOT ILIKE operators to LIKE and NOT LIKE * test: Testing ILIKE and NOT ILIKE translation
aadel · Aug 6, 2024 · 1ee5d85 · 1ee5d85
1 parent 96bd392
commit 1ee5d85
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/src/sqlalchemy_solr/base.py b/src/sqlalchemy_solr/base.py
@@ -182,6 +182,13 @@ def visit_clauselist(self, clauselist, **kw):
     def visit_array(self, element, **kw):
         return f"ARRAY[{self.visit_clauselist(element, **kw)}]"
 
+    # Solr LIKE and NOT LIKE operators are case insensetive
+    def visit_ilike_op_binary(self, binary, operator, **kw):
+        return super().visit_like_op_binary(binary, operator, **kw)
+
+    def visit_not_ilike_op_binary(self, binary, operator, **kw):
+        return super().visit_not_like_op_binary(binary, operator, **kw)
+
     def unescape_colon(self, s: str) -> str:
         """Unescape colon if present in datetime value"""
         return s.replace(r"\:", ":")

diff --git a/tests/test_operator_translation.py b/tests/test_operator_translation.py
@@ -0,0 +1,39 @@
+from sqlalchemy import select
+from tests.setup import prepare_orm
+
+
+class TestOperatorTranslation:
+
+    def test_ilike_operator_translation(self, settings):
+        engine, t = prepare_orm(settings)
+
+        select_statement_1 = (
+            "SELECT sales_test_.`CITY_s` \nFROM sales_test_ "
+            "\nWHERE sales_test_.`CITY_s` LIKE ?\n LIMIT ?"
+        )
+
+        qry = (select(t.c.CITY_s).where(t.c.CITY_s.ilike("%york%"))).limit(
+            1
+        )  # pylint: disable=unsubscriptable-object
+
+        with engine.connect() as connection:
+            result = connection.execute(qry)
+
+        assert result.context.statement == select_statement_1
+
+    def test_not_ilike_operator_translation(self, settings):
+        engine, t = prepare_orm(settings)
+
+        select_statement_1 = (
+            "SELECT sales_test_.`CITY_s` \nFROM sales_test_ "
+            "\nWHERE sales_test_.`CITY_s` NOT LIKE ?\n LIMIT ?"
+        )
+
+        qry = (select(t.c.CITY_s).where(t.c.CITY_s.notilike("%york%"))).limit(
+            1
+        )  # pylint: disable=unsubscriptable-object
+
+        with engine.connect() as connection:
+            result = connection.execute(qry)
+
+        assert result.context.statement == select_statement_1