Add tests/signer/commands/shutdown.rs

Co-authored-by: Nikos Baxevanis <[email protected]>

Author: BowTiedRadone

testnet/stacks-node/src/tests/signer/commands/shutdown.rs

@@ -0,0 +1,44 @@
+use super::context::{SignerTestContext, SignerTestState};
+use crate::tests::signer::v0::MultipleMinerTest;
+use madhouse::{Command, CommandWrapper};
+use proptest::prelude::{Just, Strategy};
+use std::sync::{Arc, Mutex};
+
+pub struct ShutdownMinersCommand {
+    miners: Arc<Mutex<MultipleMinerTest>>,
+}
+
+impl ShutdownMinersCommand {
+    pub fn new(miners: Arc<Mutex<MultipleMinerTest>>) -> Self {
+        Self { miners }
+    }
+}
+
+impl Command<SignerTestState, SignerTestContext> for ShutdownMinersCommand {
+    fn check(&self, _state: &SignerTestState) -> bool {
+        println!("Checking: Shutting down miners. Result: {:?}", true);
+        true
+    }
+
+    fn apply(&self, _state: &mut SignerTestState) {
+        println!("Applying: Shutting down miners");
+
+        if let Ok(miners_arc) = Arc::try_unwrap(self.miners.clone()) {
+            if let Ok(miners) = miners_arc.into_inner() {
+                miners.shutdown();
+            }
+        }
+    }
+
+    fn label(&self) -> String {
+        "SHUTDOWN_MINERS".to_string()
+    }
+
+    fn build(
+        ctx: Arc<SignerTestContext>,
+    ) -> impl Strategy<Value = CommandWrapper<SignerTestState, SignerTestContext>> {
+        Just(CommandWrapper::new(ShutdownMinersCommand::new(
+            ctx.miners.clone(),
+        )))
+    }
+}