X-Git-Url: http://these/git/?a=blobdiff_plain;f=crates%2Fdb_schema%2Fsrc%2Fimpls%2Fpassword_reset_request.rs;h=dfa56e209508161d3676676ef8632e19062baf02;hb=c9f140742925d6da20103124b49f2b58a35fc2b8;hp=780c6c2804abd14698b1072b30319df10ac0ee3b;hpb=4e6409f325bca5b2727b19c24d77ffa2b59109b1;p=lemmy.git diff --git a/crates/db_schema/src/impls/password_reset_request.rs b/crates/db_schema/src/impls/password_reset_request.rs index 780c6c28..dfa56e20 100644 --- a/crates/db_schema/src/impls/password_reset_request.rs +++ b/crates/db_schema/src/impls/password_reset_request.rs @@ -10,18 +10,18 @@ use sha2::{Digest, Sha256}; impl Crud for PasswordResetRequest { type Form = PasswordResetRequestForm; type IdType = i32; - fn read(conn: &PgConnection, password_reset_request_id: i32) -> Result { + fn read(conn: &mut PgConnection, password_reset_request_id: i32) -> Result { password_reset_request .find(password_reset_request_id) .first::(conn) } - fn create(conn: &PgConnection, form: &PasswordResetRequestForm) -> Result { + fn create(conn: &mut PgConnection, form: &PasswordResetRequestForm) -> Result { insert_into(password_reset_request) .values(form) .get_result::(conn) } fn update( - conn: &PgConnection, + conn: &mut PgConnection, password_reset_request_id: i32, form: &PasswordResetRequestForm, ) -> Result { @@ -33,7 +33,7 @@ impl Crud for PasswordResetRequest { impl PasswordResetRequest { pub fn create_token( - conn: &PgConnection, + conn: &mut PgConnection, from_local_user_id: LocalUserId, token: &str, ) -> Result { @@ -48,7 +48,10 @@ impl PasswordResetRequest { Self::create(conn, &form) } - pub fn read_from_token(conn: &PgConnection, token: &str) -> Result { + pub fn read_from_token( + conn: &mut PgConnection, + token: &str, + ) -> Result { let mut hasher = Sha256::new(); hasher.update(token); let token_hash: String = bytes_to_hex(hasher.finalize().to_vec()); @@ -83,7 +86,7 @@ mod tests { #[test] #[serial] fn test_crud() { - let conn = establish_unpooled_connection(); + let conn = &mut establish_unpooled_connection(); let new_person = PersonForm { name: "thommy prw".into(), @@ -91,7 +94,7 @@ mod tests { ..PersonForm::default() }; - let inserted_person = Person::create(&conn, &new_person).unwrap(); + let inserted_person = Person::create(conn, &new_person).unwrap(); let new_local_user = LocalUserForm { person_id: Some(inserted_person.id), @@ -99,13 +102,13 @@ mod tests { ..LocalUserForm::default() }; - let inserted_local_user = LocalUser::create(&conn, &new_local_user).unwrap(); + let inserted_local_user = LocalUser::create(conn, &new_local_user).unwrap(); let token = "nope"; let token_encrypted_ = "ca3704aa0b06f5954c79ee837faa152d84d6b2d42838f0637a15eda8337dbdce"; let inserted_password_reset_request = - PasswordResetRequest::create_token(&conn, inserted_local_user.id, token).unwrap(); + PasswordResetRequest::create_token(conn, inserted_local_user.id, token).unwrap(); let expected_password_reset_request = PasswordResetRequest { id: inserted_password_reset_request.id, @@ -114,8 +117,8 @@ mod tests { published: inserted_password_reset_request.published, }; - let read_password_reset_request = PasswordResetRequest::read_from_token(&conn, token).unwrap(); - let num_deleted = Person::delete(&conn, inserted_person.id).unwrap(); + let read_password_reset_request = PasswordResetRequest::read_from_token(conn, token).unwrap(); + let num_deleted = Person::delete(conn, inserted_person.id).unwrap(); assert_eq!(expected_password_reset_request, read_password_reset_request); assert_eq!(