Scheduler::is_run_queued(&Arc<ProcessControlBlock>)

Be able to check if a scheduler is still scheduling a process.

Author: KronicDeth

lumen_runtime/src/run/queues.rs

@@ -24,6 +24,14 @@ pub struct Queues {
 }
 
 impl Queues {
+    #[cfg(test)]
+    pub fn contains(&self, value: &Arc<ProcessControlBlock>) -> bool {
+        self.waiting.contains(value)
+            || self.normal_low.contains(value)
+            || self.high.contains(value)
+            || self.max.contains(value)
+    }
+
     #[cfg(test)]
     pub fn run_queue_len(&self, priority: Priority) -> usize {
         match priority {
@@ -117,6 +125,11 @@ impl Next {
 pub struct Waiting(HashSet<Arc<ProcessControlBlock>>);
 
 impl Waiting {
+    #[cfg(test)]
+    fn contains(&self, value: &Arc<ProcessControlBlock>) -> bool {
+        self.0.contains(value)
+    }
+
     fn get<Q: ?Sized>(&self, value: &Q) -> Option<&Arc<ProcessControlBlock>>
     where
         Arc<ProcessControlBlock>: Borrow<Q>,

lumen_runtime/src/run/queues/delayed.rs

@@ -13,6 +13,13 @@ use crate::run::Run;
 pub struct Delayed(VecDeque<DelayedProcess>);
 
 impl Delayed {
+    #[cfg(test)]
+    pub fn contains(&self, value: &Arc<ProcessControlBlock>) -> bool {
+        self.0
+            .iter()
+            .any(|delayed_process| &delayed_process.arc_process == value)
+    }
+
     pub fn len(&self) -> usize {
         self.0.len()
     }

lumen_runtime/src/run/queues/immediate.rs

@@ -10,6 +10,11 @@ use crate::run::Run;
 pub struct Immediate(VecDeque<Arc<ProcessControlBlock>>);
 
 impl Immediate {
+    #[cfg(test)]
+    pub fn contains(&self, value: &Arc<ProcessControlBlock>) -> bool {
+        self.0.contains(value)
+    }
+
     pub fn len(&self) -> usize {
         self.0.len()
     }

lumen_runtime/src/scheduler.rs

@@ -184,6 +184,11 @@ impl Scheduler {
         self.run_queues.read().run_queue_len(priority)
     }
 
+    #[cfg(test)]
+    pub fn is_run_queued(&self, value: &Arc<ProcessControlBlock>) -> bool {
+        self.run_queues.read().contains(value)
+    }
+
     /// Returns `true` if `arc_process` was run; otherwise, `false`.
     #[must_use]
     pub fn run_through(&self, arc_process: &Arc<ProcessControlBlock>) -> bool {