From 91751153aa780759e2882c0566b461f68d21cc6a Mon Sep 17 00:00:00 2001 From: Felix Ableitner Date: Fri, 1 Mar 2024 15:07:59 +0100 Subject: [PATCH] add test --- Cargo.lock | 1 + crates/db_views_actor/Cargo.toml | 1 + crates/db_views_actor/src/person_view.rs | 108 +++++++++++++---------- 3 files changed, 64 insertions(+), 46 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index 82c9afa6f..a5194f393 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -2755,6 +2755,7 @@ dependencies = [ "diesel", "diesel-async", "lemmy_db_schema", + "lemmy_utils", "pretty_assertions", "serde", "serde_with", diff --git a/crates/db_views_actor/Cargo.toml b/crates/db_views_actor/Cargo.toml index dd2b1aeb6..1892055d1 100644 --- a/crates/db_views_actor/Cargo.toml +++ b/crates/db_views_actor/Cargo.toml @@ -40,6 +40,7 @@ serial_test = { workspace = true } tokio = { workspace = true } pretty_assertions = { workspace = true } url.workspace = true +lemmy_utils.workspace = true [package.metadata.cargo-machete] ignored = ["strum"] diff --git a/crates/db_views_actor/src/person_view.rs b/crates/db_views_actor/src/person_view.rs index 24b6d691f..aee56748e 100644 --- a/crates/db_views_actor/src/person_view.rs +++ b/crates/db_views_actor/src/person_view.rs @@ -179,6 +179,7 @@ mod tests { traits::Crud, utils::build_db_pool_for_tests, }; + use lemmy_utils::error::LemmyResult; use pretty_assertions::assert_eq; use serial_test::serial; @@ -189,64 +190,59 @@ mod tests { bob_local_user: LocalUser, } - async fn init_data(pool: &mut DbPool<'_>) -> Data { - let inserted_instance = Instance::read_or_create(pool, "my_domain.tld".to_string()) - .await - .unwrap(); + async fn init_data(pool: &mut DbPool<'_>) -> LemmyResult { + let inserted_instance = Instance::read_or_create(pool, "my_domain.tld".to_string()).await?; let alice_form = PersonInsertForm::builder() .name("alice".to_string()) .public_key("pubkey".to_string()) .instance_id(inserted_instance.id) + .local(Some(true)) .build(); - let alice = Person::create(pool, &alice_form).await.unwrap(); + let alice = Person::create(pool, &alice_form).await?; let alice_local_user_form = LocalUserInsertForm::builder() .person_id(alice.id) .password_encrypted(String::new()) .build(); - let alice_local_user = LocalUser::create(pool, &alice_local_user_form) - .await - .unwrap(); + let alice_local_user = LocalUser::create(pool, &alice_local_user_form).await?; let bob_form = PersonInsertForm::builder() .name("bob".to_string()) .bot_account(Some(true)) .public_key("pubkey".to_string()) .instance_id(inserted_instance.id) + .local(Some(false)) .build(); - let bob = Person::create(pool, &bob_form).await.unwrap(); + let bob = Person::create(pool, &bob_form).await?; let bob_local_user_form = LocalUserInsertForm::builder() .person_id(bob.id) .password_encrypted(String::new()) .build(); - let bob_local_user = LocalUser::create(pool, &bob_local_user_form).await.unwrap(); + let bob_local_user = LocalUser::create(pool, &bob_local_user_form).await?; - Data { + Ok(Data { alice, alice_local_user, bob, bob_local_user, - } + }) } - async fn cleanup(data: Data, pool: &mut DbPool<'_>) { - LocalUser::delete(pool, data.alice_local_user.id) - .await - .unwrap(); - LocalUser::delete(pool, data.bob_local_user.id) - .await - .unwrap(); - Person::delete(pool, data.alice.id).await.unwrap(); - Person::delete(pool, data.bob.id).await.unwrap(); - Instance::delete(pool, data.bob.instance_id).await.unwrap(); + async fn cleanup(data: Data, pool: &mut DbPool<'_>) -> LemmyResult<()> { + LocalUser::delete(pool, data.alice_local_user.id).await?; + LocalUser::delete(pool, data.bob_local_user.id).await?; + Person::delete(pool, data.alice.id).await?; + Person::delete(pool, data.bob.id).await?; + Instance::delete(pool, data.bob.instance_id).await?; + Ok(()) } #[tokio::test] #[serial] - async fn exclude_deleted() { + async fn exclude_deleted() -> LemmyResult<()> { let pool = &build_db_pool_for_tests().await; let pool = &mut pool.into(); - let data = init_data(pool).await; + let data = init_data(pool).await?; Person::update( pool, @@ -256,8 +252,7 @@ mod tests { ..Default::default() }, ) - .await - .unwrap(); + .await?; let read = PersonView::read(pool, data.alice.id).await; assert_eq!(read.err(), Some(NotFound)); @@ -267,20 +262,19 @@ mod tests { ..Default::default() } .list(pool) - .await - .unwrap(); + .await?; assert_length!(1, list); assert_eq!(list[0].person.id, data.bob.id); - cleanup(data, pool).await; + cleanup(data, pool).await } #[tokio::test] #[serial] - async fn list_banned() { + async fn list_banned() -> LemmyResult<()> { let pool = &build_db_pool_for_tests().await; let pool = &mut pool.into(); - let data = init_data(pool).await; + let data = init_data(pool).await?; Person::update( pool, @@ -290,22 +284,21 @@ mod tests { ..Default::default() }, ) - .await - .unwrap(); + .await?; - let list = PersonView::banned(pool).await.unwrap(); + let list = PersonView::banned(pool).await?; assert_length!(1, list); assert_eq!(list[0].person.id, data.alice.id); - cleanup(data, pool).await; + cleanup(data, pool).await } #[tokio::test] #[serial] - async fn list_admins() { + async fn list_admins() -> LemmyResult<()> { let pool = &build_db_pool_for_tests().await; let pool = &mut pool.into(); - let data = init_data(pool).await; + let data = init_data(pool).await?; LocalUser::update( pool, @@ -315,22 +308,45 @@ mod tests { ..Default::default() }, ) - .await - .unwrap(); + .await?; - let list = PersonView::admins(pool).await.unwrap(); + let list = PersonView::admins(pool).await?; assert_length!(1, list); assert_eq!(list[0].person.id, data.alice.id); - let is_admin = PersonView::read(pool, data.alice.id) - .await - .unwrap() - .is_admin; + let is_admin = PersonView::read(pool, data.alice.id).await?.is_admin; assert!(is_admin); - let is_admin = PersonView::read(pool, data.bob.id).await.unwrap().is_admin; + let is_admin = PersonView::read(pool, data.bob.id).await?.is_admin; assert!(!is_admin); - cleanup(data, pool).await; + cleanup(data, pool).await + } + + #[tokio::test] + #[serial] + async fn listing_type() -> LemmyResult<()> { + let pool = &build_db_pool_for_tests().await; + let pool = &mut pool.into(); + let data = init_data(pool).await?; + + let list = PersonQuery { + listing_type: Some(ListingType::Local), + ..Default::default() + } + .list(pool) + .await?; + assert_length!(1, list); + assert_eq!(list[0].person.id, data.alice.id); + + let list = PersonQuery { + listing_type: Some(ListingType::All), + ..Default::default() + } + .list(pool) + .await?; + assert_length!(2, list); + + cleanup(data, pool).await } }