chore(mpt): account conversion tests

anton-rs · Oct 7, 2024 · d1cb7a4 · d1cb7a4
1 parent 5cdb416
commit d1cb7a4
Showing 1 changed file with 42 additions and 0 deletions.
diff --git a/crates/mpt/src/db/account.rs b/crates/mpt/src/db/account.rs
@@ -38,3 +38,45 @@ impl From<(AccountInfo, B256)> for TrieAccount {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use alloy_primitives::uint;
+
+    #[test]
+    fn test_trie_account_from_account() {
+        let account = Account {
+            info: AccountInfo {
+                nonce: 1,
+                balance: uint!(2_U256),
+                code_hash: B256::default(),
+                code: Default::default(),
+            },
+            status: Default::default(),
+            storage: Default::default(),
+        };
+        let storage_root = B256::default();
+        let trie_account = TrieAccount::from((account, storage_root));
+        assert_eq!(trie_account.nonce, 1);
+        assert_eq!(trie_account.balance, uint!(2_U256));
+        assert_eq!(trie_account.storage_root, B256::default());
+        assert_eq!(trie_account.code_hash, B256::default());
+    }
+
+    #[test]
+    fn test_trie_account_from_account_info() {
+        let account_info = AccountInfo {
+            nonce: 1,
+            balance: uint!(2_U256),
+            code_hash: B256::default(),
+            code: Default::default(),
+        };
+        let storage_root = B256::default();
+        let trie_account = TrieAccount::from((account_info, storage_root));
+        assert_eq!(trie_account.nonce, 1);
+        assert_eq!(trie_account.balance, uint!(2_U256));
+        assert_eq!(trie_account.storage_root, B256::default());
+        assert_eq!(trie_account.code_hash, B256::default());
+    }
+}