mirror of https://github.com/hyperledger/besu
Initial implementation of ibft_getValidatorsByBlockNumber (#499)
* Initial implementation of ibft_getValidatorsByBlockHash * Added in missing method to blockchain queries and cleaned up code.
parent
d3a353e30a
commit
76fe768198
@ -0,0 +1,55 @@ |
||||
/* |
||||
* Copyright 2019 ConsenSys AG. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on |
||||
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the |
||||
* specific language governing permissions and limitations under the License. |
||||
*/ |
||||
package tech.pegasys.pantheon.consensus.ibft.jsonrpc.methods; |
||||
|
||||
import tech.pegasys.pantheon.consensus.ibft.IbftBlockInterface; |
||||
import tech.pegasys.pantheon.ethereum.core.BlockHeader; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.JsonRpcRequest; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.methods.AbstractBlockParameterMethod; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.methods.JsonRpcMethod; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.parameters.BlockParameter; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.parameters.JsonRpcParameter; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.queries.BlockchainQueries; |
||||
|
||||
import java.util.Optional; |
||||
|
||||
public class IbftGetValidatorsByBlockNumber extends AbstractBlockParameterMethod |
||||
implements JsonRpcMethod { |
||||
|
||||
private final IbftBlockInterface ibftBlockInterface; |
||||
|
||||
public IbftGetValidatorsByBlockNumber( |
||||
final BlockchainQueries blockchainQueries, |
||||
final IbftBlockInterface ibftBlockInterface, |
||||
final JsonRpcParameter parameters) { |
||||
super(blockchainQueries, parameters); |
||||
this.ibftBlockInterface = ibftBlockInterface; |
||||
} |
||||
|
||||
@Override |
||||
protected BlockParameter blockParameter(final JsonRpcRequest request) { |
||||
return parameters().required(request.getParams(), 0, BlockParameter.class); |
||||
} |
||||
|
||||
@Override |
||||
protected Object resultByBlockNumber(final JsonRpcRequest request, final long blockNumber) { |
||||
final Optional<BlockHeader> blockHeader = |
||||
blockchainQueries().getBlockHeaderByNumber(blockNumber); |
||||
return blockHeader.map(header -> ibftBlockInterface.validatorsInBlock(header)).orElse(null); |
||||
} |
||||
|
||||
@Override |
||||
public String getName() { |
||||
return "ibft_getValidatorsByBlockNumber"; |
||||
} |
||||
} |
@ -0,0 +1,72 @@ |
||||
/* |
||||
* Copyright 2019 ConsenSys AG. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on |
||||
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the |
||||
* specific language governing permissions and limitations under the License. |
||||
*/ |
||||
package tech.pegasys.pantheon.consensus.ibft.jsonrpc.methods; |
||||
|
||||
import static org.mockito.Mockito.when; |
||||
|
||||
import tech.pegasys.pantheon.consensus.ibft.IbftBlockInterface; |
||||
import tech.pegasys.pantheon.ethereum.core.Address; |
||||
import tech.pegasys.pantheon.ethereum.core.BlockHeader; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.JsonRpcRequest; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.parameters.BlockParameter; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.parameters.JsonRpcParameter; |
||||
import tech.pegasys.pantheon.ethereum.jsonrpc.internal.queries.BlockchainQueries; |
||||
|
||||
import java.util.Collections; |
||||
import java.util.List; |
||||
import java.util.Optional; |
||||
|
||||
import org.assertj.core.api.Assertions; |
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.mockito.Mock; |
||||
import org.mockito.junit.MockitoJUnitRunner; |
||||
|
||||
@RunWith(MockitoJUnitRunner.class) |
||||
public class IbftGetValidatorsByBlockNumberTest { |
||||
|
||||
@Mock private BlockchainQueries blockchainQueries; |
||||
@Mock private BlockHeader blockHeader; |
||||
@Mock private IbftBlockInterface ibftBlockInterface; |
||||
@Mock private JsonRpcRequest request; |
||||
|
||||
private final JsonRpcParameter parameters = new JsonRpcParameter(); |
||||
private IbftGetValidatorsByBlockNumber method; |
||||
|
||||
@Before |
||||
public void setUp() { |
||||
method = new IbftGetValidatorsByBlockNumber(blockchainQueries, ibftBlockInterface, parameters); |
||||
} |
||||
|
||||
@Test |
||||
public void blockParameterIsParameter0() { |
||||
request = new JsonRpcRequest("?", "ignore", new String[] {"0x1245"}); |
||||
BlockParameter blockParameter = method.blockParameter(request); |
||||
Assertions.assertThat(blockParameter.getNumber().getAsLong()).isEqualTo(0x1245); |
||||
} |
||||
|
||||
@Test |
||||
public void nameShouldBeCorrect() { |
||||
Assertions.assertThat(method.getName()).isEqualTo("ibft_getValidatorsByBlockNumber"); |
||||
} |
||||
|
||||
@Test |
||||
public void shouldReturnListOfValidatorsFromBlock() { |
||||
when(blockchainQueries.getBlockHeaderByNumber(12)).thenReturn(Optional.of(blockHeader)); |
||||
final List<Address> addresses = Collections.singletonList(Address.ID); |
||||
when(ibftBlockInterface.validatorsInBlock(blockHeader)).thenReturn(addresses); |
||||
Object result = method.resultByBlockNumber(request, 12); |
||||
Assertions.assertThat(result).isEqualTo(addresses); |
||||
} |
||||
} |
Loading…
Reference in new issue