feat: more coverage

fix: add postgresql db to workflow

fix: apply db migrations

asdfasdfasfd

Author: bitfl0wer

.github/workflows/build-test-coverage.yaml

@@ -21,6 +21,13 @@ jobs:
     timeout-minutes: 30
     steps:
       - uses: actions/checkout@v4
+      - uses: ikalnytskyi/action-setup-postgres@v6
+        with:
+          username: sonata
+          password: sljkdhfghjklsdfghjkldfsghjklsdfghjklsdfg
+          database: sonata
+          port: 5432
+        id: postgres
       - uses: dtolnay/rust-toolchain@nightly
         with:
           components: "llvm-tools-preview"
@@ -32,8 +39,14 @@ jobs:
           tool: cargo-nextest,cargo-binstall
       - name: Install dependencies
         run: |
-          cargo binstall --no-confirm --force --only-signed cargo-llvm-cov
+          cargo binstall --no-confirm --force --only-signed cargo-llvm-cov sqlx-cli
+      - name: Apply migrations
+        env:
+          DATABASE_URL: ${{ steps.postgres.outputs.connection-uri }}
+        run: cargo sqlx migrate run
       - name: Build, Test, Publish Coverage
+        env:
+          DATABASE_URL: ${{ steps.postgres.outputs.connection-uri }}
         run: |
           if [ -n "${{ secrets.COVERALLS_REPO_TOKEN }}" ]; then
             mkdir -p coverage

src/api/admin/invitations.rs

@@ -5,4 +5,5 @@
 use poem::{IntoResponse, handler};
 
 #[handler]
+#[cfg_attr(coverage_nightly, coverage(off))]
 pub(crate) async fn create_invite() -> impl IntoResponse {}

src/api/mod.rs

@@ -19,6 +19,7 @@ pub(super) mod admin;
 pub(crate) mod middlewares;
 
 #[allow(clippy::expect_used)]
+#[cfg_attr(coverage_nightly, coverage(off))]
 /// Build the API [Route]s and start a `tokio::task`, which is a poem [Server] processing incoming
 /// HTTP API requests.
 pub(super) fn start_api(api_config: ApiConfig, db: Database) -> tokio::task::JoinHandle<()> {

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());
+    }
+}