146 lines
3.8 KiB
Rust
146 lines
3.8 KiB
Rust
use serde_json::Value;
|
|
use std::error::Error;
|
|
use std::time::Duration;
|
|
|
|
use assert_json_diff::assert_json_eq;
|
|
use async_std::io::prelude::*;
|
|
use async_std::task::{block_on, sleep};
|
|
use http_service::Body;
|
|
use http_service_mock::{make_server, TestBackend};
|
|
use meilisearch_http::data::Data;
|
|
use meilisearch_http::option::Opt;
|
|
use meilisearch_http::routes;
|
|
use serde_json::json;
|
|
use tempdir::TempDir;
|
|
use tide::server::Service;
|
|
|
|
pub fn setup_server() -> Result<TestBackend<Service<Data>>, Box<dyn Error>> {
|
|
let tmp_dir = TempDir::new("meilisearch")?;
|
|
|
|
let opt = Opt {
|
|
db_path: tmp_dir.path().to_str().unwrap().to_string(),
|
|
http_addr: "127.0.0.1:7700".to_owned(),
|
|
api_key: None,
|
|
no_analytics: true,
|
|
};
|
|
|
|
let data = Data::new(opt.clone());
|
|
let mut app = tide::with_state(data);
|
|
routes::load_routes(&mut app);
|
|
let http_server = app.into_http_service();
|
|
Ok(make_server(http_server)?)
|
|
}
|
|
|
|
pub fn enrich_server_with_movies_index(
|
|
server: &mut TestBackend<Service<Data>>,
|
|
) -> Result<(), Box<dyn Error>> {
|
|
let body = json!({
|
|
"uid": "movies",
|
|
})
|
|
.to_string()
|
|
.into_bytes();
|
|
|
|
let req = http::Request::post("/indexes")
|
|
.body(Body::from(body))
|
|
.unwrap();
|
|
let _res = server.simulate(req).unwrap();
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn enrich_server_with_movies_settings(
|
|
server: &mut TestBackend<Service<Data>>,
|
|
) -> Result<(), Box<dyn Error>> {
|
|
let json = json!({
|
|
"rankingRules": [
|
|
"_typo",
|
|
"_words",
|
|
"_proximity",
|
|
"_attribute",
|
|
"_words_position",
|
|
"dsc(popularity)",
|
|
"_exact",
|
|
"dsc(vote_average)",
|
|
],
|
|
"rankingDistinct": null,
|
|
"identifier": "id",
|
|
"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",
|
|
],
|
|
"stopWords": null,
|
|
"synonyms": null,
|
|
"indexNewFields": false,
|
|
});
|
|
|
|
let body = json.to_string().into_bytes();
|
|
|
|
let req = http::Request::post("/indexes/movies/settings")
|
|
.body(Body::from(body))
|
|
.unwrap();
|
|
let _res = server.simulate(req).unwrap();
|
|
|
|
block_on(sleep(Duration::from_secs(5)));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn enrich_server_with_movies_documents(
|
|
server: &mut TestBackend<Service<Data>>,
|
|
) -> Result<(), Box<dyn Error>> {
|
|
let body = include_bytes!("assets/movies.json").to_vec();
|
|
|
|
let req = http::Request::post("/indexes/movies/documents")
|
|
.body(Body::from(body))
|
|
.unwrap();
|
|
let _res = server.simulate(req).unwrap();
|
|
|
|
block_on(sleep(Duration::from_secs(5)));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn search(server: &mut TestBackend<Service<Data>>, query: &str, expect: Value) {
|
|
let req = http::Request::get(format!("/indexes/movies/search?{}", query))
|
|
.body(Body::empty())
|
|
.unwrap();
|
|
let res = server.simulate(req).unwrap();
|
|
|
|
let mut buf = Vec::new();
|
|
block_on(res.into_body().read_to_end(&mut buf)).unwrap();
|
|
let response: Value = serde_json::from_slice(&buf).unwrap();
|
|
|
|
assert_json_eq!(expect, response["hits"].clone(), ordered: false)
|
|
}
|
|
|
|
pub fn update_config(server: &mut TestBackend<Service<Data>>, config: Value) {
|
|
let body = config.to_string().into_bytes();
|
|
|
|
let req = http::Request::post("/indexes/movies/settings")
|
|
.body(Body::from(body))
|
|
.unwrap();
|
|
let res = server.simulate(req).unwrap();
|
|
assert_eq!(res.status(), 202);
|
|
|
|
block_on(sleep(Duration::from_secs(5)));
|
|
}
|