crates/db_schema/src/impls/moderator.rs

   1 use crate::{source::moderator::*, traits::Crud};
   2 use diesel::{dsl::*, result::Error, *};
   3
   4 impl Crud for ModRemovePost {
   5   type Form = ModRemovePostForm;
   6   type IdType = i32;
   7   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
   8     use crate::schema::mod_remove_post::dsl::*;
   9     mod_remove_post.find(from_id).first::<Self>(conn)
  10   }
  11
  12   fn create(conn: &PgConnection, form: &ModRemovePostForm) -> Result<Self, Error> {
  13     use crate::schema::mod_remove_post::dsl::*;
  14     insert_into(mod_remove_post)
  15       .values(form)
  16       .get_result::<Self>(conn)
  17   }
  18
  19   fn update(conn: &PgConnection, from_id: i32, form: &ModRemovePostForm) -> Result<Self, Error> {
  20     use crate::schema::mod_remove_post::dsl::*;
  21     diesel::update(mod_remove_post.find(from_id))
  22       .set(form)
  23       .get_result::<Self>(conn)
  24   }
  25 }
  26
  27 impl Crud for ModLockPost {
  28   type Form = ModLockPostForm;
  29   type IdType = i32;
  30   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
  31     use crate::schema::mod_lock_post::dsl::*;
  32     mod_lock_post.find(from_id).first::<Self>(conn)
  33   }
  34
  35   fn create(conn: &PgConnection, form: &ModLockPostForm) -> Result<Self, Error> {
  36     use crate::schema::mod_lock_post::dsl::*;
  37     insert_into(mod_lock_post)
  38       .values(form)
  39       .get_result::<Self>(conn)
  40   }
  41
  42   fn update(conn: &PgConnection, from_id: i32, form: &ModLockPostForm) -> Result<Self, Error> {
  43     use crate::schema::mod_lock_post::dsl::*;
  44     diesel::update(mod_lock_post.find(from_id))
  45       .set(form)
  46       .get_result::<Self>(conn)
  47   }
  48 }
  49
  50 impl Crud for ModStickyPost {
  51   type Form = ModStickyPostForm;
  52   type IdType = i32;
  53   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
  54     use crate::schema::mod_sticky_post::dsl::*;
  55     mod_sticky_post.find(from_id).first::<Self>(conn)
  56   }
  57
  58   fn create(conn: &PgConnection, form: &ModStickyPostForm) -> Result<Self, Error> {
  59     use crate::schema::mod_sticky_post::dsl::*;
  60     insert_into(mod_sticky_post)
  61       .values(form)
  62       .get_result::<Self>(conn)
  63   }
  64
  65   fn update(conn: &PgConnection, from_id: i32, form: &ModStickyPostForm) -> Result<Self, Error> {
  66     use crate::schema::mod_sticky_post::dsl::*;
  67     diesel::update(mod_sticky_post.find(from_id))
  68       .set(form)
  69       .get_result::<Self>(conn)
  70   }
  71 }
  72
  73 impl Crud for ModRemoveComment {
  74   type Form = ModRemoveCommentForm;
  75   type IdType = i32;
  76   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
  77     use crate::schema::mod_remove_comment::dsl::*;
  78     mod_remove_comment.find(from_id).first::<Self>(conn)
  79   }
  80
  81   fn create(conn: &PgConnection, form: &ModRemoveCommentForm) -> Result<Self, Error> {
  82     use crate::schema::mod_remove_comment::dsl::*;
  83     insert_into(mod_remove_comment)
  84       .values(form)
  85       .get_result::<Self>(conn)
  86   }
  87
  88   fn update(conn: &PgConnection, from_id: i32, form: &ModRemoveCommentForm) -> Result<Self, Error> {
  89     use crate::schema::mod_remove_comment::dsl::*;
  90     diesel::update(mod_remove_comment.find(from_id))
  91       .set(form)
  92       .get_result::<Self>(conn)
  93   }
  94 }
  95
  96 impl Crud for ModRemoveCommunity {
  97   type Form = ModRemoveCommunityForm;
  98   type IdType = i32;
  99   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
 100     use crate::schema::mod_remove_community::dsl::*;
 101     mod_remove_community.find(from_id).first::<Self>(conn)
 102   }
 103
 104   fn create(conn: &PgConnection, form: &ModRemoveCommunityForm) -> Result<Self, Error> {
 105     use crate::schema::mod_remove_community::dsl::*;
 106     insert_into(mod_remove_community)
 107       .values(form)
 108       .get_result::<Self>(conn)
 109   }
 110
 111   fn update(
 112     conn: &PgConnection,
 113     from_id: i32,
 114     form: &ModRemoveCommunityForm,
 115   ) -> Result<Self, Error> {
 116     use crate::schema::mod_remove_community::dsl::*;
 117     diesel::update(mod_remove_community.find(from_id))
 118       .set(form)
 119       .get_result::<Self>(conn)
 120   }
 121 }
 122
 123 impl Crud for ModBanFromCommunity {
 124   type Form = ModBanFromCommunityForm;
 125   type IdType = i32;
 126   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
 127     use crate::schema::mod_ban_from_community::dsl::*;
 128     mod_ban_from_community.find(from_id).first::<Self>(conn)
 129   }
 130
 131   fn create(conn: &PgConnection, form: &ModBanFromCommunityForm) -> Result<Self, Error> {
 132     use crate::schema::mod_ban_from_community::dsl::*;
 133     insert_into(mod_ban_from_community)
 134       .values(form)
 135       .get_result::<Self>(conn)
 136   }
 137
 138   fn update(
 139     conn: &PgConnection,
 140     from_id: i32,
 141     form: &ModBanFromCommunityForm,
 142   ) -> Result<Self, Error> {
 143     use crate::schema::mod_ban_from_community::dsl::*;
 144     diesel::update(mod_ban_from_community.find(from_id))
 145       .set(form)
 146       .get_result::<Self>(conn)
 147   }
 148 }
 149
 150 impl Crud for ModBan {
 151   type Form = ModBanForm;
 152   type IdType = i32;
 153   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
 154     use crate::schema::mod_ban::dsl::*;
 155     mod_ban.find(from_id).first::<Self>(conn)
 156   }
 157
 158   fn create(conn: &PgConnection, form: &ModBanForm) -> Result<Self, Error> {
 159     use crate::schema::mod_ban::dsl::*;
 160     insert_into(mod_ban).values(form).get_result::<Self>(conn)
 161   }
 162
 163   fn update(conn: &PgConnection, from_id: i32, form: &ModBanForm) -> Result<Self, Error> {
 164     use crate::schema::mod_ban::dsl::*;
 165     diesel::update(mod_ban.find(from_id))
 166       .set(form)
 167       .get_result::<Self>(conn)
 168   }
 169 }
 170
 171 impl Crud for ModHideCommunity {
 172   type Form = ModHideCommunityForm;
 173   type IdType = i32;
 174
 175   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
 176     use crate::schema::mod_hide_community::dsl::*;
 177     mod_hide_community.find(from_id).first::<Self>(conn)
 178   }
 179
 180   fn create(conn: &PgConnection, form: &ModHideCommunityForm) -> Result<Self, Error> {
 181     use crate::schema::mod_hide_community::dsl::*;
 182     insert_into(mod_hide_community)
 183       .values(form)
 184       .get_result::<Self>(conn)
 185   }
 186
 187   fn update(conn: &PgConnection, from_id: i32, form: &ModHideCommunityForm) -> Result<Self, Error> {
 188     use crate::schema::mod_hide_community::dsl::*;
 189     diesel::update(mod_hide_community.find(from_id))
 190       .set(form)
 191       .get_result::<Self>(conn)
 192   }
 193 }
 194
 195 impl Crud for ModAddCommunity {
 196   type Form = ModAddCommunityForm;
 197   type IdType = i32;
 198   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
 199     use crate::schema::mod_add_community::dsl::*;
 200     mod_add_community.find(from_id).first::<Self>(conn)
 201   }
 202
 203   fn create(conn: &PgConnection, form: &ModAddCommunityForm) -> Result<Self, Error> {
 204     use crate::schema::mod_add_community::dsl::*;
 205     insert_into(mod_add_community)
 206       .values(form)
 207       .get_result::<Self>(conn)
 208   }
 209
 210   fn update(conn: &PgConnection, from_id: i32, form: &ModAddCommunityForm) -> Result<Self, Error> {
 211     use crate::schema::mod_add_community::dsl::*;
 212     diesel::update(mod_add_community.find(from_id))
 213       .set(form)
 214       .get_result::<Self>(conn)
 215   }
 216 }
 217
 218 impl Crud for ModTransferCommunity {
 219   type Form = ModTransferCommunityForm;
 220   type IdType = i32;
 221   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
 222     use crate::schema::mod_transfer_community::dsl::*;
 223     mod_transfer_community.find(from_id).first::<Self>(conn)
 224   }
 225
 226   fn create(conn: &PgConnection, form: &ModTransferCommunityForm) -> Result<Self, Error> {
 227     use crate::schema::mod_transfer_community::dsl::*;
 228     insert_into(mod_transfer_community)
 229       .values(form)
 230       .get_result::<Self>(conn)
 231   }
 232
 233   fn update(
 234     conn: &PgConnection,
 235     from_id: i32,
 236     form: &ModTransferCommunityForm,
 237   ) -> Result<Self, Error> {
 238     use crate::schema::mod_transfer_community::dsl::*;
 239     diesel::update(mod_transfer_community.find(from_id))
 240       .set(form)
 241       .get_result::<Self>(conn)
 242   }
 243 }
 244
 245 impl Crud for ModAdd {
 246   type Form = ModAddForm;
 247   type IdType = i32;
 248   fn read(conn: &PgConnection, from_id: i32) -> Result<Self, Error> {
 249     use crate::schema::mod_add::dsl::*;
 250     mod_add.find(from_id).first::<Self>(conn)
 251   }
 252
 253   fn create(conn: &PgConnection, form: &ModAddForm) -> Result<Self, Error> {
 254     use crate::schema::mod_add::dsl::*;
 255     insert_into(mod_add).values(form).get_result::<Self>(conn)
 256   }
 257
 258   fn update(conn: &PgConnection, from_id: i32, form: &ModAddForm) -> Result<Self, Error> {
 259     use crate::schema::mod_add::dsl::*;
 260     diesel::update(mod_add.find(from_id))
 261       .set(form)
 262       .get_result::<Self>(conn)
 263   }
 264 }
 265
 266 #[cfg(test)]
 267 mod tests {
 268   use crate::{
 269     establish_unpooled_connection,
 270     source::{comment::*, community::*, moderator::*, person::*, post::*},
 271     traits::Crud,
 272   };
 273   use serial_test::serial;
 274
 275   // use Crud;
 276   #[test]
 277   #[serial]
 278   fn test_crud() {
 279     let conn = establish_unpooled_connection();
 280
 281     let new_mod = PersonForm {
 282       name: "the mod".into(),
 283       ..PersonForm::default()
 284     };
 285
 286     let inserted_mod = Person::create(&conn, &new_mod).unwrap();
 287
 288     let new_person = PersonForm {
 289       name: "jim2".into(),
 290       ..PersonForm::default()
 291     };
 292
 293     let inserted_person = Person::create(&conn, &new_person).unwrap();
 294
 295     let new_community = CommunityForm {
 296       name: "mod_community".to_string(),
 297       title: "nada".to_owned(),
 298       ..CommunityForm::default()
 299     };
 300
 301     let inserted_community = Community::create(&conn, &new_community).unwrap();
 302
 303     let new_post = PostForm {
 304       name: "A test post thweep".into(),
 305       creator_id: inserted_person.id,
 306       community_id: inserted_community.id,
 307       ..PostForm::default()
 308     };
 309
 310     let inserted_post = Post::create(&conn, &new_post).unwrap();
 311
 312     let comment_form = CommentForm {
 313       content: "A test comment".into(),
 314       creator_id: inserted_person.id,
 315       post_id: inserted_post.id,
 316       ..CommentForm::default()
 317     };
 318
 319     let inserted_comment = Comment::create(&conn, &comment_form).unwrap();
 320
 321     // Now the actual tests
 322
 323     // remove post
 324     let mod_remove_post_form = ModRemovePostForm {
 325       mod_person_id: inserted_mod.id,
 326       post_id: inserted_post.id,
 327       reason: None,
 328       removed: None,
 329     };
 330     let inserted_mod_remove_post = ModRemovePost::create(&conn, &mod_remove_post_form).unwrap();
 331     let read_mod_remove_post = ModRemovePost::read(&conn, inserted_mod_remove_post.id).unwrap();
 332     let expected_mod_remove_post = ModRemovePost {
 333       id: inserted_mod_remove_post.id,
 334       post_id: inserted_post.id,
 335       mod_person_id: inserted_mod.id,
 336       reason: None,
 337       removed: Some(true),
 338       when_: inserted_mod_remove_post.when_,
 339     };
 340
 341     // lock post
 342
 343     let mod_lock_post_form = ModLockPostForm {
 344       mod_person_id: inserted_mod.id,
 345       post_id: inserted_post.id,
 346       locked: None,
 347     };
 348     let inserted_mod_lock_post = ModLockPost::create(&conn, &mod_lock_post_form).unwrap();
 349     let read_mod_lock_post = ModLockPost::read(&conn, inserted_mod_lock_post.id).unwrap();
 350     let expected_mod_lock_post = ModLockPost {
 351       id: inserted_mod_lock_post.id,
 352       post_id: inserted_post.id,
 353       mod_person_id: inserted_mod.id,
 354       locked: Some(true),
 355       when_: inserted_mod_lock_post.when_,
 356     };
 357
 358     // sticky post
 359
 360     let mod_sticky_post_form = ModStickyPostForm {
 361       mod_person_id: inserted_mod.id,
 362       post_id: inserted_post.id,
 363       stickied: None,
 364     };
 365     let inserted_mod_sticky_post = ModStickyPost::create(&conn, &mod_sticky_post_form).unwrap();
 366     let read_mod_sticky_post = ModStickyPost::read(&conn, inserted_mod_sticky_post.id).unwrap();
 367     let expected_mod_sticky_post = ModStickyPost {
 368       id: inserted_mod_sticky_post.id,
 369       post_id: inserted_post.id,
 370       mod_person_id: inserted_mod.id,
 371       stickied: Some(true),
 372       when_: inserted_mod_sticky_post.when_,
 373     };
 374
 375     // comment
 376
 377     let mod_remove_comment_form = ModRemoveCommentForm {
 378       mod_person_id: inserted_mod.id,
 379       comment_id: inserted_comment.id,
 380       reason: None,
 381       removed: None,
 382     };
 383     let inserted_mod_remove_comment =
 384       ModRemoveComment::create(&conn, &mod_remove_comment_form).unwrap();
 385     let read_mod_remove_comment =
 386       ModRemoveComment::read(&conn, inserted_mod_remove_comment.id).unwrap();
 387     let expected_mod_remove_comment = ModRemoveComment {
 388       id: inserted_mod_remove_comment.id,
 389       comment_id: inserted_comment.id,
 390       mod_person_id: inserted_mod.id,
 391       reason: None,
 392       removed: Some(true),
 393       when_: inserted_mod_remove_comment.when_,
 394     };
 395
 396     // community
 397
 398     let mod_remove_community_form = ModRemoveCommunityForm {
 399       mod_person_id: inserted_mod.id,
 400       community_id: inserted_community.id,
 401       reason: None,
 402       removed: None,
 403       expires: None,
 404     };
 405     let inserted_mod_remove_community =
 406       ModRemoveCommunity::create(&conn, &mod_remove_community_form).unwrap();
 407     let read_mod_remove_community =
 408       ModRemoveCommunity::read(&conn, inserted_mod_remove_community.id).unwrap();
 409     let expected_mod_remove_community = ModRemoveCommunity {
 410       id: inserted_mod_remove_community.id,
 411       community_id: inserted_community.id,
 412       mod_person_id: inserted_mod.id,
 413       reason: None,
 414       removed: Some(true),
 415       expires: None,
 416       when_: inserted_mod_remove_community.when_,
 417     };
 418
 419     // ban from community
 420
 421     let mod_ban_from_community_form = ModBanFromCommunityForm {
 422       mod_person_id: inserted_mod.id,
 423       other_person_id: inserted_person.id,
 424       community_id: inserted_community.id,
 425       reason: None,
 426       banned: None,
 427       expires: None,
 428     };
 429     let inserted_mod_ban_from_community =
 430       ModBanFromCommunity::create(&conn, &mod_ban_from_community_form).unwrap();
 431     let read_mod_ban_from_community =
 432       ModBanFromCommunity::read(&conn, inserted_mod_ban_from_community.id).unwrap();
 433     let expected_mod_ban_from_community = ModBanFromCommunity {
 434       id: inserted_mod_ban_from_community.id,
 435       community_id: inserted_community.id,
 436       mod_person_id: inserted_mod.id,
 437       other_person_id: inserted_person.id,
 438       reason: None,
 439       banned: Some(true),
 440       expires: None,
 441       when_: inserted_mod_ban_from_community.when_,
 442     };
 443
 444     // ban
 445
 446     let mod_ban_form = ModBanForm {
 447       mod_person_id: inserted_mod.id,
 448       other_person_id: inserted_person.id,
 449       reason: None,
 450       banned: None,
 451       expires: None,
 452     };
 453     let inserted_mod_ban = ModBan::create(&conn, &mod_ban_form).unwrap();
 454     let read_mod_ban = ModBan::read(&conn, inserted_mod_ban.id).unwrap();
 455     let expected_mod_ban = ModBan {
 456       id: inserted_mod_ban.id,
 457       mod_person_id: inserted_mod.id,
 458       other_person_id: inserted_person.id,
 459       reason: None,
 460       banned: Some(true),
 461       expires: None,
 462       when_: inserted_mod_ban.when_,
 463     };
 464
 465     // mod add community
 466
 467     let mod_add_community_form = ModAddCommunityForm {
 468       mod_person_id: inserted_mod.id,
 469       other_person_id: inserted_person.id,
 470       community_id: inserted_community.id,
 471       removed: None,
 472     };
 473     let inserted_mod_add_community =
 474       ModAddCommunity::create(&conn, &mod_add_community_form).unwrap();
 475     let read_mod_add_community =
 476       ModAddCommunity::read(&conn, inserted_mod_add_community.id).unwrap();
 477     let expected_mod_add_community = ModAddCommunity {
 478       id: inserted_mod_add_community.id,
 479       community_id: inserted_community.id,
 480       mod_person_id: inserted_mod.id,
 481       other_person_id: inserted_person.id,
 482       removed: Some(false),
 483       when_: inserted_mod_add_community.when_,
 484     };
 485
 486     // mod add
 487
 488     let mod_add_form = ModAddForm {
 489       mod_person_id: inserted_mod.id,
 490       other_person_id: inserted_person.id,
 491       removed: None,
 492     };
 493     let inserted_mod_add = ModAdd::create(&conn, &mod_add_form).unwrap();
 494     let read_mod_add = ModAdd::read(&conn, inserted_mod_add.id).unwrap();
 495     let expected_mod_add = ModAdd {
 496       id: inserted_mod_add.id,
 497       mod_person_id: inserted_mod.id,
 498       other_person_id: inserted_person.id,
 499       removed: Some(false),
 500       when_: inserted_mod_add.when_,
 501     };
 502
 503     Comment::delete(&conn, inserted_comment.id).unwrap();
 504     Post::delete(&conn, inserted_post.id).unwrap();
 505     Community::delete(&conn, inserted_community.id).unwrap();
 506     Person::delete(&conn, inserted_person.id).unwrap();
 507     Person::delete(&conn, inserted_mod.id).unwrap();
 508
 509     assert_eq!(expected_mod_remove_post, read_mod_remove_post);
 510     assert_eq!(expected_mod_lock_post, read_mod_lock_post);
 511     assert_eq!(expected_mod_sticky_post, read_mod_sticky_post);
 512     assert_eq!(expected_mod_remove_comment, read_mod_remove_comment);
 513     assert_eq!(expected_mod_remove_community, read_mod_remove_community);
 514     assert_eq!(expected_mod_ban_from_community, read_mod_ban_from_community);
 515     assert_eq!(expected_mod_ban, read_mod_ban);
 516     assert_eq!(expected_mod_add_community, read_mod_add_community);
 517     assert_eq!(expected_mod_add, read_mod_add);
 518   }
 519 }