141 lines
4.2 KiB
Rust
141 lines
4.2 KiB
Rust
#[cfg(test)]
|
|
mod tests {
|
|
use std::sync::Mutex;
|
|
|
|
use actix_web::{test, web, App};
|
|
use rustysearch::{
|
|
handlers::search,
|
|
search::engine::{remove_index_from_disk, SearchEngine},
|
|
types::app_state::AppStateWithSearchEngine,
|
|
};
|
|
|
|
#[actix_web::test]
|
|
async fn test_add_document_to_index() {
|
|
let search_engine = SearchEngine::new(1.5, 0.75);
|
|
|
|
let app_state = web::Data::new(AppStateWithSearchEngine {
|
|
search_engine: Mutex::new(search_engine.clone()),
|
|
});
|
|
|
|
let app = test::init_service(App::new().app_data(app_state.clone()).route(
|
|
"/search/index/document",
|
|
web::post().to(search::add_document_to_index),
|
|
))
|
|
.await;
|
|
|
|
let data = search::AddDocumentRequest {
|
|
url: "https://example.com".to_string(),
|
|
content: "This is an example document".to_string(),
|
|
};
|
|
|
|
let req = test::TestRequest::post()
|
|
.uri("/search/index/document")
|
|
.set_json(data)
|
|
.to_request();
|
|
|
|
let resp = test::call_service(&app, req).await;
|
|
assert!(resp.status() == 201);
|
|
|
|
remove_index_from_disk();
|
|
}
|
|
|
|
#[actix_web::test]
|
|
async fn test_get_number_of_documents() {
|
|
let mut search_engine = SearchEngine::new(1.5, 0.75);
|
|
search_engine.index("https://example.com", "This is an example document");
|
|
|
|
let app_state = web::Data::new(AppStateWithSearchEngine {
|
|
search_engine: Mutex::new(search_engine.clone()),
|
|
});
|
|
|
|
let app = test::init_service(App::new().app_data(app_state.clone()).route(
|
|
"/search/index/number_of_documents",
|
|
web::get().to(search::get_number_of_documents),
|
|
))
|
|
.await;
|
|
|
|
let req = test::TestRequest::get()
|
|
.uri("/search/index/number_of_documents")
|
|
.to_request();
|
|
|
|
let resp = test::call_service(&app, req).await;
|
|
assert!(resp.status() == 200);
|
|
|
|
remove_index_from_disk();
|
|
}
|
|
|
|
#[actix_web::test]
|
|
async fn test_search() {
|
|
let mut search_engine = SearchEngine::new(1.5, 0.75);
|
|
search_engine.index("https://example.com", "This is an example document");
|
|
|
|
let app_state = web::Data::new(AppStateWithSearchEngine {
|
|
search_engine: Mutex::new(search_engine.clone()),
|
|
});
|
|
|
|
let app = test::init_service(
|
|
App::new()
|
|
.app_data(app_state.clone())
|
|
.route("/search", web::get().to(search::search)),
|
|
)
|
|
.await;
|
|
|
|
let req = test::TestRequest::get()
|
|
.uri("/search?query=example")
|
|
.to_request();
|
|
|
|
let resp = test::call_service(&app, req).await;
|
|
assert!(resp.status() == 200);
|
|
|
|
remove_index_from_disk();
|
|
}
|
|
|
|
#[actix_web::test]
|
|
async fn test_search_without_query() {
|
|
let mut search_engine = SearchEngine::new(1.5, 0.75);
|
|
search_engine.index("https://example.com", "This is an example document");
|
|
|
|
let app_state = web::Data::new(AppStateWithSearchEngine {
|
|
search_engine: Mutex::new(search_engine.clone()),
|
|
});
|
|
|
|
let app = test::init_service(
|
|
App::new()
|
|
.app_data(app_state.clone())
|
|
.route("/search", web::get().to(search::search)),
|
|
)
|
|
.await;
|
|
|
|
let req = test::TestRequest::get().uri("/search?query=").to_request();
|
|
|
|
let resp = test::call_service(&app, req).await;
|
|
assert!(resp.status() == 400);
|
|
|
|
remove_index_from_disk();
|
|
}
|
|
|
|
#[actix_web::test]
|
|
async fn test_debug_index() {
|
|
let search_engine = SearchEngine::new(1.5, 0.75);
|
|
|
|
let app_state = web::Data::new(AppStateWithSearchEngine {
|
|
search_engine: Mutex::new(search_engine.clone()),
|
|
});
|
|
|
|
let app = test::init_service(
|
|
App::new()
|
|
.app_data(app_state.clone())
|
|
.route("/search/debug", web::get().to(search::debug_index)),
|
|
)
|
|
.await;
|
|
|
|
let req = test::TestRequest::get().uri("/search/debug").to_request();
|
|
|
|
let resp = test::call_service(&app, req).await;
|
|
assert!(resp.status() == 200);
|
|
|
|
remove_index_from_disk();
|
|
}
|
|
}
|
|
|