Add vector similarity functions integration tests

crates/validator/src/common.rs

@@ -151,13 +151,20 @@ pub(crate) async fn create_index(
     client: &HttpClient,
     table: &str,
     column: &str,
+    options: Option<String>,
 ) -> IndexInfo {
     let index = format!("idx_{}", Uuid::new_v4().simple());
 
+    let extra = if let Some(options) = options {
+        format!("WITH OPTIONS = {options}")
+    } else {
+        String::new()
+    };
+
     // Create index
     session
         .query_unpaged(
-            format!("CREATE INDEX {index} ON {table}({column}) USING 'vector_index'"),
+            format!("CREATE INDEX {index} ON {table}({column}) USING 'vector_index' {extra}"),
             (),
         )
         .await

crates/validator/src/tests/ann.rs

@@ -61,7 +61,7 @@ async fn ann_query_returns_expected_results(actors: TestActors) {
             .expect("failed to insert data");
     }
 
-    let index = create_index(&session, &client, &table, "v").await;
+    let index = create_index(&session, &client, &table, "v", None).await;
 
     wait_for(
         || async { client.count(&index.keyspace, &index.index).await == Some(1000) },
@@ -121,7 +121,7 @@ async fn ann_query_respects_limit(actors: TestActors) {
     }
 
     // Create index
-    let index = create_index(&session, &client, &table, "v").await;
+    let index = create_index(&session, &client, &table, "v", None).await;
 
     wait_for(
         || async { client.count(&index.keyspace, &index.index).await == Some(10) },
@@ -189,7 +189,7 @@ async fn ann_query_respects_limit_over_1000_vectors(actors: TestActors) {
             .expect("failed to insert data");
     }
 
-    let index = create_index(&session, &client, &table, "v").await;
+    let index = create_index(&session, &client, &table, "v", None).await;
 
     wait_for(
         || async { client.count(&index.keyspace, &index.index).await == Some(1111) },

crates/validator/src/tests/crud.rs

@@ -38,7 +38,7 @@ async fn simple_create_drop_index(actors: TestActors) {
     )
     .await;
 
-    let index = create_index(&session, &client, &table, "embedding").await;
+    let index = create_index(&session, &client, &table, "embedding", None).await;
 
     assert_eq!(index.keyspace.as_ref(), &keyspace);
 
@@ -71,7 +71,7 @@ async fn simple_create_drop_multiple_indexes(actors: TestActors) {
     .await;
 
     // Create index on column v1
-    let index1 = create_index(&session, &client, &table, "v1").await;
+    let index1 = create_index(&session, &client, &table, "v1", None).await;
 
     // Wait for the full scan to complete and check if ANN query succeeds on v1
     wait_for(
@@ -99,7 +99,7 @@ async fn simple_create_drop_multiple_indexes(actors: TestActors) {
         .expect_err("ANN query should fail when index does not exist");
 
     // Create index on column v2
-    let index2 = create_index(&session, &client, &table, "v2").await;
+    let index2 = create_index(&session, &client, &table, "v2", None).await;
 
     // Check if ANN query on v1 still succeeds
     session

crates/validator/src/tests/full_scan.rs

@@ -44,7 +44,7 @@ async fn full_scan_is_completed_when_responding_to_messages_concurrently(actors:
             .expect("failed to insert data");
     }
 
-    let index = create_index(&session, &client, &table, "embedding").await;
+    let index = create_index(&session, &client, &table, "embedding", None).await;
 
     let result = session
         .query_unpaged(

crates/validator/src/tests/mod.rs

@@ -8,6 +8,7 @@ mod crud;
 mod full_scan;
 mod reconnect;
 mod serde;
+mod vector_similarity;
 
 use crate::ServicesSubnet;
 use crate::dns::Dns;
@@ -221,6 +222,7 @@ pub(crate) async fn register() -> Vec<(String, TestCase)> {
         ("full_scan", full_scan::new().await),
         ("reconnect", reconnect::new().await),
         ("serde", serde::new().await),
+        ("vector_similarity", vector_similarity::new().await),
     ]
     .into_iter()
     .map(|(name, test_case)| (name.to_string(), test_case))

crates/validator/src/tests/reconnect.rs

@@ -49,7 +49,7 @@ async fn reconnect_doesnt_break_fullscan(actors: TestActors) {
             .expect("failed to insert a row");
     }
 
-    let index = create_index(&session, &client, &table, "embedding").await;
+    let index = create_index(&session, &client, &table, "embedding", None).await;
 
     let result = session
         .query_unpaged(