new(tests): EOF - EIP-7620: context variables

tests/prague/eip7692_eof_v1/eip7620_eof_create/test_eofcreate.py

@@ -4,6 +4,7 @@
 
 import pytest
 
+from ethereum_test_base_types.conversions import to_fixed_size_bytes
 from ethereum_test_exceptions import EOFException
 from ethereum_test_specs import EOFTestFiller
 from ethereum_test_tools import (
@@ -16,6 +17,7 @@
 )
 from ethereum_test_tools.eof.v1 import Container, Section
 from ethereum_test_tools.vm.opcode import Opcodes as Op
+from ethereum_test_vm.bytecode import Bytecode
 
 from .. import EOF_FORK_NAME
 from .helpers import (
@@ -590,3 +592,82 @@ def test_eofcreate_invalid_index(
         ),
         expect_exception=EOFException.INVALID_CONTAINER_SECTION_INDEX,
     )
+
+
+@pytest.mark.parametrize(
+    ["destination_code", "expected_result"],
+    [
+        pytest.param(Op.ADDRESS, "destination"),
+        pytest.param(Op.CALLER, "caller"),
+        pytest.param(Op.CALLVALUE, "eofcreate_value"),
+        pytest.param(Op.ORIGIN, "sender"),
+    ],
+)
+def test_eofcreate_context(
+    state_test: StateTestFiller,
+    pre: Alloc,
+    destination_code: Bytecode,
+    expected_result: str,
+):
+    """Test EOFCREATE's initcode context instructions"""
+    env = Environment()
+    sender = pre.fund_eoa()
+    value = 0x1123
+    eofcreate_value = 0x13
+
+    initcode = Container(
+        sections=[
+            Section.Code(
+                Op.SSTORE(slot_call_result, destination_code) + Op.RETURNCONTRACT[0](0, 0)
+            ),
+            Section.Container(smallest_runtime_subcontainer),
+        ]
+    )
+
+    caller_contract = Container(
+        sections=[
+            Section.Code(
+                Op.SSTORE(slot_code_worked, value_code_worked)
+                + Op.EOFCREATE[0](eofcreate_value, 0, 0, 0)
+                + Op.STOP
+            ),
+            Section.Container(initcode),
+        ]
+    )
+    calling_contract_address = pre.deploy_contract(caller_contract)
+
+    destination_contract_address = compute_eofcreate_address(calling_contract_address, 0, initcode)
+
+    tx = Transaction(
+        sender=sender, to=calling_contract_address, gas_limit=1000000, data="", value=value
+    )
+
+    if expected_result == "destination":
+        expected_bytes: bytes = destination_contract_address
+    elif expected_result == "caller":
+        expected_bytes = calling_contract_address
+    elif expected_result == "sender":
+        expected_bytes = sender
+    elif expected_result == "eofcreate_value":
+        expected_bytes = to_fixed_size_bytes(eofcreate_value, 32)
+    else:
+        raise TypeError("Unexpected expected_result", expected_result)
+
+    calling_storage = {
+        slot_code_worked: value_code_worked,
+    }
+    destination_contract_storage = {
+        slot_call_result: expected_bytes,
+    }
+
+    post = {
+        calling_contract_address: Account(storage=calling_storage),
+        destination_contract_address: Account(storage=destination_contract_storage),
+    }
+
+    state_test(
+        env=env,
+        pre=pre,
+        post=post,
+        tx=tx,
+    )