test_utils: Add helpers module with add_team_to_crate()

Author: Turbo87

crates/crates_io_test_utils/src/helpers.rs

@@ -0,0 +1,18 @@
+use crates_io_database::models::{Crate, CrateOwner, Team, User};
+use diesel::prelude::*;
+use diesel_async::AsyncPgConnection;
+
+pub async fn add_team_to_crate(
+    t: &Team,
+    krate: &Crate,
+    u: &User,
+    conn: &mut AsyncPgConnection,
+) -> QueryResult<()> {
+    CrateOwner::builder()
+        .crate_id(krate.id)
+        .team_id(t.id)
+        .created_by(u.id)
+        .build()
+        .insert(conn)
+        .await
+}

crates/crates_io_test_utils/src/lib.rs

@@ -1,2 +1,3 @@
 pub mod builders;
 pub mod github;
+pub mod helpers;

src/tests/mod.rs

@@ -1,4 +1,4 @@
-use crate::models::{Crate, CrateOwner, NewCategory, NewTeam, NewUser, Team, User};
+use crate::models::{NewCategory, NewTeam, NewUser};
 use crate::tests::util::{RequestHelper, TestApp};
 use crate::views::{
     EncodableCategory, EncodableCrate, EncodableKeyword, EncodableOwner, EncodableVersion,
@@ -7,8 +7,6 @@ use crate::views::{
 
 use crate::tests::util::github::next_gh_id;
 use crate::util::gh_token_encryption::GitHubTokenEncryption;
-use diesel::prelude::*;
-use diesel_async::AsyncPgConnection;
 use serde::{Deserialize, Serialize};
 use std::sync::LazyLock;
 
@@ -116,20 +114,7 @@ fn new_team(login: &str) -> NewTeam<'_> {
         .build()
 }
 
-pub async fn add_team_to_crate(
-    t: &Team,
-    krate: &Crate,
-    u: &User,
-    conn: &mut AsyncPgConnection,
-) -> QueryResult<()> {
-    CrateOwner::builder()
-        .crate_id(krate.id)
-        .team_id(t.id)
-        .created_by(u.id)
-        .build()
-        .insert(conn)
-        .await
-}
+pub use crates_io_test_utils::helpers::add_team_to_crate;
 
 fn new_category<'a>(category: &'a str, slug: &'a str, description: &'a str) -> NewCategory<'a> {
     NewCategory {

src/typosquat/database.rs

@@ -170,9 +170,9 @@ impl From<crate::models::Owner> for Owner {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::tests::add_team_to_crate;
     use crate::typosquat::test_util::faker;
     use crates_io_test_db::TestDatabase;
+    use crates_io_test_utils::helpers::add_team_to_crate;
     use thiserror::Error;
 
     #[tokio::test]