feat: more coverage

Author: bitfl0wer

src/database/api_keys.rs

@@ -12,11 +12,15 @@ use crate::StdError;
 use crate::database::Database;
 use crate::errors::SonataDbError;
 
+/// Constant used to determine how long auto-generated tokens are supposed to be.
+pub const STANDARD_TOKEN_LENGTH: usize = 128;
+
 #[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Clone)]
 pub struct ApiKey {
     token: String,
 }
 
+#[cfg_attr(coverage_nightly, coverage(off))]
 impl Deref for ApiKey {
     type Target = str;
 
@@ -41,13 +45,15 @@ impl ApiKey {
         Ok(Self { token: token.to_owned() })
     }
 
+    #[cfg_attr(coverage_nightly, coverage(off))]
+    /// Getter for the internal token.
     pub fn token(&self) -> &str {
         &self.token
     }
 
-    /// Generates a new, random [ApiKey] which is 128 characters in length.
+    /// Generates a new, random [ApiKey] which is [STANDARD_TOKEN_LENGTH] characters in length.
     pub fn new_random(rng: &mut ThreadRng) -> Self {
-        Self { token: Alphanumeric.sample_string(rng, 128) }
+        Self { token: Alphanumeric.sample_string(rng, STANDARD_TOKEN_LENGTH) }
     }
 }
 
@@ -69,3 +75,30 @@ pub(crate) async fn add_api_key_to_database(
         .map_err(SonataDbError::Sqlx)?;
     Ok(key)
 }
+
+#[cfg(test)]
+mod test {
+    use rand::rng;
+    use sqlx::{Pool, Postgres};
+
+    use super::*;
+
+    #[test]
+    fn token_length() {
+        assert!(ApiKey::new("").is_err());
+        assert!(ApiKey::new("token").is_err());
+        assert!(ApiKey::new(['c'; 256].iter().collect::<String>().as_str()).is_err());
+        assert!(ApiKey::new("transrightsarehumanrights_thefirstpridewasariot").is_ok());
+    }
+
+    #[test]
+    fn auto_gen_token() {
+        assert_eq!(ApiKey::new_random(&mut rng()).len(), STANDARD_TOKEN_LENGTH);
+    }
+
+    #[sqlx::test]
+    async fn insert_key_into_db(db: Pool<Postgres>) {
+        let key = ApiKey::new_random(&mut rng());
+        assert!(add_api_key_to_database(key.token(), &Database { pool: db }).await.is_ok());
+    }
+}