-
Notifications
You must be signed in to change notification settings - Fork 110
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add Support for BlockchainTests (#2)
* Readme changes * Add support for BlockchainTests * Update README * Fix chain_id test * Check evm binary on init * Type fixes and optimizations * Change dataclasses to keyword only * Simplify type parsing/dumping * Rename method * Fix block_hashes assumption * Nit fixes for tox * Add missing docstrings * Update mypy version to fix dataclass error * Directory change for test fixtures * Fix path in test * Fix removal of invalid txs * Name and format fixes * Fix baseFee on all fillers * Fix base fee default
- Loading branch information
Showing
20 changed files
with
1,794 additions
and
193 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,110 @@ | ||
| """ | ||
| Generic Ethereum test base class | ||
| """ | ||
| from abc import abstractmethod | ||
| from typing import Any, Callable, Dict, Generator, List, Mapping, Tuple, Union | ||
|
|
||
| from evm_block_builder import BlockBuilder | ||
| from evm_transition_tool import TransitionTool | ||
|
|
||
| from .types import Account, FixtureBlock, FixtureHeader, Transaction | ||
|
|
||
|
|
||
| def normalize_address(address: str) -> str: | ||
| """ | ||
| Normalizes an address to be able to look it up in the alloc that is | ||
| produced by the transition tool. | ||
| """ | ||
| address = address.lower() | ||
| if address.startswith("0x"): | ||
| address = address[2:] | ||
| address.rjust(40, "0") | ||
| if len(address) > 40: | ||
| raise Exception("invalid address") | ||
|
|
||
| return "0x" + address | ||
|
|
||
|
|
||
| def verify_transactions( | ||
| txs: Union[List[Transaction], None], result | ||
| ) -> List[int]: | ||
| """ | ||
| Verify rejected transactions (if any) against the expected outcome. | ||
| Raises exception on unexpected rejections or unexpected successful txs. | ||
| """ | ||
| rejected_txs: Dict[int, Any] = {} | ||
| if "rejected" in result: | ||
| for rejected_tx in result["rejected"]: | ||
| if "index" not in rejected_tx or "error" not in rejected_tx: | ||
| raise Exception("badly formatted result") | ||
| rejected_txs[rejected_tx["index"]] = rejected_tx["error"] | ||
|
|
||
| if txs is not None: | ||
| for i, tx in enumerate(txs): | ||
| error = rejected_txs[i] if i in rejected_txs else None | ||
| if tx.error and not error: | ||
| raise Exception("tx expected to fail succeeded") | ||
| elif not tx.error and error: | ||
| raise Exception(f"tx unexpectedly failed: {error}") | ||
|
|
||
| # TODO: Also we need a way to check we actually got the | ||
| # correct error | ||
| return list(rejected_txs.keys()) | ||
|
|
||
|
|
||
| def verify_post_alloc(post: Mapping[str, Account], alloc: Mapping[str, Any]): | ||
| """ | ||
| Verify that an allocation matches the expected post in the test. | ||
| Raises exception on unexpected values. | ||
| """ | ||
| for address, account in post.items(): | ||
| address = normalize_address(address) | ||
| if account is None: | ||
| # If an account is None in post, it must not exist in the | ||
| # alloc. | ||
| if address in alloc: | ||
| raise Exception(f"found unexpected account: {address}") | ||
| else: | ||
| if address in alloc: | ||
| account.check_alloc(address, alloc[address]) | ||
| else: | ||
| raise Exception(f"expected account not found: {address}") | ||
|
|
||
|
|
||
| class BaseTest: | ||
| """ | ||
| Represents a base Ethereum test which must return a genesis and a | ||
| blockchain. | ||
| """ | ||
|
|
||
| pre: Mapping[str, Account] | ||
|
|
||
| @abstractmethod | ||
| def make_genesis( | ||
| self, | ||
| b11r: BlockBuilder, | ||
| t8n: TransitionTool, | ||
| fork: str, | ||
| ) -> FixtureHeader: | ||
| """ | ||
| Create a genesis block from the test definition. | ||
| """ | ||
| pass | ||
|
|
||
| @abstractmethod | ||
| def make_blocks( | ||
| self, | ||
| b11r: BlockBuilder, | ||
| t8n: TransitionTool, | ||
| genesis: FixtureHeader, | ||
| fork: str, | ||
| chain_id: int = 1, | ||
| reward: int = 0, | ||
| ) -> Tuple[List[FixtureBlock], str]: | ||
| """ | ||
| Generate the blockchain that must be executed sequentially during test. | ||
| """ | ||
| pass | ||
|
|
||
|
|
||
| TestSpec = Callable[[str], Generator[BaseTest, None, None]] |
Oops, something went wrong.