e2e test for vector index

Author: shisoft

src/apps/hnsw/mod.rs

@@ -148,9 +148,9 @@ unsafe impl Sync for VectorIndexer {}
 #[cfg(test)]
 mod e2e_tests {
     use dovahkiin::{data_map, types::{OwnedPrimArray, Type}};
-    use neb::{index::vector::MetricEncoding, ram::{cell::OwnedCell, schema::{Field, IndexType, Schema}, types::RandValue}};
+    use neb::{index::vector::{self, MetricEncoding}, ram::{cell::OwnedCell, schema::{Field, IndexType, Schema}, types::RandValue}};
 
-    use crate::tests::start_server;
+    use crate::{apps::hnsw::coordinator::Service, tests::start_server};
 
     use super::*;
 
@@ -162,9 +162,10 @@ mod e2e_tests {
     async fn test_e2e() {
         let server = start_server(5010, "vector_indexer_e2e_test").await.unwrap();
         let _partition = server.init_hnsw_index_partition_service().await.unwrap();
-        let _coordinator = server.init_hnsw_index_service().await.unwrap();
+        let coordinator = server.init_hnsw_index_service().await.unwrap();
         let indexer = VectorIndexer::new(&server).await;
         indexer.set_core();
+        assert!(vector::is_vector_index_core_set());
 
         let neb_client = server.neb_client.clone();
         let cell_schema = Schema::new_with_id(
@@ -183,8 +184,21 @@ mod e2e_tests {
         neb_client.new_schema_with_id(cell_schema).await.unwrap().unwrap();
         let cell_id = Id::rand();
         let vector1 = vec![1.0, 2.0, 3.0];
-        let map = data_map!(VECTOR: OwnedValue::PrimArray(OwnedPrimArray::F32(vector1)));
+        let map = data_map!(VECTOR: OwnedValue::PrimArray(OwnedPrimArray::F32(vector1.clone())));
         let cell1 = OwnedCell::new_with_id(CELL_SCHEMA_ID, &cell_id, OwnedValue::Map(map));
+        assert!(vector::is_vector_index_core_set());
         neb_client.write_cell(cell1).await.unwrap().unwrap();
+        let query = coordinator.query_top_k(
+            CELL_SCHEMA_ID, 
+            VECTOR_FIELD_ID, 
+            OwnedPrimArray::F32(vector1), 
+            10, 
+            32,
+            10, 
+            MetricEncoding::L2
+        ).await.unwrap();
+        assert_eq!(query.len(), 1);
+        assert_eq!(query[0].0, cell_id);
+        assert_eq!(query[0].1, 0.0);
     }
 }