Use DbConn in model tests

Plume-org · Mar 19, 2023 · 33bd290 · 33bd290
1 parent 85ab539
commit 33bd290
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/plume-models/src/comments.rs b/plume-models/src/comments.rs
@@ -422,6 +422,7 @@ impl CommentTree {
 mod tests {
     use super::*;
     use crate::blogs::Blog;
+    use crate::db_conn::DbConn;
     use crate::inbox::{inbox, tests::fill_database, InboxResult};
     use crate::safe_string::SafeString;
     use crate::tests::{db, format_datetime};

diff --git a/plume-models/src/follows.rs b/plume-models/src/follows.rs
@@ -232,7 +232,9 @@ impl IntoId for Follow {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::{tests::db, users::tests as user_tests, users::tests::fill_database};
+    use crate::{
+        db_conn::DbConn, tests::db, users::tests as user_tests, users::tests::fill_database,
+    };
     use assert_json_diff::assert_json_eq;
     use diesel::Connection;
     use serde_json::{json, to_value};

diff --git a/plume-models/src/posts.rs b/plume-models/src/posts.rs
@@ -1027,6 +1027,7 @@ impl From<PostEvent> for Arc<Post> {
 #[cfg(test)]
 mod tests {
     use super::*;
+    use crate::db_conn::DbConn;
     use crate::inbox::{inbox, tests::fill_database, InboxResult};
     use crate::mentions::{Mention, NewMention};
     use crate::safe_string::SafeString;