FediMovies/fedimovies
1
1
Fork 0
Code Issues 3 Pull requests Projects Releases Packages Activity

Add test for get_relationship function

Browse source
This commit is contained in:
silverpill 2022-02-02 18:02:25 +00:00
parent 9f1f950aa3
commit 548e000b3e
1 changed files with 51 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
src/models/relationships/queries.rs
View file

@ -38,12 +38,12 @@ pub async fn get_relationships(
EXISTS ( EXISTS (
SELECT 1 FROM follow_request SELECT 1 FROM follow_request
WHERE source_id = $1 AND target_id = actor_profile.id WHERE source_id = $1 AND target_id = actor_profile.id
AND request_status = 1 AND request_status = $3
) AS requested ) AS requested
FROM actor_profile FROM actor_profile
WHERE actor_profile.id = ANY($2) WHERE actor_profile.id = ANY($2)
", ",
&[&source_id, &target_ids], &[&source_id, &target_ids, &FollowRequestStatus::Pending],
).await?; ).await?;
let relationships = rows.iter() let relationships = rows.iter()
.map(Relationship::try_from) .map(Relationship::try_from)
@ -71,12 +71,12 @@ pub async fn get_relationship(
EXISTS ( EXISTS (
SELECT 1 FROM follow_request SELECT 1 FROM follow_request
WHERE source_id = $1 AND target_id = actor_profile.id WHERE source_id = $1 AND target_id = actor_profile.id
AND request_status = 1 AND request_status = $3
) AS requested ) AS requested
FROM actor_profile FROM actor_profile
WHERE actor_profile.id = $2 WHERE actor_profile.id = $2
", ",
&[&source_id, &target_id], &[&source_id, &target_id, &FollowRequestStatus::Pending],
).await?; ).await?;
let row = maybe_row.ok_or(DatabaseError::NotFound("profile"))?; let row = maybe_row.ok_or(DatabaseError::NotFound("profile"))?;
let relationship = Relationship::try_from(&row)?; let relationship = Relationship::try_from(&row)?;
@ -307,3 +307,50 @@ pub async fn get_following(
.collect::<Result<_, _>>()?; .collect::<Result<_, _>>()?;
Ok(profiles) Ok(profiles)
} }
#[cfg(test)]
mod tests {
use serial_test::serial;
use crate::database::test_utils::create_test_database;
use crate::models::relationships::queries::follow;
use crate::models::users::queries::create_user;
use crate::models::users::types::UserCreateData;
use super::*;
#[tokio::test]
#[serial]
async fn test_get_relationship() {
let db_client = &mut create_test_database().await;
let user_data_1 = UserCreateData {
username: "user".to_string(),
..Default::default()
};
let user_1 = create_user(db_client, user_data_1).await.unwrap();
let user_data_2 = UserCreateData {
username: "another-user".to_string(),
..Default::default()
};
let user_2 = create_user(db_client, user_data_2).await.unwrap();
// Initial state
let relationship = get_relationship(db_client, &user_1.id, &user_2.id).await.unwrap();
assert_eq!(relationship.id, user_2.id);
assert_eq!(relationship.following, false);
assert_eq!(relationship.followed_by, false);
assert_eq!(relationship.requested, false);
// Follow request
let follow_request = create_follow_request(db_client, &user_1.id, &user_2.id).await.unwrap();
let relationship = get_relationship(db_client, &user_1.id, &user_2.id).await.unwrap();
assert_eq!(relationship.id, user_2.id);
assert_eq!(relationship.following, false);
assert_eq!(relationship.followed_by, false);
assert_eq!(relationship.requested, true);
// Mutual follow
follow_request_accepted(db_client, &follow_request.id).await.unwrap();
follow(db_client, &user_2.id, &user_1.id).await.unwrap();
let relationship = get_relationship(db_client, &user_1.id, &user_2.id).await.unwrap();
assert_eq!(relationship.id, user_2.id);
assert_eq!(relationship.following, true);
assert_eq!(relationship.followed_by, true);
assert_eq!(relationship.requested, false);
}
}