lemmy_db/src/aggregates/site_aggregates.rs

   1 use diesel::{result::Error, *};
   2 use lemmy_db_schema::schema::site_aggregates;
   3 use serde::Serialize;
   4
   5 #[derive(Queryable, Associations, Identifiable, PartialEq, Debug, Serialize)]
   6 #[table_name = "site_aggregates"]
   7 pub struct SiteAggregates {
   8   pub id: i32,
   9   pub users: i64,
  10   pub posts: i64,
  11   pub comments: i64,
  12   pub communities: i64,
  13 }
  14
  15 impl SiteAggregates {
  16   pub fn read(conn: &PgConnection) -> Result<Self, Error> {
  17     site_aggregates::table.first::<Self>(conn)
  18   }
  19 }
  20
  21 #[cfg(test)]
  22 mod tests {
  23   use crate::{
  24     aggregates::site_aggregates::SiteAggregates,
  25     tests::establish_unpooled_connection,
  26     Crud,
  27     ListingType,
  28     SortType,
  29   };
  30   use lemmy_db_schema::source::{
  31     comment::{Comment, CommentForm},
  32     community::{Community, CommunityForm},
  33     post::{Post, PostForm},
  34     user::{UserForm, User_},
  35   };
  36
  37   #[test]
  38   fn test_crud() {
  39     let conn = establish_unpooled_connection();
  40
  41     let new_user = UserForm {
  42       name: "thommy_site_agg".into(),
  43       preferred_username: None,
  44       password_encrypted: "nope".into(),
  45       email: None,
  46       matrix_user_id: None,
  47       avatar: None,
  48       banner: None,
  49       admin: false,
  50       banned: Some(false),
  51       published: None,
  52       updated: None,
  53       show_nsfw: false,
  54       theme: "browser".into(),
  55       default_sort_type: SortType::Hot as i16,
  56       default_listing_type: ListingType::Subscribed as i16,
  57       lang: "browser".into(),
  58       show_avatars: true,
  59       send_notifications_to_email: false,
  60       actor_id: None,
  61       bio: None,
  62       local: true,
  63       private_key: None,
  64       public_key: None,
  65       last_refreshed_at: None,
  66     };
  67
  68     let inserted_user = User_::create(&conn, &new_user).unwrap();
  69
  70     let new_community = CommunityForm {
  71       name: "TIL_site_agg".into(),
  72       creator_id: inserted_user.id,
  73       title: "nada".to_owned(),
  74       description: None,
  75       category_id: 1,
  76       nsfw: false,
  77       removed: None,
  78       deleted: None,
  79       updated: None,
  80       actor_id: None,
  81       local: true,
  82       private_key: None,
  83       public_key: None,
  84       last_refreshed_at: None,
  85       published: None,
  86       icon: None,
  87       banner: None,
  88     };
  89
  90     let inserted_community = Community::create(&conn, &new_community).unwrap();
  91
  92     let new_post = PostForm {
  93       name: "A test post".into(),
  94       url: None,
  95       body: None,
  96       creator_id: inserted_user.id,
  97       community_id: inserted_community.id,
  98       removed: None,
  99       deleted: None,
 100       locked: None,
 101       stickied: None,
 102       nsfw: false,
 103       updated: None,
 104       embed_title: None,
 105       embed_description: None,
 106       embed_html: None,
 107       thumbnail_url: None,
 108       ap_id: None,
 109       local: true,
 110       published: None,
 111     };
 112
 113     // Insert two of those posts
 114     let inserted_post = Post::create(&conn, &new_post).unwrap();
 115     let _inserted_post_again = Post::create(&conn, &new_post).unwrap();
 116
 117     let comment_form = CommentForm {
 118       content: "A test comment".into(),
 119       creator_id: inserted_user.id,
 120       post_id: inserted_post.id,
 121       removed: None,
 122       deleted: None,
 123       read: None,
 124       parent_id: None,
 125       published: None,
 126       updated: None,
 127       ap_id: None,
 128       local: true,
 129     };
 130
 131     // Insert two of those comments
 132     let inserted_comment = Comment::create(&conn, &comment_form).unwrap();
 133
 134     let child_comment_form = CommentForm {
 135       content: "A test comment".into(),
 136       creator_id: inserted_user.id,
 137       post_id: inserted_post.id,
 138       removed: None,
 139       deleted: None,
 140       read: None,
 141       parent_id: Some(inserted_comment.id),
 142       published: None,
 143       updated: None,
 144       ap_id: None,
 145       local: true,
 146     };
 147
 148     let _inserted_child_comment = Comment::create(&conn, &child_comment_form).unwrap();
 149
 150     let site_aggregates_before_delete = SiteAggregates::read(&conn).unwrap();
 151
 152     assert_eq!(1, site_aggregates_before_delete.users);
 153     assert_eq!(1, site_aggregates_before_delete.communities);
 154     assert_eq!(2, site_aggregates_before_delete.posts);
 155     assert_eq!(2, site_aggregates_before_delete.comments);
 156
 157     // Try a post delete
 158     Post::delete(&conn, inserted_post.id).unwrap();
 159     let site_aggregates_after_post_delete = SiteAggregates::read(&conn).unwrap();
 160     assert_eq!(1, site_aggregates_after_post_delete.posts);
 161     assert_eq!(0, site_aggregates_after_post_delete.comments);
 162
 163     // This shouuld delete all the associated rows, and fire triggers
 164     let user_num_deleted = User_::delete(&conn, inserted_user.id).unwrap();
 165     assert_eq!(1, user_num_deleted);
 166
 167     let site_aggregates_after_delete = SiteAggregates::read(&conn).unwrap();
 168     assert_eq!(0, site_aggregates_after_delete.users);
 169     assert_eq!(0, site_aggregates_after_delete.communities);
 170     assert_eq!(0, site_aggregates_after_delete.posts);
 171     assert_eq!(0, site_aggregates_after_delete.comments);
 172   }
 173 }