Merge pull request #112 from cipherstash/expand-order-by-testing

Expand ordering test using operator family

Author: tobyhede

src/operators/order_by_test.sql

@@ -25,4 +25,51 @@ DECLARE
         format('SELECT id FROM ore WHERE e < %L ORDER BY eql_v2.order_by(e) ASC LIMIT 1', ore_term),
         '1');
   END;
+$$ LANGUAGE plpgsql;
+
+
+--
+-- ORE - ORDER BY without function
+--
+DO $$
+DECLARE
+    e eql_v2_encrypted;
+    ore_term eql_v2_encrypted;
+  BEGIN
+      -- Pull a record from the ore table with value of "42"
+      SELECT ore.e FROM ore WHERE id = 42 INTO ore_term;
+
+      -- lt
+      PERFORM assert_count(
+        'ORDER BY eql_v2.order_by(e) DESC',
+        format('SELECT id FROM ore WHERE e < %L ORDER BY e DESC', ore_term),
+        41);
+
+      PERFORM assert_result(
+        'ORDER BY e DESC returns correct record',
+        format('SELECT id FROM ore WHERE e < %L ORDER BY e DESC LIMIT 1', ore_term),
+        '41');
+
+      PERFORM assert_result(
+        'ORDER BY e ASC',
+        format('SELECT id FROM ore WHERE e < %L ORDER BY e ASC LIMIT 1', ore_term),
+        '1');
+
+      -- gt
+      PERFORM assert_count(
+        'ORDER BY eql_v2.order_by(e) DESC',
+        format('SELECT id FROM ore WHERE e > %L ORDER BY e ASC', ore_term),
+        57);
+
+      PERFORM assert_result(
+        'ORDER BY e DESC returns correct record',
+        format('SELECT id FROM ore WHERE e > %L ORDER BY e DESC LIMIT 1', ore_term),
+        '99');
+
+      PERFORM assert_result(
+        'ORDER BY e ASC',
+        format('SELECT id FROM ore WHERE e > %L ORDER BY e ASC LIMIT 1', ore_term),
+        '43');
+
+  END;
 $$ LANGUAGE plpgsql;