mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-06-01 00:28:05 +00:00
600ae662a5
Since DB tests execute diesel migrations automatically, concurrent execution causes flaky failures from simultaneous migrations. This can be worked around using `cargo test --workspace -- --test-threads=1`, which is what the CI config does, but this is not intuitive for newcomer developers and unnecessarily slows down the test suite for the majority of tests which are safe to run concurrently. This fixes this issue by integrating with the small test crate `serial_test` and using it to explicitly mark DB tests to run sequentially while allowing all other tests to run in parallel. Additionally, this greatly improves the speed of `cargo test` by disabling doc-tests in all crates, since these are aren't currently used and cargo's doc-test pass, even when no doc-tests exist, has significant overhead. On my machine, this change significantly improves test suite times by about 85%, making it much more practical to develop with tools like `cargo watch` auto-running tests.
259 lines
7.6 KiB
Rust
259 lines
7.6 KiB
Rust
use diesel::{result::Error, *};
|
|
use lemmy_db_schema::schema::user_aggregates;
|
|
use serde::Serialize;
|
|
|
|
#[derive(Queryable, Associations, Identifiable, PartialEq, Debug, Serialize, Clone)]
|
|
#[table_name = "user_aggregates"]
|
|
pub struct UserAggregates {
|
|
pub id: i32,
|
|
pub user_id: i32,
|
|
pub post_count: i64,
|
|
pub post_score: i64,
|
|
pub comment_count: i64,
|
|
pub comment_score: i64,
|
|
}
|
|
|
|
impl UserAggregates {
|
|
pub fn read(conn: &PgConnection, user_id: i32) -> Result<Self, Error> {
|
|
user_aggregates::table
|
|
.filter(user_aggregates::user_id.eq(user_id))
|
|
.first::<Self>(conn)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use crate::{
|
|
aggregates::user_aggregates::UserAggregates,
|
|
establish_unpooled_connection,
|
|
Crud,
|
|
Likeable,
|
|
ListingType,
|
|
SortType,
|
|
};
|
|
use lemmy_db_schema::source::{
|
|
comment::{Comment, CommentForm, CommentLike, CommentLikeForm},
|
|
community::{Community, CommunityForm},
|
|
post::{Post, PostForm, PostLike, PostLikeForm},
|
|
user::{UserForm, User_},
|
|
};
|
|
use serial_test::serial;
|
|
|
|
#[test]
|
|
#[serial]
|
|
fn test_crud() {
|
|
let conn = establish_unpooled_connection();
|
|
|
|
let new_user = UserForm {
|
|
name: "thommy_user_agg".into(),
|
|
preferred_username: None,
|
|
password_encrypted: "nope".into(),
|
|
email: None,
|
|
matrix_user_id: None,
|
|
avatar: None,
|
|
banner: None,
|
|
admin: false,
|
|
banned: Some(false),
|
|
published: None,
|
|
updated: None,
|
|
show_nsfw: false,
|
|
theme: "browser".into(),
|
|
default_sort_type: SortType::Hot as i16,
|
|
default_listing_type: ListingType::Subscribed as i16,
|
|
lang: "browser".into(),
|
|
show_avatars: true,
|
|
send_notifications_to_email: false,
|
|
actor_id: None,
|
|
bio: None,
|
|
local: true,
|
|
private_key: None,
|
|
public_key: None,
|
|
last_refreshed_at: None,
|
|
inbox_url: None,
|
|
shared_inbox_url: None,
|
|
};
|
|
|
|
let inserted_user = User_::create(&conn, &new_user).unwrap();
|
|
|
|
let another_user = UserForm {
|
|
name: "jerry_user_agg".into(),
|
|
preferred_username: None,
|
|
password_encrypted: "nope".into(),
|
|
email: None,
|
|
matrix_user_id: None,
|
|
avatar: None,
|
|
banner: None,
|
|
admin: false,
|
|
banned: Some(false),
|
|
published: None,
|
|
updated: None,
|
|
show_nsfw: false,
|
|
theme: "browser".into(),
|
|
default_sort_type: SortType::Hot as i16,
|
|
default_listing_type: ListingType::Subscribed as i16,
|
|
lang: "browser".into(),
|
|
show_avatars: true,
|
|
send_notifications_to_email: false,
|
|
actor_id: None,
|
|
bio: None,
|
|
local: true,
|
|
private_key: None,
|
|
public_key: None,
|
|
last_refreshed_at: None,
|
|
inbox_url: None,
|
|
shared_inbox_url: None,
|
|
};
|
|
|
|
let another_inserted_user = User_::create(&conn, &another_user).unwrap();
|
|
|
|
let new_community = CommunityForm {
|
|
name: "TIL_site_agg".into(),
|
|
creator_id: inserted_user.id,
|
|
title: "nada".to_owned(),
|
|
description: None,
|
|
nsfw: false,
|
|
removed: None,
|
|
deleted: None,
|
|
updated: None,
|
|
actor_id: None,
|
|
local: true,
|
|
private_key: None,
|
|
public_key: None,
|
|
last_refreshed_at: None,
|
|
published: None,
|
|
icon: None,
|
|
banner: None,
|
|
followers_url: None,
|
|
inbox_url: None,
|
|
shared_inbox_url: None,
|
|
};
|
|
|
|
let inserted_community = Community::create(&conn, &new_community).unwrap();
|
|
|
|
let new_post = PostForm {
|
|
name: "A test post".into(),
|
|
url: None,
|
|
body: None,
|
|
creator_id: inserted_user.id,
|
|
community_id: inserted_community.id,
|
|
removed: None,
|
|
deleted: None,
|
|
locked: None,
|
|
stickied: None,
|
|
nsfw: false,
|
|
updated: None,
|
|
embed_title: None,
|
|
embed_description: None,
|
|
embed_html: None,
|
|
thumbnail_url: None,
|
|
ap_id: None,
|
|
local: true,
|
|
published: None,
|
|
};
|
|
|
|
let inserted_post = Post::create(&conn, &new_post).unwrap();
|
|
|
|
let post_like = PostLikeForm {
|
|
post_id: inserted_post.id,
|
|
user_id: inserted_user.id,
|
|
score: 1,
|
|
};
|
|
|
|
let _inserted_post_like = PostLike::like(&conn, &post_like).unwrap();
|
|
|
|
let comment_form = CommentForm {
|
|
content: "A test comment".into(),
|
|
creator_id: inserted_user.id,
|
|
post_id: inserted_post.id,
|
|
removed: None,
|
|
deleted: None,
|
|
read: None,
|
|
parent_id: None,
|
|
published: None,
|
|
updated: None,
|
|
ap_id: None,
|
|
local: true,
|
|
};
|
|
|
|
let inserted_comment = Comment::create(&conn, &comment_form).unwrap();
|
|
|
|
let mut comment_like = CommentLikeForm {
|
|
comment_id: inserted_comment.id,
|
|
user_id: inserted_user.id,
|
|
post_id: inserted_post.id,
|
|
score: 1,
|
|
};
|
|
|
|
let _inserted_comment_like = CommentLike::like(&conn, &comment_like).unwrap();
|
|
|
|
let mut child_comment_form = CommentForm {
|
|
content: "A test comment".into(),
|
|
creator_id: inserted_user.id,
|
|
post_id: inserted_post.id,
|
|
removed: None,
|
|
deleted: None,
|
|
read: None,
|
|
parent_id: Some(inserted_comment.id),
|
|
published: None,
|
|
updated: None,
|
|
ap_id: None,
|
|
local: true,
|
|
};
|
|
|
|
let inserted_child_comment = Comment::create(&conn, &child_comment_form).unwrap();
|
|
|
|
let child_comment_like = CommentLikeForm {
|
|
comment_id: inserted_child_comment.id,
|
|
user_id: another_inserted_user.id,
|
|
post_id: inserted_post.id,
|
|
score: 1,
|
|
};
|
|
|
|
let _inserted_child_comment_like = CommentLike::like(&conn, &child_comment_like).unwrap();
|
|
|
|
let user_aggregates_before_delete = UserAggregates::read(&conn, inserted_user.id).unwrap();
|
|
|
|
assert_eq!(1, user_aggregates_before_delete.post_count);
|
|
assert_eq!(1, user_aggregates_before_delete.post_score);
|
|
assert_eq!(2, user_aggregates_before_delete.comment_count);
|
|
assert_eq!(2, user_aggregates_before_delete.comment_score);
|
|
|
|
// Remove a post like
|
|
PostLike::remove(&conn, inserted_user.id, inserted_post.id).unwrap();
|
|
let after_post_like_remove = UserAggregates::read(&conn, inserted_user.id).unwrap();
|
|
assert_eq!(0, after_post_like_remove.post_score);
|
|
|
|
// Remove a parent comment (the scores should also be removed)
|
|
Comment::delete(&conn, inserted_comment.id).unwrap();
|
|
let after_parent_comment_delete = UserAggregates::read(&conn, inserted_user.id).unwrap();
|
|
assert_eq!(0, after_parent_comment_delete.comment_count);
|
|
assert_eq!(0, after_parent_comment_delete.comment_score);
|
|
|
|
// Add in the two comments again, then delete the post.
|
|
let new_parent_comment = Comment::create(&conn, &comment_form).unwrap();
|
|
child_comment_form.parent_id = Some(new_parent_comment.id);
|
|
Comment::create(&conn, &child_comment_form).unwrap();
|
|
comment_like.comment_id = new_parent_comment.id;
|
|
CommentLike::like(&conn, &comment_like).unwrap();
|
|
let after_comment_add = UserAggregates::read(&conn, inserted_user.id).unwrap();
|
|
assert_eq!(2, after_comment_add.comment_count);
|
|
assert_eq!(1, after_comment_add.comment_score);
|
|
|
|
Post::delete(&conn, inserted_post.id).unwrap();
|
|
let after_post_delete = UserAggregates::read(&conn, inserted_user.id).unwrap();
|
|
assert_eq!(0, after_post_delete.comment_score);
|
|
assert_eq!(0, after_post_delete.comment_count);
|
|
assert_eq!(0, after_post_delete.post_score);
|
|
assert_eq!(0, after_post_delete.post_count);
|
|
|
|
// This should delete all the associated rows, and fire triggers
|
|
let user_num_deleted = User_::delete(&conn, inserted_user.id).unwrap();
|
|
assert_eq!(1, user_num_deleted);
|
|
User_::delete(&conn, another_inserted_user.id).unwrap();
|
|
|
|
// Should be none found
|
|
let after_delete = UserAggregates::read(&conn, inserted_user.id);
|
|
assert!(after_delete.is_err());
|
|
}
|
|
}
|