tests::util: Extract db_new_scoped_token() function

Author: Turbo87

src/tests/util.rs

@@ -29,6 +29,7 @@ use conduit::{BoxError, Handler, Method};
 use conduit_cookie::SessionMiddleware;
 use conduit_test::MockRequest;
 
+use cargo_registry::models::token::{CrateScope, EndpointScope};
 use conduit::header;
 use cookie::Cookie;
 use std::collections::HashMap;
@@ -263,9 +264,22 @@ impl MockCookieUser {
     ///
     /// This method updates the database directly
     pub fn db_new_token(&self, name: &str) -> MockTokenUser {
-        let token = self
-            .app
-            .db(|conn| ApiToken::insert(conn, self.user.id, name).unwrap());
+        self.db_new_scoped_token(name, None, None)
+    }
+
+    /// Creates a scoped token and wraps it in a helper struct
+    ///
+    /// This method updates the database directly
+    pub fn db_new_scoped_token(
+        &self,
+        name: &str,
+        crate_scopes: Option<Vec<CrateScope>>,
+        endpoint_scopes: Option<Vec<EndpointScope>>,
+    ) -> MockTokenUser {
+        let token = self.app.db(|conn| {
+            ApiToken::insert_with_scopes(conn, self.user.id, name, crate_scopes, endpoint_scopes)
+                .unwrap()
+        });
         MockTokenUser {
             app: self.app.clone(),
             token,