1
0
Fork 0
mirror of https://github.com/Nutomic/ibis.git synced 2024-11-22 12:31:08 +00:00

add test helper functions

This commit is contained in:
Felix Ableitner 2023-11-16 13:01:42 +01:00
parent a5da3de8c2
commit b69b0fd30a

View file

@ -7,6 +7,7 @@ use fediwiki::federation::objects::instance::DbInstance;
use fediwiki::start; use fediwiki::start;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use reqwest::Client; use reqwest::Client;
use serde::{Deserialize, Serialize};
use serial_test::serial; use serial_test::serial;
use std::sync::Once; use std::sync::Once;
use tracing::log::LevelFilter; use tracing::log::LevelFilter;
@ -35,10 +36,7 @@ async fn test_get_article() -> MyResult<()> {
}); });
let title = "Manu_Chao"; let title = "Manu_Chao";
let res: DbArticle = reqwest::get(format!("http://{hostname}/api/v1/article/{title}")) let res: DbArticle = get(hostname, &format!("article/{title}")).await?;
.await?
.json()
.await?;
assert_eq!(title, res.title); assert_eq!(title, res.title);
assert!(res.local); assert!(res.local);
handle.abort(); handle.abort();
@ -59,32 +57,17 @@ async fn test_follow_instance() -> MyResult<()> {
}); });
// check initial state // check initial state
let alpha_instance: DbInstance = CLIENT let alpha_instance: DbInstance = get(hostname_alpha, "instance").await?;
.get(format!("http://{hostname_alpha}/api/v1/instance"))
.send()
.await?
.json()
.await?;
assert_eq!(0, alpha_instance.follows.len()); assert_eq!(0, alpha_instance.follows.len());
let beta_instance: DbInstance = CLIENT let beta_instance: DbInstance = get(hostname_beta, "instance").await?;
.get(format!("http://{hostname_beta}/api/v1/instance"))
.send()
.await?
.json()
.await?;
assert_eq!(0, beta_instance.followers.len()); assert_eq!(0, beta_instance.followers.len());
// fetch beta instance on alpha // fetch beta instance on alpha
let resolve_object = ResolveObject { let resolve_object = ResolveObject {
id: Url::parse(&format!("http://{hostname_beta}"))?, id: Url::parse(&format!("http://{hostname_beta}"))?,
}; };
let beta_instance_resolved: DbInstance = CLIENT let beta_instance_resolved: DbInstance =
.get(format!("http://{hostname_alpha}/api/v1/resolve_object")) get_query(hostname_beta, "resolve_object", Some(resolve_object)).await?;
.query(&resolve_object)
.send()
.await?
.json()
.await?;
// send follow // send follow
let follow_instance = FollowInstance { let follow_instance = FollowInstance {
@ -97,23 +80,33 @@ async fn test_follow_instance() -> MyResult<()> {
.await?; .await?;
// check that follow was federated // check that follow was federated
let beta_instance: DbInstance = CLIENT let beta_instance: DbInstance = get(hostname_beta, "instance").await?;
.get(format!("http://{hostname_beta}/api/v1/instance"))
.send()
.await?
.json()
.await?;
assert_eq!(1, beta_instance.followers.len()); assert_eq!(1, beta_instance.followers.len());
let alpha_instance: DbInstance = CLIENT let alpha_instance: DbInstance = get(hostname_alpha, "instance").await?;
.get(format!("http://{hostname_alpha}/api/v1/instance"))
.send()
.await?
.json()
.await?;
assert_eq!(1, alpha_instance.follows.len()); assert_eq!(1, alpha_instance.follows.len());
handle_alpha.abort(); handle_alpha.abort();
handle_beta.abort(); handle_beta.abort();
Ok(()) Ok(())
} }
async fn get<T>(hostname: &str, endpoint: &str) -> MyResult<T>
where
T: for<'de> Deserialize<'de>,
{
get_query(hostname, endpoint, None::<i32>).await
}
async fn get_query<T, R>(hostname: &str, endpoint: &str, query: Option<R>) -> MyResult<T>
where
T: for<'de> Deserialize<'de>,
R: Serialize,
{
let mut res = CLIENT.get(format!("http://{}/api/v1/{}", hostname, endpoint));
if let Some(query) = query {
res = res.query(&query);
}
let alpha_instance: T = res.send().await?.json().await?;
Ok(alpha_instance)
}