#![allow(clippy::unwrap_used)] mod common; use crate::common::{TestData, TEST_ARTICLE_DEFAULT_TEXT}; use ibis::{ common::{ utils::extract_domain, ArticleView, CreateArticleForm, EditArticleForm, ForkArticleForm, GetArticleForm, GetUserForm, ListArticlesForm, LoginUserForm, Notification, ProtectArticleForm, RegisterUserForm, SearchArticleForm, }, frontend::error::MyResult, }; use pretty_assertions::{assert_eq, assert_ne}; use retry_future::{LinearRetryStrategy, RetryFuture, RetryPolicy}; use url::Url; #[tokio::test] async fn test_create_read_and_edit_local_article() -> MyResult<()> { let data = TestData::start(false).await; // create article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; assert_eq!(create_form.title, create_res.article.title); assert!(create_res.article.local); // now article can be read let get_article_data = GetArticleForm { title: Some(create_res.article.title.clone()), domain: None, id: None, }; let get_res = data.alpha.get_article(get_article_data.clone()).await?; assert_eq!(create_form.title, get_res.article.title); assert_eq!(TEST_ARTICLE_DEFAULT_TEXT, get_res.article.text); assert!(get_res.article.local); // error on article which wasnt federated let not_found = data.beta.get_article(get_article_data.clone()).await; assert!(not_found.is_err()); // edit article let edit_form = EditArticleForm { article_id: create_res.article.id, new_text: "Lorem Ipsum 2\n".to_string(), summary: "summary".to_string(), previous_version_id: get_res.latest_version, resolve_conflict_id: None, }; let edit_res = data.alpha.edit_article(&edit_form).await?; assert_eq!(edit_form.new_text, edit_res.article.text); assert_eq!(2, edit_res.edits.len()); assert_eq!(edit_form.summary, edit_res.edits[1].edit.summary); let search_form = SearchArticleForm { query: create_form.title.clone(), }; let search_res = data.alpha.search(&search_form).await?; assert_eq!(1, search_res.len()); assert_eq!(edit_res.article, search_res[0]); let list_articles = data .alpha .list_articles(ListArticlesForm { only_local: Some(false), instance_id: None, }) .await?; assert_eq!(2, list_articles.len()); assert_eq!(edit_res.article, list_articles[0]); data.stop() } #[tokio::test] async fn test_create_duplicate_article() -> MyResult<()> { let data = TestData::start(false).await; // create article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; assert_eq!(create_form.title, create_res.article.title); assert!(create_res.article.local); let create_res = data.alpha.create_article(&create_form).await; assert!(create_res.is_err()); data.stop() } #[tokio::test] async fn test_follow_instance() -> MyResult<()> { let data = TestData::start(false).await; // check initial state let alpha_user = data.alpha.site().await?.my_profile.unwrap(); assert_eq!(0, alpha_user.following.len()); let beta_instance = data.beta.get_local_instance().await?; assert_eq!(0, beta_instance.followers.len()); data.alpha .follow_instance_with_resolve(&data.beta.hostname) .await?; // check that follow was federated let alpha_user = data.alpha.site().await?.my_profile.unwrap(); assert_eq!(1, alpha_user.following.len()); assert_eq!(beta_instance.instance.ap_id, alpha_user.following[0].ap_id); let beta_instance = data.beta.get_local_instance().await?; assert_eq!(1, beta_instance.followers.len()); assert_eq!(alpha_user.person.ap_id, beta_instance.followers[0].ap_id); data.stop() } #[tokio::test] async fn test_synchronize_articles() -> MyResult<()> { let data = TestData::start(false).await; // create article on alpha let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; assert_eq!(create_form.title, create_res.article.title); assert_eq!(1, create_res.edits.len()); assert!(create_res.article.local); // edit the article let edit_form = EditArticleForm { article_id: create_res.article.id, new_text: "Lorem Ipsum 2\n".to_string(), summary: "summary".to_string(), previous_version_id: create_res.latest_version, resolve_conflict_id: None, }; data.alpha.edit_article(&edit_form).await?; // fetch alpha instance on beta, articles are also fetched automatically let instance = data .beta .resolve_instance(Url::parse(&format!("http://{}", &data.alpha.hostname))?) .await?; let get_article_data = GetArticleForm { title: Some(create_res.article.title.clone()), ..Default::default() }; // try to read remote article by name, fails without domain let get_res = data.beta.get_article(get_article_data.clone()).await; assert!(get_res.is_err()); // get the article with instance id and compare let get_res = RetryFuture::new( || async { let get_article_data = GetArticleForm { title: Some(create_res.article.title.clone()), domain: Some(instance.domain.clone()), id: None, }; let res = data.beta.get_article(get_article_data).await; match res { Err(_) => Err(RetryPolicy::::Retry(None)), Ok(a) if a.edits.len() < 2 => Err(RetryPolicy::Retry(None)), Ok(a) => Ok(a), } }, LinearRetryStrategy::new(), ) .await?; assert_eq!(create_res.article.ap_id, get_res.article.ap_id); assert_eq!(create_form.title, get_res.article.title); assert_eq!(2, get_res.edits.len()); assert_eq!(edit_form.new_text, get_res.article.text); assert!(!get_res.article.local); data.stop() } #[tokio::test] async fn test_edit_local_article() -> MyResult<()> { let data = TestData::start(false).await; let beta_instance = data .alpha .follow_instance_with_resolve(&data.beta.hostname) .await?; // create new article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.beta.create_article(&create_form).await?; assert_eq!(create_form.title, create_res.article.title); assert!(create_res.article.local); // article should be federated to alpha let get_article_data = GetArticleForm { title: Some(create_res.article.title.to_string()), domain: Some(beta_instance.domain), id: None, }; let get_res = data.alpha.get_article(get_article_data.clone()).await?; assert_eq!(create_res.article.title, get_res.article.title); assert_eq!(1, get_res.edits.len()); assert!(!get_res.article.local); assert_eq!(create_res.article.text, get_res.article.text); // edit the article let edit_form = EditArticleForm { article_id: create_res.article.id, new_text: "Lorem Ipsum 2\n".to_string(), summary: "summary".to_string(), previous_version_id: get_res.latest_version, resolve_conflict_id: None, }; let edit_res = data.beta.edit_article(&edit_form).await?; assert_eq!(edit_res.article.text, edit_form.new_text); assert_eq!(edit_res.edits.len(), 2); assert!(edit_res.edits[0] .edit .ap_id .to_string() .starts_with(&edit_res.article.ap_id.to_string())); // edit should be federated to alpha let get_res = data.alpha.get_article(get_article_data).await?; assert_eq!(edit_res.article.title, get_res.article.title); assert_eq!(edit_res.edits.len(), 2); assert_eq!(edit_res.article.text, get_res.article.text); data.stop() } #[tokio::test] async fn test_edit_remote_article() -> MyResult<()> { let data = TestData::start(false).await; let beta_id_on_alpha = data .alpha .follow_instance_with_resolve(&data.beta.hostname) .await?; let beta_id_on_gamma = data .gamma .follow_instance_with_resolve(&data.beta.hostname) .await?; // create new article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.beta.create_article(&create_form).await?; assert_eq!(&create_form.title, &create_res.article.title); assert!(create_res.article.local); // article should be federated to alpha and gamma let get_article_data_alpha = GetArticleForm { title: Some(create_res.article.title.to_string()), domain: Some(beta_id_on_alpha.domain), id: None, }; let get_res = data .alpha .get_article(get_article_data_alpha.clone()) .await?; assert_eq!(create_res.article.title, get_res.article.title); assert_eq!(1, get_res.edits.len()); assert!(!get_res.article.local); let get_article_data_gamma = GetArticleForm { title: Some(create_res.article.title.to_string()), domain: Some(beta_id_on_gamma.domain), id: None, }; let get_res = data .gamma .get_article(get_article_data_gamma.clone()) .await?; assert_eq!(create_res.article.title, get_res.article.title); assert_eq!(create_res.article.text, get_res.article.text); let edit_form = EditArticleForm { article_id: get_res.article.id, new_text: "Lorem Ipsum 2\n".to_string(), summary: "summary".to_string(), previous_version_id: get_res.latest_version, resolve_conflict_id: None, }; let edit_res = data.alpha.edit_article(&edit_form).await?; assert_eq!(edit_form.new_text, edit_res.article.text); assert_eq!(2, edit_res.edits.len()); assert!(!edit_res.article.local); assert!(edit_res.edits[0] .edit .ap_id .to_string() .starts_with(&edit_res.article.ap_id.to_string())); // edit should be federated to beta and gamma let get_res = data.alpha.get_article(get_article_data_alpha).await?; assert_eq!(edit_res.article.title, get_res.article.title); assert_eq!(edit_res.edits.len(), 2); assert_eq!(edit_res.article.text, get_res.article.text); let get_res = data.gamma.get_article(get_article_data_gamma).await?; assert_eq!(edit_res.article.title, get_res.article.title); assert_eq!(edit_res.edits.len(), 2); assert_eq!(edit_res.article.text, get_res.article.text); data.stop() } #[tokio::test] async fn test_local_edit_conflict() -> MyResult<()> { let data = TestData::start(false).await; // create new article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; assert_eq!(create_form.title, create_res.article.title); assert!(create_res.article.local); // one user edits article let edit_form = EditArticleForm { article_id: create_res.article.id, new_text: "Lorem Ipsum\n".to_string(), summary: "summary".to_string(), previous_version_id: create_res.latest_version.clone(), resolve_conflict_id: None, }; let edit_res = data.alpha.edit_article(&edit_form).await?; assert_eq!(edit_res.article.text, edit_form.new_text); assert_eq!(2, edit_res.edits.len()); // another user edits article, without being aware of previous edit let edit_form = EditArticleForm { article_id: create_res.article.id, new_text: "Ipsum Lorem\n".to_string(), summary: "summary".to_string(), previous_version_id: create_res.latest_version, resolve_conflict_id: None, }; let edit_res = data .alpha .edit_article_with_conflict(&edit_form) .await? .unwrap(); assert_eq!("<<<<<<< ours\nIpsum Lorem\n||||||| original\nsome\nexample\ntext\n=======\nLorem Ipsum\n>>>>>>> theirs\n", edit_res.three_way_merge); let notifications = data.alpha.notifications_list().await?; assert_eq!(1, notifications.len()); let Notification::EditConflict(conflict) = ¬ifications[0] else { panic!() }; assert_eq!(conflict, &edit_res); let edit_form = EditArticleForm { article_id: create_res.article.id, new_text: "Lorem Ipsum and Ipsum Lorem\n".to_string(), summary: "summary".to_string(), previous_version_id: edit_res.previous_version_id, resolve_conflict_id: Some(edit_res.id), }; let edit_res = data.alpha.edit_article(&edit_form).await?; assert_eq!(edit_form.new_text, edit_res.article.text); assert_eq!(0, data.alpha.notifications_count().await?); data.stop() } #[tokio::test] async fn test_federated_edit_conflict() -> MyResult<()> { let data = TestData::start(false).await; let beta_id_on_alpha = data .alpha .follow_instance_with_resolve(&data.beta.hostname) .await?; // create new article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.beta.create_article(&create_form).await?; assert_eq!(create_form.title, create_res.article.title); assert!(create_res.article.local); // fetch article to gamma let resolve_res: ArticleView = data .gamma .resolve_article(create_res.article.ap_id.inner().clone()) .await?; assert_eq!(create_res.article.text, resolve_res.article.text); // alpha edits article let get_article_data = GetArticleForm { title: Some(create_form.title.to_string()), domain: Some(beta_id_on_alpha.domain), id: None, }; let get_res = data.alpha.get_article(get_article_data).await?; assert_eq!(&create_res.edits.len(), &get_res.edits.len()); assert_eq!(&create_res.edits[0].edit.hash, &get_res.edits[0].edit.hash); let edit_form = EditArticleForm { article_id: get_res.article.id, new_text: "Lorem Ipsum\n".to_string(), summary: "summary".to_string(), previous_version_id: create_res.latest_version.clone(), resolve_conflict_id: None, }; let edit_res = data.alpha.edit_article(&edit_form).await?; assert_eq!(edit_res.article.text, edit_form.new_text); assert_eq!(2, edit_res.edits.len()); assert!(!edit_res.article.local); assert!(edit_res.edits[1] .edit .ap_id .to_string() .starts_with(&edit_res.article.ap_id.to_string())); // gamma also edits, as its not the latest version there is a conflict. local version should // not be updated with this conflicting version, instead user needs to handle the conflict let edit_form = EditArticleForm { article_id: resolve_res.article.id, new_text: "aaaa\n".to_string(), summary: "summary".to_string(), previous_version_id: create_res.latest_version, resolve_conflict_id: None, }; let edit_res = data.gamma.edit_article(&edit_form).await?; assert_ne!(edit_form.new_text, edit_res.article.text); assert_eq!(1, edit_res.edits.len()); assert!(!edit_res.article.local); assert_eq!(1, data.gamma.notifications_count().await?); let notifications = data.gamma.notifications_list().await?; assert_eq!(1, notifications.len()); let Notification::EditConflict(conflict) = ¬ifications[0] else { panic!() }; // resolve the conflict let edit_form = EditArticleForm { article_id: resolve_res.article.id, new_text: "aaaa\n".to_string(), summary: "summary".to_string(), previous_version_id: conflict.previous_version_id.clone(), resolve_conflict_id: Some(conflict.id), }; let edit_res = data.gamma.edit_article(&edit_form).await?; assert_eq!(edit_form.new_text, edit_res.article.text); assert_eq!(3, edit_res.edits.len()); assert_eq!(0, data.gamma.notifications_count().await?); let notifications = data.gamma.notifications_list().await?; assert_eq!(0, notifications.len()); data.stop() } #[tokio::test] async fn test_overlapping_edits_no_conflict() -> MyResult<()> { let data = TestData::start(false).await; // create new article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; assert_eq!(create_form.title, create_res.article.title); assert!(create_res.article.local); // one user edits article let edit_form = EditArticleForm { article_id: create_res.article.id, new_text: "my\nexample\ntext\n".to_string(), summary: "summary".to_string(), previous_version_id: create_res.latest_version.clone(), resolve_conflict_id: None, }; let edit_res = data.alpha.edit_article(&edit_form).await?; assert_eq!(edit_res.article.text, edit_form.new_text); assert_eq!(2, edit_res.edits.len()); // another user edits article, without being aware of previous edit let edit_form = EditArticleForm { article_id: create_res.article.id, new_text: "some\nexample\narticle\n".to_string(), summary: "summary".to_string(), previous_version_id: create_res.latest_version, resolve_conflict_id: None, }; let edit_res = data.alpha.edit_article(&edit_form).await?; assert_eq!(0, data.alpha.notifications_count().await?); assert_eq!(3, edit_res.edits.len()); assert_eq!("my\nexample\narticle\n", edit_res.article.text); data.stop() } #[tokio::test] async fn test_fork_article() -> MyResult<()> { let data = TestData::start(false).await; // create article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; assert_eq!(create_form.title, create_res.article.title); assert!(create_res.article.local); // fetch on beta let resolve_res = data .beta .resolve_article(create_res.article.ap_id.into_inner()) .await?; let resolved_article = resolve_res.article; assert_eq!(create_res.edits.len(), resolve_res.edits.len()); // fork the article to local instance let fork_form = ForkArticleForm { article_id: resolved_article.id, new_title: resolved_article.title.clone(), }; let fork_res = data.beta.fork_article(&fork_form).await?; let forked_article = fork_res.article; assert_eq!(resolved_article.title, forked_article.title); assert_eq!(resolved_article.text, forked_article.text); assert_eq!(resolve_res.edits.len(), fork_res.edits.len()); assert_eq!(resolve_res.edits[0].edit.diff, fork_res.edits[0].edit.diff); assert_eq!(resolve_res.edits[0].edit.hash, fork_res.edits[0].edit.hash); assert_ne!(resolve_res.edits[0].edit.id, fork_res.edits[0].edit.id); assert_eq!(resolve_res.latest_version, fork_res.latest_version); assert_ne!(resolved_article.ap_id, forked_article.ap_id); assert!(forked_article.local); let beta_instance = data.beta.get_local_instance().await?; assert_eq!(forked_article.instance_id, beta_instance.instance.id); // now search returns two articles for this title (original and forked) let search_form = SearchArticleForm { query: create_form.title.clone(), }; let search_res = data.beta.search(&search_form).await?; assert_eq!(2, search_res.len()); data.stop() } #[tokio::test] async fn test_user_registration_login() -> MyResult<()> { let data = TestData::start(false).await; let username = "my_user"; let password = "hunter2"; let register_data = RegisterUserForm { username: username.to_string(), password: password.to_string(), }; data.alpha.register(register_data).await?; let login_data = LoginUserForm { username: username.to_string(), password: "asd123".to_string(), }; let invalid_login = data.alpha.login(login_data).await; assert!(invalid_login.is_err()); let login_data = LoginUserForm { username: username.to_string(), password: password.to_string(), }; data.alpha.login(login_data).await?; let my_profile = data.alpha.site().await?.my_profile.unwrap(); assert_eq!(username, my_profile.person.username); data.alpha.logout().await?; let my_profile_after_logout = data.alpha.site().await?.my_profile; assert!(my_profile_after_logout.is_none()); data.stop() } #[tokio::test] async fn test_user_profile() -> MyResult<()> { let data = TestData::start(false).await; // Create an article and federate it, in order to federate the user who created it let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; data.beta .resolve_article(create_res.article.ap_id.into_inner()) .await?; let domain = extract_domain(&data.alpha.site().await?.my_profile.unwrap().person.ap_id); // Now we can fetch the remote user from local api let params = GetUserForm { name: "alpha".to_string(), domain: Some(domain), }; let user = data.beta.get_user(params).await?; assert_eq!("alpha", user.username); assert!(!user.local); data.stop() } #[tokio::test] async fn test_lock_article() -> MyResult<()> { let data = TestData::start(false).await; // create article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; assert!(!create_res.article.protected); // lock from normal user fails let lock_form = ProtectArticleForm { article_id: create_res.article.id, protected: true, }; let lock_res = data.alpha.protect_article(&lock_form).await; assert!(lock_res.is_err()); // login as admin to lock article let form = LoginUserForm { username: "ibis".to_string(), password: "ibis".to_string(), }; data.alpha.login(form).await?; let lock_res = data.alpha.protect_article(&lock_form).await?; assert!(lock_res.protected); let resolve_res: ArticleView = data .gamma .resolve_article(create_res.article.ap_id.inner().clone()) .await?; let edit_form = EditArticleForm { article_id: resolve_res.article.id, new_text: "test".to_string(), summary: "test".to_string(), previous_version_id: resolve_res.latest_version, resolve_conflict_id: None, }; let edit_res = data.gamma.edit_article(&edit_form).await; assert!(edit_res.is_err()); data.stop() } #[tokio::test] async fn test_synchronize_instances() -> MyResult<()> { let data = TestData::start(false).await; // fetch alpha instance on beta data.beta .resolve_instance(Url::parse(&format!("http://{}", &data.alpha.hostname))?) .await?; let beta_instances = data.beta.list_instances().await?; assert_eq!(1, beta_instances.len()); // fetch beta instance on gamma data.gamma .resolve_instance(Url::parse(&format!("http://{}", &data.beta.hostname))?) .await?; // wait until instance collection is fetched let gamma_instances = RetryFuture::new( || async { let res = data.gamma.list_instances().await; match res { Err(_) => Err(RetryPolicy::::Retry(None)), Ok(i) if i.len() < 2 => Err(RetryPolicy::Retry(None)), Ok(i) => Ok(i), } }, LinearRetryStrategy::new(), ) .await?; // now gamma also knows about alpha assert_eq!(2, gamma_instances.len()); assert!(gamma_instances .iter() .any(|i| i.domain == data.alpha.hostname)); data.stop() } #[tokio::test] async fn test_article_approval_required() -> MyResult<()> { let data = TestData::start(true).await; // create article let create_form = CreateArticleForm { title: "Manu_Chao".to_string(), text: TEST_ARTICLE_DEFAULT_TEXT.to_string(), summary: "create article".to_string(), }; let create_res = data.alpha.create_article(&create_form).await?; assert!(!create_res.article.approved); let list_all = data.alpha.list_articles(Default::default()).await?; assert_eq!(1, list_all.len()); assert!(list_all.iter().all(|a| a.id != create_res.article.id)); // login as admin to handle approvals let form = LoginUserForm { username: "ibis".to_string(), password: "ibis".to_string(), }; data.alpha.login(form).await?; assert_eq!(1, data.alpha.notifications_count().await?); let notifications = data.alpha.notifications_list().await?; assert_eq!(1, notifications.len()); let Notification::ArticleApprovalRequired(notif) = ¬ifications[0] else { panic!() }; assert_eq!(create_res.article.id, notif.id); data.alpha.approve_article(notif.id, true).await?; let form = GetArticleForm { id: Some(create_res.article.id), ..Default::default() }; let approved = data.alpha.get_article(form).await?; assert_eq!(create_res.article.id, approved.article.id); assert!(approved.article.approved); let list_all = data.alpha.list_articles(Default::default()).await?; assert_eq!(2, list_all.len()); assert!(list_all.iter().any(|a| a.id == create_res.article.id)); data.stop() }