chia-blockchain 2.5.0rc1__py3-none-any.whl → 2.5.1__py3-none-any.whl

chia/_tests/util/misc.py

@@ -12,36 +12,15 @@ import pathlib
 import ssl
 import subprocess
 import sys
+from collections.abc import Awaitable, Collection, Iterator
 from concurrent.futures import Future
 from dataclasses import dataclass, field
 from enum import Enum
-from inspect import getframeinfo, stack
-from pathlib import Path
 from statistics import mean
 from textwrap import dedent
 from time import thread_time
 from types import TracebackType
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Awaitable,
-    Callable,
-    ClassVar,
-    Collection,
-    Dict,
-    Iterable,
-    Iterator,
-    List,
-    Optional,
-    Protocol,
-    TextIO,
-    Tuple,
-    Type,
-    TypeVar,
-    Union,
-    cast,
-    final,
-)
+from typing import TYPE_CHECKING, Any, Callable, ClassVar, Optional, Protocol, TextIO, TypeVar, Union, cast, final
 
 import aiohttp
 import pytest
@@ -55,14 +34,13 @@ import chia
 import chia._tests
 from chia._tests import ether
 from chia._tests.core.data_layer.util import ChiaRoot
-from chia.consensus.difficulty_adjustment import get_next_sub_slot_iters_and_difficulty
-from chia.full_node.full_node import FullNode
+from chia._tests.util.time_out_assert import DataTypeProtocol, caller_file_and_line
 from chia.full_node.mempool import Mempool
+from chia.protocols.protocol_message_types import ProtocolMessageTypes
+from chia.server.api_protocol import ApiMetadata, ApiProtocol
+from chia.server.outbound_message import Message
 from chia.types.blockchain_format.sized_bytes import bytes32
 from chia.types.condition_opcodes import ConditionOpcode
-from chia.types.full_block import FullBlock
-from chia.types.peer_info import PeerInfo
-from chia.util.batches import to_batches
 from chia.util.hash import std_hash
 from chia.util.ints import uint16, uint32, uint64
 from chia.util.network import WebServer
@@ -186,7 +164,7 @@ def measure_overhead(
     ],
     cycles: int = 10,
 ) -> float:
-    times: List[float] = []
+    times: list[float] = []
 
     for _ in range(cycles):
         with manager_maker() as results:
@@ -257,10 +235,10 @@ class BenchmarkData:
 
     label: str
 
-    __match_args__: ClassVar[Tuple[str, ...]] = ()
+    __match_args__: ClassVar[tuple[str, ...]] = ()
 
     @classmethod
-    def unmarshal(cls, marshalled: Dict[str, Any]) -> BenchmarkData:
+    def unmarshal(cls, marshalled: dict[str, Any]) -> BenchmarkData:
         return cls(
             duration=marshalled["duration"],
             path=pathlib.Path(marshalled["path"]),
@@ -269,7 +247,7 @@ class BenchmarkData:
             label=marshalled["label"],
         )
 
-    def marshal(self) -> Dict[str, Any]:
+    def marshal(self) -> dict[str, Any]:
         return {
             "duration": self.duration,
             "path": self.path.as_posix(),
@@ -336,7 +314,7 @@ class _AssertRuntime:
 
     def __exit__(
         self,
-        exc_type: Optional[Type[BaseException]],
+        exc_type: Optional[type[BaseException]],
         exc: Optional[BaseException],
         traceback: Optional[TracebackType],
     ) -> None:
@@ -373,7 +351,7 @@ class _AssertRuntime:
                 label=self.label,
             )
 
-            ether.record_property(  # pylint: disable=E1102
+            ether.record_property(
                 data.tag,
                 json.dumps(data.marshal(), ensure_ascii=True, sort_keys=True),
             )
@@ -404,7 +382,7 @@ def assert_rpc_error(error: str) -> Iterator[None]:
 
 
 @contextlib.contextmanager
-def closing_chia_root_popen(chia_root: ChiaRoot, args: List[str]) -> Iterator[subprocess.Popen[Any]]:
+def closing_chia_root_popen(chia_root: ChiaRoot, args: list[str]) -> Iterator[subprocess.Popen[Any]]:
     environment = {**os.environ, "CHIA_ROOT": os.fspath(chia_root.path)}
 
     with subprocess.Popen(args=args, env=environment) as process:
@@ -475,7 +453,7 @@ class CoinGenerator:
         return HintedCoin(Coin(parent_coin_id, self._get_hash(), self._get_amount()), hint)
 
 
-def coin_creation_args(hinted_coin: HintedCoin) -> List[Any]:
+def coin_creation_args(hinted_coin: HintedCoin) -> list[Any]:
     if hinted_coin.hint is not None:
         memos = [hinted_coin.hint]
     else:
@@ -514,7 +492,7 @@ async def wallet_height_at_least(wallet_node: WalletNode, h: uint32) -> bool:
 @dataclass
 class RecordingWebServer:
     web_server: WebServer
-    requests: List[web.Request] = field(default_factory=list)
+    requests: list[web.Request] = field(default_factory=list)
 
     @classmethod
     async def create(
@@ -540,7 +518,7 @@ class RecordingWebServer:
         await web_server.start()
         return self
 
-    def get_routes(self) -> Dict[str, Callable[[web.Request], Awaitable[web.Response]]]:
+    def get_routes(self) -> dict[str, Callable[[web.Request], Awaitable[web.Response]]]:
         return {"/{path:.*}": self.handler}
 
     async def handler(self, request: web.Request) -> web.Response:
@@ -563,12 +541,12 @@ class RecordingWebServer:
 @dataclasses.dataclass(frozen=True)
 class TestId:
     platform: str
-    test_path: Tuple[str, ...]
-    ids: Tuple[str, ...]
+    test_path: tuple[str, ...]
+    ids: tuple[str, ...]
 
     @classmethod
     def create(cls, node: Node, platform: str = sys.platform) -> TestId:
-        test_path: List[str] = []
+        test_path: list[str] = []
         temp_node = node
         while True:
             name: str
@@ -587,8 +565,8 @@ class TestId:
             temp_node = temp_node.parent
 
         # TODO: can we avoid parsing the id's etc from the node name?
-        test_name, delimiter, rest = node.name.partition("[")
-        ids: Tuple[str, ...]
+        _test_name, delimiter, rest = node.name.partition("[")
+        ids: tuple[str, ...]
         if delimiter == "":
             ids = ()
         else:
@@ -601,14 +579,14 @@ class TestId:
         )
 
     @classmethod
-    def unmarshal(cls, marshalled: Dict[str, Any]) -> TestId:
+    def unmarshal(cls, marshalled: dict[str, Any]) -> TestId:
         return cls(
             platform=marshalled["platform"],
             test_path=tuple(marshalled["test_path"]),
             ids=tuple(marshalled["ids"]),
         )
 
-    def marshal(self) -> Dict[str, Any]:
+    def marshal(self) -> dict[str, Any]:
         return {
             "platform": self.platform,
             "test_path": self.test_path,
@@ -616,27 +594,6 @@ class TestId:
         }
 
 
-T = TypeVar("T")
-
-
-@dataclasses.dataclass(frozen=True)
-class DataTypeProtocol(Protocol):
-    tag: ClassVar[str]
-
-    line: int
-    path: Path
-    label: str
-    duration: float
-    limit: float
-
-    __match_args__: ClassVar[Tuple[str, ...]] = ()
-
-    @classmethod
-    def unmarshal(cls: Type[T], marshalled: Dict[str, Any]) -> T: ...
-
-    def marshal(self) -> Dict[str, Any]: ...
-
-
 T_ComparableEnum = TypeVar("T_ComparableEnum", bound="ComparableEnum")
 
 
@@ -678,45 +635,41 @@ class ComparableEnum(Enum):
         return self.value.__ge__(other.value)
 
 
-def caller_file_and_line(distance: int = 1, relative_to: Iterable[Path] = ()) -> Tuple[str, int]:
-    caller = getframeinfo(stack()[distance + 1][0])
-
-    caller_path = Path(caller.filename)
-    options: List[str] = [caller_path.as_posix()]
-    for path in relative_to:
-        try:
-            options.append(caller_path.relative_to(path).as_posix())
-        except ValueError:
-            pass
-
-    return min(options, key=len), caller.lineno
+def is_attribute_local(o: object, name: str) -> bool:
+    return name in getattr(o, "__dict__", ()) or name in getattr(o, "__slots__", ())
 
 
-async def add_blocks_in_batches(
-    blocks: List[FullBlock],
-    full_node: FullNode,
-    header_hash: Optional[bytes32] = None,
-) -> None:
-    if header_hash is None:
-        diff = full_node.constants.DIFFICULTY_STARTING
-        ssi = full_node.constants.SUB_SLOT_ITERS_STARTING
-    else:
-        block_record = await full_node.blockchain.get_block_record_from_db(header_hash)
-        ssi, diff = get_next_sub_slot_iters_and_difficulty(
-            full_node.constants, True, block_record, full_node.blockchain
-        )
-    prev_ses_block = None
-    for block_batch in to_batches(blocks, 64):
-        b = block_batch.entries[0]
-        if (b.height % 128) == 0:
-            print(f"main chain: {b.height:4} weight: {b.weight}")
-        success, _, ssi, diff, prev_ses_block, err = await full_node.add_block_batch(
-            block_batch.entries,
-            PeerInfo("0.0.0.0", 0),
-            None,
-            current_ssi=ssi,
-            current_difficulty=diff,
-            prev_ses_block=prev_ses_block,
-        )
-        assert err is None
-        assert success is True
+@contextlib.contextmanager
+def patch_request_handler(
+    api: Union[ApiProtocol, type[ApiProtocol]],
+    handler: Callable[..., Awaitable[Optional[Message]]],
+    request_type: Optional[ProtocolMessageTypes] = None,
+) -> Iterator[None]:
+    if request_type is None:
+        request_type = ProtocolMessageTypes[handler.__name__]
+
+    metadata = ApiMetadata.copy(api.metadata)
+    original_request = metadata.message_type_to_request.pop(request_type)
+    decorator = metadata.request(
+        peer_required=original_request.peer_required,
+        bytes_required=original_request.bytes_required,
+        execute_task=original_request.execute_task,
+        reply_types=original_request.reply_types,
+        request_type=original_request.request_type,
+    )
+    decorator(handler)
+
+    was_local = is_attribute_local(api, "metadata")
+    original = api.metadata
+    # when an instance is passed, this is intentionally assigning to an instance
+    # counter to the class variable hint
+    api.metadata = metadata  # type: ignore[misc]
+    try:
+        yield
+    finally:
+        if was_local:
+            # when an instance is passed, this is intentionally assigning to an instance
+            # counter to the class variable hint
+            api.metadata = original  # type: ignore[misc]
+        else:
+            del api.metadata

chia/_tests/util/network_protocol_data.py

@@ -1,5 +1,3 @@
-# flake8: noqa
-
 from __future__ import annotations
 
 from chia_rs import G1Element, G2Element, RewardChainBlockUnfinished
@@ -46,7 +44,7 @@ error_without_data = Error(int16(Err.UNKNOWN.value), "Unknown", None)
 error_with_data = Error(int16(Err.UNKNOWN.value), "Unknown", b"extra data")
 
 
-### FARMER PROTOCOL
+# FARMER PROTOCOL
 new_signage_point = farmer_protocol.NewSignagePoint(
     bytes32(bytes.fromhex("34b2a753b0dc864e7218f8facf23ca0e2b636351df5289b76f5845d9a78b7026")),
     bytes32(bytes.fromhex("9dc8b9d685c79acdf8780d994416dfcfb118e0adc99769ecfa94e1f40aa5bbe5")),
@@ -142,7 +140,7 @@ signed_values = farmer_protocol.SignedValues(
 )
 
 
-### FULL NODE PROTOCOL.
+# FULL NODE PROTOCOL.
 new_peak = full_node_protocol.NewPeak(
     bytes32(bytes.fromhex("8a346e8dc02e9b44c0571caa74fd99f163d4c5d7deae9f8ddb00528721493f7a")),
     uint32(2653549198),
@@ -503,7 +501,7 @@ timestamped_peer_info = TimestampedPeerInfo("127.0.0.1", uint16(8444), uint64(10
 respond_peers = full_node_protocol.RespondPeers([timestamped_peer_info])
 
 
-## WALLET PROTOCOL
+# WALLET PROTOCOL
 request_puzzle_solution = wallet_protocol.RequestPuzzleSolution(
     bytes32(bytes.fromhex("6edddb46bd154f50566b49c95812e0f1131a0a7162630349fc8d1d696e463e47")),
     uint32(3905474497),
@@ -755,7 +753,7 @@ respond_cost_info = wallet_protocol.RespondCostInfo(
 )
 
 
-### HARVESTER PROTOCOL
+# HARVESTER PROTOCOL
 pool_difficulty = harvester_protocol.PoolDifficulty(
     uint64(14819251421858580996),
     uint64(12852879676624401630),
@@ -899,7 +897,7 @@ respond_plots = harvester_protocol.RespondPlots(
     ["str"],
 )
 
-### INTRODUCER PROTOCOL
+# INTRODUCER PROTOCOL
 request_peers_introducer = introducer_protocol.RequestPeersIntroducer()
 
 respond_peers_introducer = introducer_protocol.RespondPeersIntroducer(
@@ -913,7 +911,7 @@ respond_peers_introducer = introducer_protocol.RespondPeersIntroducer(
 )
 
 
-### POOL PROTOCOL
+# POOL PROTOCOL
 authentication_payload = pool_protocol.AuthenticationPayload(
     "method",
     bytes32(bytes.fromhex("0251e3b3a1aacc689091b6b085be7a8d319bd9d1a015faae969cb76d8a45607c")),
@@ -1002,7 +1000,7 @@ error_response = pool_protocol.ErrorResponse(
     "err",
 )
 
-### TIMELORD PROTOCOL
+# TIMELORD PROTOCOL
 sub_epoch_summary = SubEpochSummary(
     bytes32(bytes.fromhex("2d0550de416467e7b57e56e962c712b79bee29cae29c73cc908da5978fc9789e")),
     bytes32(bytes.fromhex("3d29f5a3fe067ce7edea76c9cebaf3a3afdebc0eb9fbd530f807f1a28ed2df6d")),