Add test for abort on tag

0xSpaceShard · Jul 9, 2024 · 448b013 · 448b013
1 parent 4e34912
commit 448b013
Showing 1 changed file with 30 additions and 2 deletions.
diff --git a/crates/starknet-devnet/tests/test_abort_blocks.rs b/crates/starknet-devnet/tests/test_abort_blocks.rs
@@ -3,7 +3,7 @@ pub mod common;
 mod abort_blocks_tests {
     use serde_json::json;
     use server::api::json_rpc::error::ApiError;
-    use starknet_rs_core::types::{BlockId, FieldElement};
+    use starknet_rs_core::types::{BlockId, BlockTag, FieldElement};
     use starknet_types::rpc::transaction_receipt::FeeUnit;
 
     use crate::common::background_devnet::BackgroundDevnet;
@@ -97,7 +97,7 @@ mod abort_blocks_tests {
     }
 
     #[tokio::test]
-    async fn abort_latest_block() {
+    async fn abort_latest_block_with_hash() {
         let devnet =
             BackgroundDevnet::spawn_with_additional_args(&["--state-archive-capacity", "full"])
                 .await
@@ -296,4 +296,32 @@ mod abort_blocks_tests {
 
         abort_blocks_error(&fork_devnet, &BlockId::Hash(fork_block_hash)).await;
     }
+
+    #[tokio::test]
+    async fn abort_latest_blocks() {
+        let devnet =
+            BackgroundDevnet::spawn_with_additional_args(&["--state-archive-capacity", "full"])
+                .await
+                .expect("Could not start Devnet");
+
+        for _ in 0..3 {
+            devnet.create_block().await.unwrap();
+        }
+        for _ in 0..3 {
+            abort_blocks(&devnet, &BlockId::Tag(BlockTag::Latest)).await;
+        }
+        abort_blocks_error(&devnet, &BlockId::Tag(BlockTag::Latest)).await; // Rolled back to genesis block, should not be possible to abort
+    }
+    #[tokio::test]
+    async fn abort_pending_block() {
+        let devnet =
+            BackgroundDevnet::spawn_with_additional_args(&["--state-archive-capacity", "full"])
+                .await
+                .expect("Could not start Devnet");
+
+        for _ in 0..3 {
+            devnet.create_block().await.unwrap();
+        }
+        abort_blocks_error(&devnet, &BlockId::Tag(BlockTag::Pending)).await; // Pending block is genesis block so not possible to abort
+    }
 }