test(sqlx): add selector-based ORE comparison tests

Add 13 Rust/SQLx test cases for selector-based ORE comparisons with index fallback:
- Tests for <, >, <=, >= operators with e->'selector' syntax
- Coverage for ore_cllw_u64_8 index type ($.n numeric selector)
- Coverage for ore_cllw_var_8 index type ($.hello string selector)
- Fallback behavior when requested index type missing

Includes new helper function get_ste_vec_selector_term() to extract selector terms using SQL helper functions, matching the pattern from legacy SQL tests.

These tests verify that ORE comparisons work correctly with selector extraction and gracefully fall back to JSONB literal comparison when the expected index type is not present.

Author: tobyhede

tests/sqlx/src/helpers.rs

@@ -82,3 +82,88 @@ pub async fn get_ore_encrypted_as_jsonb(pool: &PgPool, id: i32) -> Result<String
 
     result.with_context(|| format!("ore table returned NULL for id={}", id))
 }
+
+/// Fetch STE vec encrypted value from ste_vec table
+///
+/// The ste_vec table is created by migration `003_install_ste_vec_data.sql`
+/// and contains 10 pre-seeded records (ids 1-10) with ore_cllw_u64_8 and ore_cllw_var_8 indexes.
+///
+/// Test data structure:
+/// - Records have selectors for $.hello (a7cea93975ed8c01f861ccb6bd082784) with ore_cllw_var_8
+/// - Records have selectors for $.n (2517068c0d1f9d4d41d2c666211f785e) with ore_cllw_u64_8
+pub async fn get_ste_vec_encrypted(pool: &PgPool, id: i32) -> Result<String> {
+    let sql = format!("SELECT e::text FROM ste_vec WHERE id = {}", id);
+    let row = sqlx::query(&sql)
+        .fetch_one(pool)
+        .await
+        .with_context(|| format!("fetching ste_vec encrypted value for id={}", id))?;
+
+    let result: Option<String> = row
+        .try_get(0)
+        .with_context(|| format!("extracting text column for id={}", id))?;
+
+    result.with_context(|| format!("ste_vec table returned NULL for id={}", id))
+}
+
+/// Extract selector term using SQL helper functions
+///
+/// Uses the get_numeric_ste_vec_*() helper functions to extract a selector term.
+/// This matches the SQL test pattern exactly.
+///
+/// # Arguments
+/// * `pool` - Database connection pool
+/// * `value` - Which STE vec value to use (10, 20, 30, or 42)
+/// * `selector` - Selector hash to extract (e.g., Selectors::N, Selectors::HELLO)
+///
+/// # Example
+/// ```ignore
+/// // Extract $.n selector from n=30 test data
+/// let term = get_ste_vec_selector_term(&pool, 30, Selectors::N).await?;
+/// ```
+pub async fn get_ste_vec_selector_term(
+    pool: &PgPool,
+    value: i32,
+    selector: &str,
+) -> Result<String> {
+    // Call the appropriate get_numeric_ste_vec_*() function
+    let func_name = match value {
+        10 => "get_numeric_ste_vec_10",
+        20 => "get_numeric_ste_vec_20",
+        30 => "get_numeric_ste_vec_30",
+        42 => "get_numeric_ste_vec_42",
+        _ => {
+            return Err(anyhow::anyhow!(
+                "Invalid value: {}. Must be 10, 20, 30, or 42",
+                value
+            ))
+        }
+    };
+
+    // SQL equivalent: sv := get_numeric_ste_vec_30()::eql_v2_encrypted;
+    //                  term := sv->'2517068c0d1f9d4d41d2c666211f785e'::text;
+    let sql = format!(
+        "SELECT ({}()::eql_v2_encrypted -> '{}'::text)::text",
+        func_name, selector
+    );
+
+    let row = sqlx::query(&sql).fetch_one(pool).await.with_context(|| {
+        format!(
+            "extracting selector '{}' from ste_vec value={}",
+            selector, value
+        )
+    })?;
+
+    let result: Option<String> = row.try_get(0).with_context(|| {
+        format!(
+            "getting text column for selector '{}' from value={}",
+            selector, value
+        )
+    })?;
+
+    result.with_context(|| {
+        format!(
+            "selector extraction returned NULL for selector='{}', value={}",
+            selector, value
+        )
+    })
+}

tests/sqlx/src/lib.rs

@@ -10,7 +10,10 @@ pub mod index_types;
 pub mod selectors;
 
 pub use assertions::QueryAssertion;
-pub use helpers::{get_encrypted_term, get_ore_encrypted, get_ore_encrypted_as_jsonb};
+pub use helpers::{
+    get_encrypted_term, get_ore_encrypted, get_ore_encrypted_as_jsonb, get_ste_vec_encrypted,
+    get_ste_vec_selector_term,
+};
 pub use index_types as IndexTypes;
 pub use selectors::Selectors;