new(tests): TSTORE: ensure transient storage is cleared after transactions (#798)

* new(tests): TSTORE: ensure transient storage is cleared after transactions

* address review

* tests: ensure eof/legacy gets properly switched

* chore: fix mypy tox errors

* address comment

---------

Co-authored-by: spencer-tb <[email protected]>

Author: jochem-brouwer

tests/cancun/eip1153_tstore/test_tstorage_clear_after_tx.py

@@ -0,0 +1,109 @@
+"""
+Ethereum Transient Storage EIP Tests
+https://eips.ethereum.org/EIPS/eip-1153
+"""
+
+from typing import Optional
+
+import pytest
+
+from ethereum_test_tools import (
+    Account,
+    Alloc,
+    Block,
+    BlockchainTestFiller,
+    Environment,
+    EVMCodeType,
+    Initcode,
+    Transaction,
+)
+from ethereum_test_tools.eof.v1 import Container
+from ethereum_test_tools.vm.opcode import Opcodes as Op
+
+from .spec import ref_spec_1153
+
+REFERENCE_SPEC_GIT_PATH = ref_spec_1153.git_path
+REFERENCE_SPEC_VERSION = ref_spec_1153.version
+
+
+@pytest.mark.valid_from("Cancun")
+@pytest.mark.with_all_evm_code_types
+def test_tstore_clear_after_deployment_tx(
+    blockchain_test: BlockchainTestFiller,
+    pre: Alloc,
+    evm_code_type: EVMCodeType,
+):
+    """
+    This test first creates a contract, which TSTOREs a value 1 in slot 1.
+    After creating the contract, a new tx will call this contract, storing TLOAD(1) into slot 1.
+    The transient storage should be cleared after creating the contract (at tx-level), so
+    the storage should stay empty.
+    """
+    env = Environment()
+
+    init_code = Op.TSTORE(1, 1)
+    deploy_code = Op.SSTORE(1, Op.TLOAD(1))
+
+    code: Optional[Container | Initcode] = None
+    if evm_code_type == EVMCodeType.EOF_V1:
+        code = Container.Init(
+            deploy_container=Container.Code(deploy_code), initcode_prefix=init_code
+        )
+    else:
+        code = Initcode(deploy_code=deploy_code, initcode_prefix=init_code)
+
+    sender = pre.fund_eoa()
+
+    deployment_tx = Transaction(
+        gas_limit=100000,
+        data=code,
+        to=None,
+        sender=sender,
+    )
+
+    address = deployment_tx.created_contract
+
+    invoke_contract_tx = Transaction(gas_limit=100000, to=address, sender=sender)
+
+    txs = [deployment_tx, invoke_contract_tx]
+
+    post = {
+        address: Account(storage={0x01: 0x00}),
+    }
+
+    blockchain_test(genesis_environment=env, pre=pre, post=post, blocks=[Block(txs=txs)])
+
+
+@pytest.mark.valid_from("Cancun")
+@pytest.mark.with_all_evm_code_types
+def test_tstore_clear_after_tx(
+    blockchain_test: BlockchainTestFiller,
+    pre: Alloc,
+):
+    """
+    This test first SSTOREs the TLOAD value of key 1 in slot 1. Then, it TSTOREs 1 in slot 1.
+    The second tx will re-call the contract. The storage should stay empty,
+    because the transient storage is cleared after the transaction.
+    """
+    env = Environment()
+
+    code = Op.SSTORE(1, Op.TLOAD(1)) + Op.TSTORE(1, 1)
+    account = pre.deploy_contract(code)
+
+    sender = pre.fund_eoa()
+
+    poke_tstore_tx = Transaction(
+        gas_limit=100000,
+        to=account,
+        sender=sender,
+    )
+
+    re_poke_tstore_tx = Transaction(gas_limit=100000, to=account, sender=sender)
+
+    txs = [poke_tstore_tx, re_poke_tstore_tx]
+
+    post = {
+        account: Account(storage={0x01: 0x00}),
+    }
+
+    blockchain_test(genesis_environment=env, pre=pre, post=post, blocks=[Block(txs=txs)])