You've already forked AstralRinth
aecsocket
d2a66bb2b0
* Allow searching by project dependencies * change field name * use query macro
225 lines
8.4 KiB
Rust
225 lines
8.4 KiB
Rust
use actix_http::StatusCode;
|
|
use common::api_v3::ApiV3;
|
|
use common::database::*;
|
|
|
|
use common::dummy_data::DUMMY_CATEGORIES;
|
|
|
|
use ariadne::ids::base62_impl::{parse_base62, to_base62};
|
|
use common::environment::TestEnvironment;
|
|
use common::environment::with_test_environment;
|
|
use common::search::setup_search_projects;
|
|
use futures::stream::StreamExt;
|
|
use serde_json::json;
|
|
|
|
use crate::common::api_common::Api;
|
|
use crate::common::api_common::ApiProject;
|
|
|
|
pub mod common;
|
|
|
|
// TODO: Revisit this wit h the new modify_json in the version maker
|
|
// That change here should be able to simplify it vastly
|
|
|
|
#[actix_rt::test]
|
|
async fn search_projects() {
|
|
// Test setup and dummy data
|
|
with_test_environment(
|
|
Some(10),
|
|
|test_env: TestEnvironment<ApiV3>| async move {
|
|
let id_conversion = setup_search_projects(&test_env).await;
|
|
|
|
let api = &test_env.api;
|
|
let test_name = test_env.db.database_name.clone();
|
|
let dependency_project_id = id_conversion
|
|
.iter()
|
|
.find_map(|(project_id, test_id)| {
|
|
(*test_id == 1).then_some(to_base62(*project_id))
|
|
})
|
|
.unwrap();
|
|
|
|
// Pairs of:
|
|
// 1. vec of search facets
|
|
// 2. expected project ids to be returned by this search
|
|
let pairs = vec![
|
|
(
|
|
json!([["categories:fabric"]]),
|
|
vec![0, 1, 2, 3, 4, 5, 6, 7, 9],
|
|
),
|
|
(json!([["categories:forge"]]), vec![7]),
|
|
(
|
|
json!([["categories:fabric", "categories:forge"]]),
|
|
vec![0, 1, 2, 3, 4, 5, 6, 7, 9],
|
|
),
|
|
(json!([["categories:fabric"], ["categories:forge"]]), vec![]),
|
|
(
|
|
json!([
|
|
["categories:fabric"],
|
|
[&format!("categories:{}", DUMMY_CATEGORIES[0])],
|
|
]),
|
|
vec![1, 2, 3, 4],
|
|
),
|
|
(json!([["project_types:modpack"]]), vec![4]),
|
|
(json!([["environment:server_only"]]), vec![0, 2, 3]),
|
|
(
|
|
json!([["environment:client_or_server_prefers_both"]]),
|
|
vec![6, 7],
|
|
),
|
|
(json!([["open_source:true"]]), vec![0, 1, 2, 4, 5, 6, 7, 9]),
|
|
(json!([["license:MIT"]]), vec![1, 2, 4, 9]),
|
|
(json!([[r#"name:'Mysterious Project'"#]]), vec![2, 3]),
|
|
(json!([["author:user"]]), vec![0, 1, 2, 4, 5, 7, 9]), // Organization test '9' is included here as user is owner of org
|
|
(json!([["game_versions:1.20.5"]]), vec![4, 5]),
|
|
// bug fix
|
|
(
|
|
json!([
|
|
// Only the forge one has 1.20.2, so its true that this project 'has'
|
|
// 1.20.2 and a fabric version, but not true that it has a 1.20.2 fabric version.
|
|
["categories:fabric"],
|
|
["game_versions:1.20.2"]
|
|
]),
|
|
vec![],
|
|
),
|
|
// Project type change
|
|
// Modpack should still be able to search based on former loader, even though technically the loader is 'mrpack'
|
|
// (json!([["categories:mrpack"]]), vec![4]),
|
|
// (
|
|
// json!([["categories:fabric"]]),
|
|
// vec![4],
|
|
// ),
|
|
(
|
|
json!([["categories:fabric"], ["project_types:modpack"]]),
|
|
vec![4],
|
|
),
|
|
(
|
|
json!([[format!(
|
|
"dependency_project_ids:{dependency_project_id}"
|
|
)]]),
|
|
vec![7],
|
|
),
|
|
];
|
|
// TODO: versions, game versions
|
|
// Untested:
|
|
// - downloads (not varied)
|
|
// - color (not varied)
|
|
// - created_timestamp (not varied)
|
|
// - modified_timestamp (not varied)
|
|
// TODO: multiple different project types test
|
|
|
|
// Test searches
|
|
let stream = futures::stream::iter(pairs);
|
|
stream
|
|
.for_each_concurrent(1, |(facets, mut expected_project_ids)| {
|
|
let id_conversion = id_conversion.clone();
|
|
let test_name = test_name.clone();
|
|
async move {
|
|
let projects = api
|
|
.search_deserialized(
|
|
Some(&format!("&{test_name}")),
|
|
Some(facets.clone()),
|
|
USER_USER_PAT,
|
|
)
|
|
.await;
|
|
let mut found_project_ids: Vec<u64> = projects
|
|
.hits
|
|
.into_iter()
|
|
.map(|p| {
|
|
id_conversion
|
|
[&parse_base62(&p.project_id).unwrap()]
|
|
})
|
|
.collect();
|
|
let num_hits = projects.total_hits;
|
|
expected_project_ids.sort();
|
|
found_project_ids.sort();
|
|
println!("Facets: {facets:?}");
|
|
assert_eq!(found_project_ids, expected_project_ids);
|
|
assert_eq!(num_hits, { expected_project_ids.len() });
|
|
}
|
|
})
|
|
.await;
|
|
|
|
let projects = api
|
|
.search_deserialized(
|
|
Some(&format!("&{test_name}")),
|
|
Some(json!([[format!(
|
|
"dependency_project_ids:{dependency_project_id}"
|
|
)]])),
|
|
USER_USER_PAT,
|
|
)
|
|
.await;
|
|
assert_eq!(projects.total_hits, 1);
|
|
assert_eq!(projects.hits[0].dependency_project_ids.len(), 1);
|
|
assert_eq!(
|
|
projects.hits[0].dependency_project_ids[0],
|
|
dependency_project_id
|
|
);
|
|
assert_eq!(projects.hits[0].dependencies.len(), 1);
|
|
assert_eq!(
|
|
projects.hits[0].dependencies[0].project_id,
|
|
dependency_project_id
|
|
);
|
|
assert!(
|
|
projects.hits[0].dependencies[0]
|
|
.slug
|
|
.as_ref()
|
|
.unwrap()
|
|
.contains("searchable-project-1")
|
|
);
|
|
},
|
|
)
|
|
.await;
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn index_swaps() {
|
|
with_test_environment(
|
|
Some(10),
|
|
|test_env: TestEnvironment<ApiV3>| async move {
|
|
// Reindex
|
|
let resp = test_env.api.reset_search_index().await;
|
|
assert_status!(&resp, StatusCode::NO_CONTENT);
|
|
|
|
// Now we should get results
|
|
let projects = test_env
|
|
.api
|
|
.search_deserialized(
|
|
None,
|
|
Some(json!([["categories:fabric"]])),
|
|
USER_USER_PAT,
|
|
)
|
|
.await;
|
|
assert_eq!(projects.total_hits, 1);
|
|
assert!(projects.hits[0].slug.as_ref().unwrap().contains("alpha"));
|
|
|
|
// Delete the project
|
|
let resp =
|
|
test_env.api.remove_project("alpha", USER_USER_PAT).await;
|
|
assert_status!(&resp, StatusCode::NO_CONTENT);
|
|
|
|
// We should wait for deletions to be indexed
|
|
let projects = test_env
|
|
.api
|
|
.search_deserialized(
|
|
None,
|
|
Some(json!([["categories:fabric"]])),
|
|
USER_USER_PAT,
|
|
)
|
|
.await;
|
|
assert_eq!(projects.total_hits, 0);
|
|
|
|
// When we reindex, it should be still gone
|
|
let resp = test_env.api.reset_search_index().await;
|
|
assert_status!(&resp, StatusCode::NO_CONTENT);
|
|
|
|
let projects = test_env
|
|
.api
|
|
.search_deserialized(
|
|
None,
|
|
Some(json!([["categories:fabric"]])),
|
|
USER_USER_PAT,
|
|
)
|
|
.await;
|
|
assert_eq!(projects.total_hits, 0);
|
|
},
|
|
)
|
|
.await;
|
|
}
|