202 lines
6.3 KiB
Rust
Raw Normal View History

2021-02-18 22:10:50 +01:00
use std::time::Duration;
2021-02-18 19:50:52 +01:00
use actix_web::http::StatusCode;
2021-04-20 12:07:22 +02:00
use paste::paste;
2021-02-18 19:50:52 +01:00
use serde_json::{json, Value};
2021-03-10 14:47:04 +01:00
use tokio::time::sleep;
2021-02-18 19:50:52 +01:00
use super::service::Service;
2021-04-20 12:07:22 +02:00
macro_rules! make_settings_test_routes {
($($name:ident),+) => {
$(paste! {
pub async fn [<update_$name>](&self, value: Value) -> (Value, StatusCode) {
let url = format!("/indexes/{}/settings/{}", self.uid, stringify!($name).replace("_", "-"));
self.service.post(url, value).await
}
pub async fn [<get_$name>](&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}/settings/{}", self.uid, stringify!($name).replace("_", "-"));
self.service.get(url).await
}
})*
};
}
2021-02-18 19:50:52 +01:00
pub struct Index<'a> {
pub uid: String,
pub service: &'a Service,
}
2021-04-20 12:07:22 +02:00
#[allow(dead_code)]
2021-02-18 19:50:52 +01:00
impl Index<'_> {
pub async fn get(&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}", self.uid);
self.service.get(url).await
}
2021-02-19 19:55:00 +01:00
pub async fn load_test_set(&self) -> u64 {
let url = format!("/indexes/{}/documents", self.uid);
2021-03-15 18:11:10 +01:00
let (response, code) = self
.service
.post_str(url, include_str!("../assets/test_set.json"))
.await;
assert_eq!(code, 202);
2021-02-19 19:55:00 +01:00
let update_id = response["updateId"].as_i64().unwrap();
self.wait_update_id(update_id as u64).await;
update_id as u64
}
pub async fn create(& self, primary_key: Option<&str>) -> (Value, StatusCode) {
2021-02-18 19:50:52 +01:00
let body = json!({
"uid": self.uid,
"primaryKey": primary_key,
});
self.service.post("/indexes", body).await
}
2021-02-18 20:28:10 +01:00
pub async fn update(&self, primary_key: Option<&str>) -> (Value, StatusCode) {
let body = json!({
"primaryKey": primary_key,
});
let url = format!("/indexes/{}", self.uid);
self.service.put(url, body).await
}
2021-02-18 20:44:33 +01:00
pub async fn delete(&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}", self.uid);
self.service.delete(url).await
}
2021-02-18 22:10:50 +01:00
pub async fn add_documents(
&self,
documents: Value,
primary_key: Option<&str>,
) -> (Value, StatusCode) {
let url = match primary_key {
Some(key) => format!("/indexes/{}/documents?primaryKey={}", self.uid, key),
None => format!("/indexes/{}/documents", self.uid),
};
self.service.post(url, documents).await
}
2021-03-15 18:11:10 +01:00
pub async fn update_documents(
&self,
documents: Value,
primary_key: Option<&str>,
) -> (Value, StatusCode) {
2021-02-19 19:55:00 +01:00
let url = match primary_key {
Some(key) => format!("/indexes/{}/documents?primaryKey={}", self.uid, key),
None => format!("/indexes/{}/documents", self.uid),
};
self.service.put(url, documents).await
}
2021-02-22 16:03:17 +01:00
pub async fn wait_update_id(&self, update_id: u64) -> Value {
2021-02-18 22:10:50 +01:00
// try 10 times to get status, or panic to not wait forever
let url = format!("/indexes/{}/updates/{}", self.uid, update_id);
for _ in 0..10 {
let (response, status_code) = self.service.get(&url).await;
2021-03-10 18:04:20 +01:00
assert_eq!(status_code, 200, "response: {}", response);
2021-02-18 22:10:50 +01:00
if response["status"] == "processed" || response["status"] == "failed" {
2021-02-22 16:03:17 +01:00
return response;
2021-02-18 22:10:50 +01:00
}
2021-03-10 14:47:04 +01:00
sleep(Duration::from_secs(1)).await;
2021-02-18 22:10:50 +01:00
}
panic!("Timeout waiting for update id");
}
2021-03-10 18:04:20 +01:00
pub async fn get_update(&self, update_id: u64) -> (Value, StatusCode) {
let url = format!("/indexes/{}/updates/{}", self.uid, update_id);
2021-02-18 22:10:50 +01:00
self.service.get(url).await
}
pub async fn list_updates(&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}/updates", self.uid);
self.service.get(url).await
}
2021-02-19 19:55:00 +01:00
2021-03-15 18:11:10 +01:00
pub async fn get_document(
&self,
id: u64,
_options: Option<GetDocumentOptions>,
) -> (Value, StatusCode) {
2021-02-19 19:55:00 +01:00
let url = format!("/indexes/{}/documents/{}", self.uid, id);
self.service.get(url).await
}
pub async fn get_all_documents(&self, options: GetAllDocumentsOptions) -> (Value, StatusCode) {
let mut url = format!("/indexes/{}/documents?", self.uid);
if let Some(limit) = options.limit {
url.push_str(&format!("limit={}&", limit));
}
if let Some(offset) = options.offset {
url.push_str(&format!("offset={}&", offset));
}
if let Some(attributes_to_retrieve) = options.attributes_to_retrieve {
2021-03-15 18:11:10 +01:00
url.push_str(&format!(
"attributesToRetrieve={}&",
attributes_to_retrieve.join(",")
));
}
2021-02-19 19:55:00 +01:00
self.service.get(url).await
}
2021-02-22 16:03:17 +01:00
pub async fn delete_document(&self, id: u64) -> (Value, StatusCode) {
let url = format!("/indexes/{}/documents/{}", self.uid, id);
self.service.delete(url).await
}
pub async fn clear_all_documents(&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}/documents", self.uid);
self.service.delete(url).await
}
2021-02-23 14:13:43 +01:00
pub async fn delete_batch(&self, ids: Vec<u64>) -> (Value, StatusCode) {
let url = format!("/indexes/{}/documents/delete-batch", self.uid);
2021-03-15 18:11:10 +01:00
self.service
.post(url, serde_json::to_value(&ids).unwrap())
.await
2021-02-23 14:13:43 +01:00
}
2021-02-24 09:30:51 +01:00
pub async fn settings(&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}/settings", self.uid);
self.service.get(url).await
}
pub async fn update_settings(&self, settings: Value) -> (Value, StatusCode) {
let url = format!("/indexes/{}/settings", self.uid);
self.service.post(url, settings).await
}
2021-02-24 10:14:36 +01:00
pub async fn delete_settings(&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}/settings", self.uid);
self.service.delete(url).await
}
2021-04-01 21:54:37 +03:00
pub async fn stats(&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}/stats", self.uid);
self.service.get(url).await
}
2021-04-20 12:07:22 +02:00
make_settings_test_routes!(
distinct_attribute
);
2021-02-18 19:50:52 +01:00
}
2021-02-19 19:55:00 +01:00
2021-04-20 12:07:22 +02:00
2021-02-19 19:55:00 +01:00
pub struct GetDocumentOptions;
#[derive(Debug, Default)]
pub struct GetAllDocumentsOptions {
pub limit: Option<usize>,
pub offset: Option<usize>,
pub attributes_to_retrieve: Option<Vec<&'static str>>,
}