add v0 table enum

crates/katana/storage/db/src/mdbx/mod.rs

@@ -111,9 +111,28 @@ impl DbEnv {
 
 #[cfg(any(test, feature = "test-utils"))]
 pub mod test_utils {
-    use std::path::Path;
+    use super::*;
+    use crate::tables::v0::Tables as V0Tables;
+
+    impl DbEnv {
+        /// Creates all the defined tables in [`Tables`], if necessary.
+        pub fn create_v0_tables(&self) -> Result<(), DatabaseError> {
+            let tx = self.0.begin_rw_txn().map_err(DatabaseError::CreateRWTx)?;
+
+            for table in V0Tables::ALL {
+                let flags = match table.table_type() {
+                    TableType::Table => DatabaseFlags::default(),
+                    TableType::DupSort => DatabaseFlags::DUP_SORT,
+                };
+
+                tx.create_db(Some(table.name()), flags).map_err(DatabaseError::CreateTable)?;
+            }
 
-    use super::{DbEnv, DbEnvKind};
+            tx.commit().map_err(DatabaseError::Commit)?;
+
+            Ok(())
+        }
+    }
 
     const ERROR_DB_CREATION: &str = "Not able to create the mdbx file.";
 

crates/katana/storage/db/src/tables/v0.rs

@@ -1,13 +1,39 @@
 use katana_primitives::block::BlockNumber;
 use katana_primitives::contract::{ContractAddress, StorageKey};
 
-use super::{DupSort, Table};
+use super::*;
 use crate::codecs::{Compress, Decode, Decompress, Encode};
 use crate::error::CodecError;
 use crate::models::contract::{ContractClassChange, ContractNonceChange};
 use crate::models::storage::ContractStorageEntry;
 use crate::models::storage::ContractStorageKey;
-use crate::{dupsort, tables};
+use crate::{define_tables_enum, dupsort, tables};
+
+// TODO(kariy): can we somehow define this without repeating existing tables, and only add the older ones?
+define_tables_enum! {[
+    (Headers, TableType::Table),
+    (BlockHashes, TableType::Table),
+    (BlockNumbers, TableType::Table),
+    (BlockBodyIndices, TableType::Table),
+    (BlockStatusses, TableType::Table),
+    (TxNumbers, TableType::Table),
+    (TxBlocks, TableType::Table),
+    (TxHashes, TableType::Table),
+    (Transactions, TableType::Table),
+    (Receipts, TableType::Table),
+    (CompiledClassHashes, TableType::Table),
+    (CompiledClasses, TableType::Table),
+    (SierraClasses, TableType::Table),
+    (ContractInfo, TableType::Table),
+    (ContractStorage, TableType::DupSort),
+    (ClassDeclarationBlock, TableType::Table),
+    (ClassDeclarations, TableType::DupSort),
+    (ContractInfoChangeSet, TableType::Table),
+    (NonceChanges, TableType::DupSort),
+    (ContractClassChanges, TableType::DupSort),
+    (StorageChanges, TableType::DupSort),
+    (StorageChangeSet, TableType::DupSort)
+]}
 
 tables! {
     /// Contract nonce changes by block.