diff --git a/src/providers/consensus/typings.py b/src/providers/consensus/typings.py index ec6d49873..01d24ae4d 100644 --- a/src/providers/consensus/typings.py +++ b/src/providers/consensus/typings.py @@ -87,13 +87,13 @@ class BlockAttestation(Nested, FromResponse): @dataclass -class BeaconBlockBody(Nested): +class BeaconBlockBody(Nested, FromResponse): execution_payload: ExecutionPayload attestations: list[BlockAttestation] @dataclass -class BlockMessage(Nested): +class BlockMessage(Nested, FromResponse): slot: str proposer_index: str parent_root: str diff --git a/tests/providers_clients/test_http_provider.py b/tests/providers_clients/test_http_provider.py index e9f13009d..2789cdeaf 100644 --- a/tests/providers_clients/test_http_provider.py +++ b/tests/providers_clients/test_http_provider.py @@ -25,7 +25,7 @@ def test_no_providers(): def test_all_fallbacks_ok(): provider = HTTPProvider(['http://localhost:1', 'http://localhost:2'], 5 * 60, 1, 1) - provider._get_without_fallbacks = lambda host, endpoint, path_params, query_params, stream: (host, endpoint) + provider._get_without_fallbacks = lambda host, endpoint, path_params, query_params: (host, endpoint) assert provider._get('test') == ('http://localhost:1', 'test') assert len(provider.get_all_providers()) == 2 @@ -60,4 +60,4 @@ def _simple_get(host, endpoint, *_): provider._get_without_fallbacks = Mock(side_effect=_simple_get) with pytest.raises(CustomError): provider._get('test', force_raise=lambda _: CustomError()) - provider._get_without_fallbacks.assert_called_once_with('http://localhost:1', 'test', None, None, False) + provider._get_without_fallbacks.assert_called_once_with('http://localhost:1', 'test', None, None)