tests

Author: syphar

src/build_queue.rs

@@ -889,9 +889,10 @@ mod tests {
     use crate::db::types::BuildStatus;
     use crate::test::{FakeBuild, TestEnvironment, V1, V2};
     use chrono::Utc;
-
     use std::time::Duration;
 
+    static KRATE: &str = "krate";
+
     #[tokio::test(flavor = "multi_thread")]
     async fn test_rebuild_when_old() -> Result<()> {
         let env = TestEnvironment::with_config(
@@ -1757,4 +1758,48 @@ mod tests {
 
         Ok(())
     }
+
+    #[tokio::test(flavor = "multi_thread")]
+    async fn test_delete_version_from_queue() -> Result<()> {
+        let env = TestEnvironment::new().await?;
+
+        let queue = env.async_build_queue();
+        assert_eq!(queue.pending_count().await?, 0);
+
+        queue.add_crate(KRATE, &V1, 0, None).await?;
+        queue.add_crate(KRATE, &V2, 0, None).await?;
+
+        assert_eq!(queue.pending_count().await?, 2);
+        queue.remove_version_from_queue(KRATE, &V1).await?;
+
+        assert_eq!(queue.pending_count().await?, 1);
+
+        // only v2 remains
+        if let [k] = queue.queued_crates().await?.as_slice() {
+            assert_eq!(k.name, KRATE);
+            assert_eq!(k.version, V2);
+        } else {
+            panic!("expected only one queued crate");
+        }
+
+        Ok(())
+    }
+
+    #[tokio::test(flavor = "multi_thread")]
+    async fn test_delete_crate_from_queue() -> Result<()> {
+        let env = TestEnvironment::new().await?;
+
+        let queue = env.async_build_queue();
+        assert_eq!(queue.pending_count().await?, 0);
+
+        queue.add_crate(KRATE, &V1, 0, None).await?;
+        queue.add_crate(KRATE, &V2, 0, None).await?;
+
+        assert_eq!(queue.pending_count().await?, 2);
+        queue.remove_crate_from_queue(KRATE).await?;
+
+        assert_eq!(queue.pending_count().await?, 0);
+
+        Ok(())
+    }
 }