Joseph T. Lyons
60e9ab8f93
Trying to delete a user record from our admin panel throws the following error: `update or delete on table "users" violates foreign key constraint "access_tokens_user_id_fkey" on table "access_tokens" Detail: Key (id)=(....) is still referenced from table "access_tokens".` We need to add a cascade delete to the `access_tokens` table. Release Notes: - N/A
97 lines
2.5 KiB
Rust
97 lines
2.5 KiB
Rust
use chrono::Utc;
|
|
|
|
use crate::{
|
|
db::{Database, NewUserParams},
|
|
test_both_dbs,
|
|
};
|
|
use std::sync::Arc;
|
|
|
|
test_both_dbs!(
|
|
test_accepted_tos,
|
|
test_accepted_tos_postgres,
|
|
test_accepted_tos_sqlite
|
|
);
|
|
|
|
async fn test_accepted_tos(db: &Arc<Database>) {
|
|
let user_id = db
|
|
.create_user(
|
|
"user1@example.com",
|
|
None,
|
|
false,
|
|
NewUserParams {
|
|
github_login: "user1".to_string(),
|
|
github_user_id: 1,
|
|
},
|
|
)
|
|
.await
|
|
.unwrap()
|
|
.user_id;
|
|
|
|
let user = db.get_user_by_id(user_id).await.unwrap().unwrap();
|
|
assert!(user.accepted_tos_at.is_none());
|
|
|
|
let accepted_tos_at = Utc::now().naive_utc();
|
|
db.set_user_accepted_tos_at(user_id, Some(accepted_tos_at))
|
|
.await
|
|
.unwrap();
|
|
|
|
let user = db.get_user_by_id(user_id).await.unwrap().unwrap();
|
|
assert!(user.accepted_tos_at.is_some());
|
|
assert_eq!(user.accepted_tos_at, Some(accepted_tos_at));
|
|
|
|
db.set_user_accepted_tos_at(user_id, None).await.unwrap();
|
|
|
|
let user = db.get_user_by_id(user_id).await.unwrap().unwrap();
|
|
assert!(user.accepted_tos_at.is_none());
|
|
}
|
|
|
|
test_both_dbs!(
|
|
test_destroy_user_cascade_deletes_access_tokens,
|
|
test_destroy_user_cascade_deletes_access_tokens_postgres,
|
|
test_destroy_user_cascade_deletes_access_tokens_sqlite
|
|
);
|
|
|
|
async fn test_destroy_user_cascade_deletes_access_tokens(db: &Arc<Database>) {
|
|
let user_id = db
|
|
.create_user(
|
|
"user1@example.com",
|
|
Some("user1"),
|
|
false,
|
|
NewUserParams {
|
|
github_login: "user1".to_string(),
|
|
github_user_id: 12345,
|
|
},
|
|
)
|
|
.await
|
|
.unwrap()
|
|
.user_id;
|
|
|
|
let user = db.get_user_by_id(user_id).await.unwrap();
|
|
assert!(user.is_some());
|
|
|
|
let token_1_id = db
|
|
.create_access_token(user_id, None, "token-1", 10)
|
|
.await
|
|
.unwrap();
|
|
|
|
let token_2_id = db
|
|
.create_access_token(user_id, None, "token-2", 10)
|
|
.await
|
|
.unwrap();
|
|
|
|
let token_1 = db.get_access_token(token_1_id).await;
|
|
let token_2 = db.get_access_token(token_2_id).await;
|
|
assert!(token_1.is_ok());
|
|
assert!(token_2.is_ok());
|
|
|
|
db.destroy_user(user_id).await.unwrap();
|
|
|
|
let user = db.get_user_by_id(user_id).await.unwrap();
|
|
assert!(user.is_none());
|
|
|
|
let token_1 = db.get_access_token(token_1_id).await;
|
|
let token_2 = db.get_access_token(token_2_id).await;
|
|
assert!(token_1.is_err());
|
|
assert!(token_2.is_err());
|
|
}
|