crates/db_schema/src/aggregates/comment_aggregates.rs

   1 use crate::{
   2   aggregates::structs::CommentAggregates,
   3   newtypes::CommentId,
   4   schema::comment_aggregates,
   5 };
   6 use diesel::{result::Error, *};
   7
   8 impl CommentAggregates {
   9   pub fn read(conn: &mut PgConnection, comment_id: CommentId) -> Result<Self, Error> {
  10     comment_aggregates::table
  11       .filter(comment_aggregates::comment_id.eq(comment_id))
  12       .first::<Self>(conn)
  13   }
  14 }
  15
  16 #[cfg(test)]
  17 mod tests {
  18   use crate::{
  19     aggregates::comment_aggregates::CommentAggregates,
  20     source::{
  21       comment::{Comment, CommentForm, CommentLike, CommentLikeForm},
  22       community::{Community, CommunityForm},
  23       person::{Person, PersonForm},
  24       post::{Post, PostForm},
  25     },
  26     traits::{Crud, Likeable},
  27     utils::establish_unpooled_connection,
  28   };
  29   use serial_test::serial;
  30
  31   #[test]
  32   #[serial]
  33   fn test_crud() {
  34     let conn = &mut establish_unpooled_connection();
  35
  36     let new_person = PersonForm {
  37       name: "thommy_comment_agg".into(),
  38       public_key: Some("pubkey".to_string()),
  39       ..PersonForm::default()
  40     };
  41
  42     let inserted_person = Person::create(conn, &new_person).unwrap();
  43
  44     let another_person = PersonForm {
  45       name: "jerry_comment_agg".into(),
  46       public_key: Some("pubkey".to_string()),
  47       ..PersonForm::default()
  48     };
  49
  50     let another_inserted_person = Person::create(conn, &another_person).unwrap();
  51
  52     let new_community = CommunityForm {
  53       name: "TIL_comment_agg".into(),
  54       title: "nada".to_owned(),
  55       public_key: Some("pubkey".to_string()),
  56       ..CommunityForm::default()
  57     };
  58
  59     let inserted_community = Community::create(conn, &new_community).unwrap();
  60
  61     let new_post = PostForm {
  62       name: "A test post".into(),
  63       creator_id: inserted_person.id,
  64       community_id: inserted_community.id,
  65       ..PostForm::default()
  66     };
  67
  68     let inserted_post = Post::create(conn, &new_post).unwrap();
  69
  70     let comment_form = CommentForm {
  71       content: "A test comment".into(),
  72       creator_id: inserted_person.id,
  73       post_id: inserted_post.id,
  74       ..CommentForm::default()
  75     };
  76
  77     let inserted_comment = Comment::create(conn, &comment_form, None).unwrap();
  78
  79     let child_comment_form = CommentForm {
  80       content: "A test comment".into(),
  81       creator_id: inserted_person.id,
  82       post_id: inserted_post.id,
  83       ..CommentForm::default()
  84     };
  85
  86     let _inserted_child_comment =
  87       Comment::create(conn, &child_comment_form, Some(&inserted_comment.path)).unwrap();
  88
  89     let comment_like = CommentLikeForm {
  90       comment_id: inserted_comment.id,
  91       post_id: inserted_post.id,
  92       person_id: inserted_person.id,
  93       score: 1,
  94     };
  95
  96     CommentLike::like(conn, &comment_like).unwrap();
  97
  98     let comment_aggs_before_delete = CommentAggregates::read(conn, inserted_comment.id).unwrap();
  99
 100     assert_eq!(1, comment_aggs_before_delete.score);
 101     assert_eq!(1, comment_aggs_before_delete.upvotes);
 102     assert_eq!(0, comment_aggs_before_delete.downvotes);
 103
 104     // Add a post dislike from the other person
 105     let comment_dislike = CommentLikeForm {
 106       comment_id: inserted_comment.id,
 107       post_id: inserted_post.id,
 108       person_id: another_inserted_person.id,
 109       score: -1,
 110     };
 111
 112     CommentLike::like(conn, &comment_dislike).unwrap();
 113
 114     let comment_aggs_after_dislike = CommentAggregates::read(conn, inserted_comment.id).unwrap();
 115
 116     assert_eq!(0, comment_aggs_after_dislike.score);
 117     assert_eq!(1, comment_aggs_after_dislike.upvotes);
 118     assert_eq!(1, comment_aggs_after_dislike.downvotes);
 119
 120     // Remove the first comment like
 121     CommentLike::remove(conn, inserted_person.id, inserted_comment.id).unwrap();
 122     let after_like_remove = CommentAggregates::read(conn, inserted_comment.id).unwrap();
 123     assert_eq!(-1, after_like_remove.score);
 124     assert_eq!(0, after_like_remove.upvotes);
 125     assert_eq!(1, after_like_remove.downvotes);
 126
 127     // Remove the parent post
 128     Post::delete(conn, inserted_post.id).unwrap();
 129
 130     // Should be none found, since the post was deleted
 131     let after_delete = CommentAggregates::read(conn, inserted_comment.id);
 132     assert!(after_delete.is_err());
 133
 134     // This should delete all the associated rows, and fire triggers
 135     Person::delete(conn, another_inserted_person.id).unwrap();
 136     let person_num_deleted = Person::delete(conn, inserted_person.id).unwrap();
 137     assert_eq!(1, person_num_deleted);
 138
 139     // Delete the community
 140     let community_num_deleted = Community::delete(conn, inserted_community.id).unwrap();
 141     assert_eq!(1, community_num_deleted);
 142   }
 143 }