lemmy_db_schema 1.0.0-beta.2

A link aggregator for the fediverse
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use crate::{
  newtypes::LocalUserId,
  source::password_reset_request::{PasswordResetRequest, PasswordResetRequestForm},
};
use diesel::{
  ExpressionMethods,
  IntoSql,
  delete,
  dsl::{IntervalDsl, insert_into, now},
  sql_types::Timestamptz,
};
use diesel_async::RunQueryDsl;
use lemmy_db_schema_file::schema::password_reset_request;
use lemmy_diesel_utils::connection::{DbPool, get_conn};
use lemmy_utils::error::{LemmyErrorExt, LemmyErrorType, LemmyResult};

impl PasswordResetRequest {
  pub async fn create(
    pool: &mut DbPool<'_>,
    local_user_id: LocalUserId,
    token_: String,
  ) -> LemmyResult<PasswordResetRequest> {
    let form = PasswordResetRequestForm {
      local_user_id,
      token: token_.into(),
    };
    let conn = &mut get_conn(pool).await?;
    insert_into(password_reset_request::table)
      .values(form)
      .get_result::<Self>(conn)
      .await
      .with_lemmy_type(LemmyErrorType::CouldntCreate)
  }

  pub async fn read_and_delete(pool: &mut DbPool<'_>, token_: &str) -> LemmyResult<Self> {
    let conn = &mut get_conn(pool).await?;
    delete(password_reset_request::table)
      .filter(password_reset_request::token.eq(token_))
      .filter(password_reset_request::published_at.gt(now.into_sql::<Timestamptz>() - 1.days()))
      .get_result(conn)
      .await
      .with_lemmy_type(LemmyErrorType::Deleted)
  }
}

#[cfg(test)]
mod tests {

  use crate::source::{
    instance::Instance,
    local_user::{LocalUser, LocalUserInsertForm},
    password_reset_request::PasswordResetRequest,
    person::{Person, PersonInsertForm},
  };
  use lemmy_diesel_utils::{connection::build_db_pool_for_tests, traits::Crud};
  use lemmy_utils::error::LemmyResult;
  use pretty_assertions::assert_eq;
  use serial_test::serial;

  #[tokio::test]
  #[serial]
  async fn test_password_reset() -> LemmyResult<()> {
    let pool = &build_db_pool_for_tests();
    let pool = &mut pool.into();

    // Setup
    let inserted_instance = Instance::read_or_create(pool, "my_domain.tld").await?;
    let new_person = PersonInsertForm::test_form(inserted_instance.id, "thommy prw");
    let inserted_person = Person::create(pool, &new_person).await?;
    let new_local_user = LocalUserInsertForm::test_form(inserted_person.id);
    let inserted_local_user = LocalUser::create(pool, &new_local_user, vec![]).await?;

    // Create password reset token
    let token = "nope";
    let inserted_password_reset_request =
      PasswordResetRequest::create(pool, inserted_local_user.id, token.to_string()).await?;

    // Read it and verify
    let read_password_reset_request = PasswordResetRequest::read_and_delete(pool, token).await?;
    assert_eq!(
      inserted_password_reset_request.id,
      read_password_reset_request.id
    );
    assert_eq!(
      inserted_password_reset_request.local_user_id,
      read_password_reset_request.local_user_id
    );
    assert_eq!(
      inserted_password_reset_request.token,
      read_password_reset_request.token
    );
    assert_eq!(
      inserted_password_reset_request.published_at,
      read_password_reset_request.published_at
    );

    // Cannot reuse same token again
    let read_password_reset_request = PasswordResetRequest::read_and_delete(pool, token).await;
    assert!(read_password_reset_request.is_err());

    // Cleanup
    let num_deleted = Person::delete(pool, inserted_person.id).await?;
    assert_eq!(1, num_deleted);

    Instance::delete(pool, inserted_instance.id).await?;
    Ok(())
  }
}