starred/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2026-08-24 10:05:29 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
40db4635a8589a92e00d9778461135808550c77a
chia-blockchain/chia/_tests/util/test_config.py
T
Matt HauffandGitHub 40db4635a8 [LABS-245] Enable PEP604 Ruff rules (#20269) 
* Enable PEP604 Ruff rules

* Fix harcoded signature in test

* Hack CLVMStreamable test with note to fast follow
2025-11-18 12:34:00 -08:00

428 lines
12 KiB
Python
Raw Blame History

from __future__ import annotations
from dataclasses import dataclass
from typing import Any
from chia_rs.sized_ints import uint16
from chia._tests.util.misc import DataCase, Marks, datacases
from chia.protocols.outbound_message import NodeType
from chia.server.resolve_peer_info import get_unresolved_peer_infos, set_peer_info
from chia.types.peer_info import UnresolvedPeerInfo
@dataclass
class GetUnresolvedPeerInfosCase(DataCase):
description: str
service_config: dict[str, Any]
requested_node_type: NodeType
expected_peer_infos: set[UnresolvedPeerInfo]
marks: Marks = ()
@property
def id(self) -> str:
return self.description
@datacases(
GetUnresolvedPeerInfosCase(
description="multiple farmer peers",
service_config={
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
requested_node_type=NodeType.FARMER,
expected_peer_infos={
UnresolvedPeerInfo(host="127.0.0.1", port=uint16(8447)),
UnresolvedPeerInfo(host="my.farmer.tld", port=uint16(18447)),
},
),
GetUnresolvedPeerInfosCase(
description="single farmer peer",
service_config={
"farmer_peer": {
"host": "my.farmer.tld",
"port": 18447,
},
},
requested_node_type=NodeType.FARMER,
expected_peer_infos={
UnresolvedPeerInfo(host="my.farmer.tld", port=uint16(18447)),
},
),
GetUnresolvedPeerInfosCase(
description="single farmer peer and multiple farmer peers",
service_config={
"farmer_peer": {
"host": "my.farmer.tld",
"port": 18447,
},
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.other.farmer.tld",
"port": 18447,
},
],
},
requested_node_type=NodeType.FARMER,
expected_peer_infos={
UnresolvedPeerInfo(host="my.farmer.tld", port=uint16(18447)),
UnresolvedPeerInfo(host="127.0.0.1", port=uint16(8447)),
UnresolvedPeerInfo(host="my.other.farmer.tld", port=uint16(18447)),
},
),
GetUnresolvedPeerInfosCase(
description="multiple full node peers",
service_config={
"full_node_peers": [
{
"host": "127.0.0.1",
"port": 8444,
},
{
"host": "my.full-node.tld",
"port": 18444,
},
],
},
requested_node_type=NodeType.FULL_NODE,
expected_peer_infos={
UnresolvedPeerInfo(host="127.0.0.1", port=uint16(8444)),
UnresolvedPeerInfo(host="my.full-node.tld", port=uint16(18444)),
},
),
GetUnresolvedPeerInfosCase(
description="single full node peer",
service_config={
"full_node_peer": {
"host": "my.full-node.tld",
"port": 18444,
},
},
requested_node_type=NodeType.FULL_NODE,
expected_peer_infos={
UnresolvedPeerInfo(host="my.full-node.tld", port=uint16(18444)),
},
),
GetUnresolvedPeerInfosCase(
description="single full node peer and multiple full node peers",
service_config={
"full_node_peer": {
"host": "my.full-node.tld",
"port": 18444,
},
"full_node_peers": [
{
"host": "127.0.0.1",
"port": 8444,
},
{
"host": "my.other.full-node.tld",
"port": 18444,
},
],
},
requested_node_type=NodeType.FULL_NODE,
expected_peer_infos={
UnresolvedPeerInfo(host="my.full-node.tld", port=uint16(18444)),
UnresolvedPeerInfo(host="127.0.0.1", port=uint16(8444)),
UnresolvedPeerInfo(host="my.other.full-node.tld", port=uint16(18444)),
},
),
GetUnresolvedPeerInfosCase(
description="no peer info in config",
service_config={},
requested_node_type=NodeType.FULL_NODE,
expected_peer_infos=set(),
),
)
def test_get_unresolved_peer_infos(case: GetUnresolvedPeerInfosCase) -> None:
assert get_unresolved_peer_infos(case.service_config, case.requested_node_type) == case.expected_peer_infos
@dataclass
class SetPeerInfoCase(DataCase):
description: str
service_config: dict[str, Any]
requested_node_type: NodeType
expected_service_config: dict[str, Any]
peer_host: str | None = None
peer_port: int | None = None
marks: Marks = ()
@property
def id(self) -> str:
return self.description
@datacases(
SetPeerInfoCase(
description="multiple peers, modify first entry, set host and port",
service_config={
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
requested_node_type=NodeType.FARMER,
peer_host="localhost",
peer_port=1337,
expected_service_config={
"farmer_peers": [
{
"host": "localhost",
"port": 1337,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
),
SetPeerInfoCase(
description="multiple peers, modify first entry, set host",
service_config={
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
requested_node_type=NodeType.FARMER,
peer_host="localhost",
expected_service_config={
"farmer_peers": [
{
"host": "localhost",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
),
SetPeerInfoCase(
description="multiple peers, modify first entry, set port",
service_config={
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
requested_node_type=NodeType.FARMER,
peer_port=1337,
expected_service_config={
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 1337,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
),
SetPeerInfoCase(
description="single peer, set host and port",
service_config={
"farmer_peer": {
"host": "127.0.0.1",
"port": 8447,
},
},
requested_node_type=NodeType.FARMER,
peer_host="localhost",
peer_port=1337,
expected_service_config={
"farmer_peer": {
"host": "localhost",
"port": 1337,
},
},
),
SetPeerInfoCase(
description="single peer, set host",
service_config={
"farmer_peer": {
"host": "127.0.0.1",
"port": 8447,
},
},
requested_node_type=NodeType.FARMER,
peer_host="localhost",
expected_service_config={
"farmer_peer": {
"host": "localhost",
"port": 8447,
},
},
),
SetPeerInfoCase(
description="single peer, set port",
service_config={
"farmer_peer": {
"host": "127.0.0.1",
"port": 8447,
},
},
requested_node_type=NodeType.FARMER,
peer_port=1337,
expected_service_config={
"farmer_peer": {
"host": "127.0.0.1",
"port": 1337,
},
},
),
SetPeerInfoCase(
description="single and multiple peers, modify single peer, set host and port",
service_config={
"farmer_peer": {
"host": "127.0.0.1",
"port": 28447,
},
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
requested_node_type=NodeType.FARMER,
peer_host="localhost",
peer_port=1337,
expected_service_config={
"farmer_peer": {
"host": "localhost",
"port": 1337,
},
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
),
SetPeerInfoCase(
description="single and multiple peers, modify single peer, set host",
service_config={
"farmer_peer": {
"host": "127.0.0.1",
"port": 28447,
},
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
requested_node_type=NodeType.FARMER,
peer_host="localhost",
expected_service_config={
"farmer_peer": {
"host": "localhost",
"port": 28447,
},
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
),
SetPeerInfoCase(
description="single and multiple peers, modify single peer, set port",
service_config={
"farmer_peer": {
"host": "127.0.0.1",
"port": 28447,
},
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
requested_node_type=NodeType.FARMER,
peer_port=1337,
expected_service_config={
"farmer_peer": {
"host": "127.0.0.1",
"port": 1337,
},
"farmer_peers": [
{
"host": "127.0.0.1",
"port": 8447,
},
{
"host": "my.farmer.tld",
"port": 18447,
},
],
},
),
)
def test_set_peer_info(case: SetPeerInfoCase) -> None:
set_peer_info(case.service_config, case.requested_node_type, peer_host=case.peer_host, peer_port=case.peer_port)
assert case.service_config == case.expected_service_config
Reference in New Issue View Git Blame Copy Permalink