#![allow(dead_code)] use serde_json::{json, Value}; use std::time::Duration; use actix_web::{http::StatusCode, test}; use meilisearch_http::data::Data; use meilisearch_http::option::Opt; use meilisearch_core::DatabaseOptions; use tempdir::TempDir; use tokio::time::delay_for; pub struct Server { uid: String, data: Data, } const DB_OPTS: DatabaseOptions = DatabaseOptions { main_map_size: 100 * 1024 * 1024 * 1024, update_map_size: 100 * 1024 * 1024 * 1024, }; impl Server { pub fn with_uid(uid: &str) -> Server { let tmp_dir = TempDir::new("meilisearch").unwrap(); let default_db_options = DB_OPTS; let opt = Opt { db_path: tmp_dir.path().to_str().unwrap().to_string(), http_addr: "127.0.0.1:7700".to_owned(), master_key: None, env: "development".to_owned(), no_analytics: true, main_map_size: default_db_options.main_map_size, update_map_size: default_db_options.update_map_size }; let data = Data::new(opt.clone()); Server { uid: uid.to_string(), data, } } pub async fn wait_update_id(&mut self, update_id: u64) { loop { let (response, status_code) = self.get_update_status(update_id).await; assert_eq!(status_code, 200); if response["status"] == "processed" || response["status"] == "error" { eprintln!("{:#?}", response); return; } delay_for(Duration::from_secs(1)).await; } } // Global Http request GET/POST/DELETE async or sync pub async fn get_request(&mut self, url: &str) -> (Value, StatusCode) { eprintln!("get_request: {}", url); let mut app = test::init_service(meilisearch_http::create_app(&self.data)).await; let req = test::TestRequest::get().uri(url).to_request(); let res = test::call_service(&mut app, req).await; let status_code = res.status().clone(); let body = test::read_body(res).await; let response = serde_json::from_slice(&body).unwrap_or_default(); (response, status_code) } pub async fn post_request(&mut self, url: &str, body: Value) -> (Value, StatusCode) { eprintln!("post_request: {}", url); let mut app = test::init_service(meilisearch_http::create_app(&self.data)).await; let req = test::TestRequest::post() .uri(url) .set_json(&body) .to_request(); let res = test::call_service(&mut app, req).await; let status_code = res.status().clone(); let body = test::read_body(res).await; let response = serde_json::from_slice(&body).unwrap_or_default(); (response, status_code) } pub async fn post_request_async(&mut self, url: &str, body: Value) -> (Value, StatusCode) { eprintln!("post_request_async: {}", url); let (response, status_code) = self.post_request(url, body).await; assert_eq!(status_code, 202); assert!(response["updateId"].as_u64().is_some()); self.wait_update_id(response["updateId"].as_u64().unwrap()) .await; (response, status_code) } pub async fn put_request(&mut self, url: &str, body: Value) -> (Value, StatusCode) { eprintln!("put_request: {}", url); let mut app = test::init_service(meilisearch_http::create_app(&self.data)).await; let req = test::TestRequest::put() .uri(url) .set_json(&body) .to_request(); let res = test::call_service(&mut app, req).await; let status_code = res.status().clone(); let body = test::read_body(res).await; let response = serde_json::from_slice(&body).unwrap_or_default(); (response, status_code) } pub async fn put_request_async(&mut self, url: &str, body: Value) -> (Value, StatusCode) { eprintln!("put_request_async: {}", url); let (response, status_code) = self.put_request(url, body).await; assert!(response["updateId"].as_u64().is_some()); assert_eq!(status_code, 202); self.wait_update_id(response["updateId"].as_u64().unwrap()) .await; (response, status_code) } pub async fn delete_request(&mut self, url: &str) -> (Value, StatusCode) { eprintln!("delete_request: {}", url); let mut app = test::init_service(meilisearch_http::create_app(&self.data)).await; let req = test::TestRequest::delete().uri(url).to_request(); let res = test::call_service(&mut app, req).await; let status_code = res.status().clone(); let body = test::read_body(res).await; let response = serde_json::from_slice(&body).unwrap_or_default(); (response, status_code) } pub async fn delete_request_async(&mut self, url: &str) -> (Value, StatusCode) { eprintln!("delete_request_async: {}", url); let (response, status_code) = self.delete_request(url).await; assert!(response["updateId"].as_u64().is_some()); assert_eq!(status_code, 202); self.wait_update_id(response["updateId"].as_u64().unwrap()) .await; (response, status_code) } // All Routes pub async fn list_indexes(&mut self) -> (Value, StatusCode) { self.get_request("/indexes").await } pub async fn create_index(&mut self, body: Value) -> (Value, StatusCode) { self.post_request("/indexes", body).await } pub async fn search_multi_index(&mut self, query: &str) -> (Value, StatusCode) { let url = format!("/indexes/search?{}", query); self.get_request(&url).await } pub async fn get_index(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}", self.uid); self.get_request(&url).await } pub async fn update_index(&mut self, body: Value) -> (Value, StatusCode) { let url = format!("/indexes/{}", self.uid); self.put_request(&url, body).await } pub async fn delete_index(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}", self.uid); self.delete_request(&url).await } pub async fn search(&mut self, query: &str) -> (Value, StatusCode) { let url = format!("/indexes/{}/search?{}", self.uid, query); self.get_request(&url).await } pub async fn get_all_updates_status(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/updates", self.uid); self.get_request(&url).await } pub async fn get_update_status(&mut self, update_id: u64) -> (Value, StatusCode) { let url = format!("/indexes/{}/updates/{}", self.uid, update_id); self.get_request(&url).await } pub async fn get_all_documents(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/documents", self.uid); self.get_request(&url).await } pub async fn add_or_replace_multiple_documents(&mut self, body: Value) { let url = format!("/indexes/{}/documents", self.uid); self.post_request_async(&url, body).await; } pub async fn add_or_replace_multiple_documents_sync( &mut self, body: Value, ) -> (Value, StatusCode) { let url = format!("/indexes/{}/documents", self.uid); self.post_request(&url, body).await } pub async fn add_or_update_multiple_documents(&mut self, body: Value) { let url = format!("/indexes/{}/documents", self.uid); self.put_request_async(&url, body).await; } pub async fn clear_all_documents(&mut self) { let url = format!("/indexes/{}/documents", self.uid); self.delete_request_async(&url).await; } pub async fn get_document(&mut self, document_id: impl ToString) -> (Value, StatusCode) { let url = format!( "/indexes/{}/documents/{}", self.uid, document_id.to_string() ); self.get_request(&url).await } pub async fn delete_document(&mut self, document_id: impl ToString) -> (Value, StatusCode) { let url = format!( "/indexes/{}/documents/{}", self.uid, document_id.to_string() ); self.delete_request_async(&url).await } pub async fn delete_multiple_documents(&mut self, body: Value) { let url = format!("/indexes/{}/documents/delete-batch", self.uid); self.post_request_async(&url, body).await; } pub async fn get_all_settings(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings", self.uid); self.get_request(&url).await } pub async fn update_all_settings(&mut self, body: Value) { let url = format!("/indexes/{}/settings", self.uid); self.post_request_async(&url, body).await; } pub async fn delete_all_settings(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings", self.uid); self.delete_request_async(&url).await } pub async fn get_ranking_rules(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/ranking-rules", self.uid); self.get_request(&url).await } pub async fn update_ranking_rules(&mut self, body: Value) { let url = format!("/indexes/{}/settings/ranking-rules", self.uid); self.post_request_async(&url, body).await; } pub async fn update_ranking_rules_sync(&mut self, body: Value) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/ranking-rules", self.uid); self.post_request(&url, body).await } pub async fn delete_ranking_rules(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/ranking-rules", self.uid); self.delete_request_async(&url).await } pub async fn get_distinct_attribute(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/distinct-attribute", self.uid); self.get_request(&url).await } pub async fn update_distinct_attribute(&mut self, body: Value) { let url = format!("/indexes/{}/settings/distinct-attribute", self.uid); self.post_request_async(&url, body).await; } pub async fn delete_distinct_attribute(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/distinct-attribute", self.uid); self.delete_request_async(&url).await } pub async fn get_primary_key(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/primary_key", self.uid); self.get_request(&url).await } pub async fn get_searchable_attributes(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/searchable-attributes", self.uid); self.get_request(&url).await } pub async fn update_searchable_attributes(&mut self, body: Value) { let url = format!("/indexes/{}/settings/searchable-attributes", self.uid); self.post_request_async(&url, body).await; } pub async fn delete_searchable_attributes(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/searchable-attributes", self.uid); self.delete_request_async(&url).await } pub async fn get_displayed_attributes(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/displayed-attributes", self.uid); self.get_request(&url).await } pub async fn update_displayed_attributes(&mut self, body: Value) { let url = format!("/indexes/{}/settings/displayed-attributes", self.uid); self.post_request_async(&url, body).await; } pub async fn delete_displayed_attributes(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/displayed-attributes", self.uid); self.delete_request_async(&url).await } pub async fn get_accept_new_fields(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/accept-new-fields", self.uid); self.get_request(&url).await } pub async fn update_accept_new_fields(&mut self, body: Value) { let url = format!("/indexes/{}/settings/accept-new-fields", self.uid); self.post_request_async(&url, body).await; } pub async fn get_synonyms(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/synonyms", self.uid); self.get_request(&url).await } pub async fn update_synonyms(&mut self, body: Value) { let url = format!("/indexes/{}/settings/synonyms", self.uid); self.post_request_async(&url, body).await; } pub async fn delete_synonyms(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/synonyms", self.uid); self.delete_request_async(&url).await } pub async fn get_stop_words(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/stop-words", self.uid); self.get_request(&url).await } pub async fn update_stop_words(&mut self, body: Value) { let url = format!("/indexes/{}/settings/stop-words", self.uid); self.post_request_async(&url, body).await; } pub async fn delete_stop_words(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/settings/stop-words", self.uid); self.delete_request_async(&url).await } pub async fn get_index_stats(&mut self) -> (Value, StatusCode) { let url = format!("/indexes/{}/stats", self.uid); self.get_request(&url).await } pub async fn list_keys(&mut self) -> (Value, StatusCode) { self.get_request("/keys").await } pub async fn get_health(&mut self) -> (Value, StatusCode) { self.get_request("/health").await } pub async fn update_health(&mut self, body: Value) -> (Value, StatusCode) { self.put_request("/health", body).await } pub async fn get_version(&mut self) -> (Value, StatusCode) { self.get_request("/version").await } pub async fn get_sys_info(&mut self) -> (Value, StatusCode) { self.get_request("/sys-info").await } pub async fn get_sys_info_pretty(&mut self) -> (Value, StatusCode) { self.get_request("/sys-info/pretty").await } // Populate routes pub async fn populate_movies(&mut self) { let body = json!({ "uid": "movies", "primaryKey": "id", }); self.create_index(body).await; let body = json!({ "rankingRules": [ "typo", "words", "proximity", "attribute", "wordsPosition", "desc(popularity)", "exactness", "desc(vote_average)", ], "searchableAttributes": [ "title", "tagline", "overview", "cast", "director", "producer", "production_companies", "genres", ], "displayedAttributes": [ "title", "director", "producer", "tagline", "genres", "id", "overview", "vote_count", "vote_average", "poster_path", "popularity", ], "acceptNewFields": false, }); self.update_all_settings(body).await; let dataset = include_bytes!("assets/movies.json"); let body: Value = serde_json::from_slice(dataset).unwrap(); self.add_or_replace_multiple_documents(body).await; } }