From e77cd830c7492c3e337dea01811e4663bbce389f Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Mon, 9 Dec 2024 04:12:32 +0100
Subject: [PATCH] add test_open_db_restart

---
 node_core/src/storage/block_store.rs | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/node_core/src/storage/block_store.rs b/node_core/src/storage/block_store.rs
index 454405f..57d4aa3 100644
--- a/node_core/src/storage/block_store.rs
+++ b/node_core/src/storage/block_store.rs
@@ -70,7 +70,7 @@ mod tests {
 
     #[test]
     fn test_open_db_with_genesis() {
-        let temp_dir: tempfile::TempDir = tempdir().unwrap();
+        let temp_dir = tempdir().unwrap();
         let path = temp_dir.path();
 
         let genesis_block = create_genesis_block();
@@ -83,4 +83,20 @@ mod tests {
         assert_eq!(stored_block.hash, genesis_block.hash);
     }
 
+    #[test]
+    fn test_open_db_restart() {
+        let temp_dir = tempdir().unwrap();
+        let path = temp_dir.path();
+
+        let genesis_block = create_genesis_block();
+        let _ = NodeBlockStore::open_db_with_genesis(path, Some(genesis_block)).unwrap();
+
+        // Restart the database
+        let node_store = NodeBlockStore::open_db_restart(path).unwrap();
+
+        // The block should no longer be available since no genesis block is set on restart
+        let result = node_store.get_block_at_id(0);
+        assert!(result.is_err());
+    }
+
 }