From 373421a1d1d13f714b8ceb369c2e337998195edc Mon Sep 17 00:00:00 2001
From: Daniel Lubarov <daniel@lubarov.com>
Date: Thu, 16 Mar 2023 14:11:40 -0700
Subject: [PATCH] Fix tests - need to supply empty code

---
 evm/tests/basic_smart_contract.rs | 2 ++
 evm/tests/empty_txn_list.rs       | 6 +++++-
 evm/tests/simple_transfer.rs      | 5 ++++-
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/evm/tests/basic_smart_contract.rs b/evm/tests/basic_smart_contract.rs
index ed326bfd..4809ea93 100644
--- a/evm/tests/basic_smart_contract.rs
+++ b/evm/tests/basic_smart_contract.rs
@@ -95,7 +95,9 @@ fn test_basic_smart_contract() -> anyhow::Result<()> {
     };
 
     let mut contract_code = HashMap::new();
+    contract_code.insert(keccak(vec![]), vec![]);
     contract_code.insert(code_hash, code.to_vec());
+
     let inputs = GenerationInputs {
         signed_txns: vec![txn.to_vec()],
         tries: tries_before,
diff --git a/evm/tests/empty_txn_list.rs b/evm/tests/empty_txn_list.rs
index 19330417..a0148e74 100644
--- a/evm/tests/empty_txn_list.rs
+++ b/evm/tests/empty_txn_list.rs
@@ -3,6 +3,7 @@ use std::time::Duration;
 
 use env_logger::{try_init_from_env, Env, DEFAULT_FILTER_ENV};
 use eth_trie_utils::partial_trie::PartialTrie;
+use keccak_hash::keccak;
 use plonky2::field::goldilocks_field::GoldilocksField;
 use plonky2::plonk::config::PoseidonGoldilocksConfig;
 use plonky2::util::timing::TimingTree;
@@ -38,6 +39,9 @@ fn test_empty_txn_list() -> anyhow::Result<()> {
     let txns_trie_root = transactions_trie.calc_hash();
     let receipts_trie_root = receipts_trie.calc_hash();
 
+    let mut contract_code = HashMap::new();
+    contract_code.insert(keccak(vec![]), vec![]);
+
     let inputs = GenerationInputs {
         signed_txns: vec![],
         tries: TrieInputs {
@@ -46,7 +50,7 @@ fn test_empty_txn_list() -> anyhow::Result<()> {
             receipts_trie,
             storage_tries,
         },
-        contract_code: HashMap::new(),
+        contract_code,
         block_metadata,
         addresses: vec![],
     };
diff --git a/evm/tests/simple_transfer.rs b/evm/tests/simple_transfer.rs
index d6e1a65f..ff2bed31 100644
--- a/evm/tests/simple_transfer.rs
+++ b/evm/tests/simple_transfer.rs
@@ -70,10 +70,13 @@ fn test_simple_transfer() -> anyhow::Result<()> {
         ..BlockMetadata::default()
     };
 
+    let mut contract_code = HashMap::new();
+    contract_code.insert(keccak(vec![]), vec![]);
+
     let inputs = GenerationInputs {
         signed_txns: vec![txn.to_vec()],
         tries: tries_before,
-        contract_code: HashMap::new(),
+        contract_code,
         block_metadata,
         addresses: vec![],
     };