CipherLabs
/
besu
mirror of https://github.com/hyperledger/besu
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add more tests for the GQ private transaction bloom filters (#2394)

Browse Source 
* add tests for the GQ private transaction bloom filters

Signed-off-by: Stefan Pingel <stefan.pingel@consensys.net>
pull/2392/head
Stefan Pingel 4 years ago committed by GitHub
parent de6c10d17f
commit b769995fe8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 305 additions and 1 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 129
      ethereum/api/src/test/java/org/hyperledger/besu/ethereum/api/query/GoQuorumPrivateTxBloomBlockchainQueriesTest.java
  2. 175
      ethereum/api/src/test/java/org/hyperledger/besu/ethereum/api/query/cache/GoQuorumPrivateTransactionLogBloomCacherTest.java
  3. 2
      ethereum/api/src/test/java/org/hyperledger/besu/ethereum/api/query/cache/TransactionLogBloomCacherTest.java

129
ethereum/api/src/test/java/org/hyperledger/besu/ethereum/api/query/GoQuorumPrivateTxBloomBlockchainQueriesTest.java
Unescape View File

@ -0,0 +1,129 @@
/*
* Copyright 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.
*
* SPDX-License-Identifier: Apache-2.0
*
*/
package org.hyperledger.besu.ethereum.api.query;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import static org.mockito.Mockito.when;
import org.hyperledger.besu.ethereum.chain.MutableBlockchain;
import org.hyperledger.besu.ethereum.core.Address;
import org.hyperledger.besu.ethereum.core.BlockBody;
import org.hyperledger.besu.ethereum.core.BlockHeader;
import org.hyperledger.besu.ethereum.core.Difficulty;
import org.hyperledger.besu.ethereum.core.Hash;
import org.hyperledger.besu.ethereum.core.Log;
import org.hyperledger.besu.ethereum.core.LogsBloomFilter;
import org.hyperledger.besu.ethereum.eth.manager.EthScheduler;
import org.hyperledger.besu.ethereum.mainnet.MainnetBlockHeaderFunctions;
import org.hyperledger.besu.ethereum.worldstate.WorldStateArchive;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import org.apache.tuweni.bytes.Bytes;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.ClassRule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
@RunWith(MockitoJUnitRunner.class)
public class GoQuorumPrivateTxBloomBlockchainQueriesTest {
@ClassRule public static TemporaryFolder cacheDir = new TemporaryFolder();
private static LogsQuery logsQuery;
private Hash testHash;
private static LogsBloomFilter testLogsBloomFilter;
@Mock MutableBlockchain blockchain;
@Mock WorldStateArchive worldStateArchive;
@Mock EthScheduler scheduler;
private BlockchainQueries blockchainQueries;
@BeforeClass
public static void setupClass() {
final Address testAddress = Address.fromHexString("0x123456");
final Bytes testMessage = Bytes.fromHexString("0x9876");
final Log testLog = new Log(testAddress, testMessage, List.of());
testLogsBloomFilter = LogsBloomFilter.builder().insertLog(testLog).build();
logsQuery = new LogsQuery(List.of(testAddress), List.of());
}
@Before
public void setup() {
final BlockHeader fakeHeader =
new BlockHeader(
Hash.EMPTY,
Hash.EMPTY,
Address.ZERO,
Hash.EMPTY,
Hash.EMPTY,
Hash.EMPTY,
LogsBloomFilter.empty(),
Difficulty.ZERO,
0,
0,
0,
0,
Bytes.EMPTY,
null,
Hash.EMPTY,
0,
new MainnetBlockHeaderFunctions(),
Optional.of(testLogsBloomFilter));
testHash = fakeHeader.getHash();
final BlockBody fakeBody = new BlockBody(Collections.emptyList(), Collections.emptyList());
when(blockchain.getBlockHeader(any())).thenReturn(Optional.of(fakeHeader));
when(blockchain.getBlockHeader(anyLong())).thenReturn(Optional.of(fakeHeader));
when(blockchain.getTxReceipts(any())).thenReturn(Optional.of(Collections.emptyList()));
when(blockchain.getBlockBody(any())).thenReturn(Optional.of(fakeBody));
blockchainQueries =
new BlockchainQueries(
blockchain,
worldStateArchive,
Optional.of(cacheDir.getRoot().toPath()),
Optional.of(scheduler));
}
/**
* Tests whether block headers containing private blooms match. The resulting list of
* LogWithMetadata would be empty, because the mocked blockchain does not return any receipts, but
* we do check that the methods on the blockchain are actually called that would be called if the
* block header matches the bloom filter.
*/
@Test
public void testPrivateBloomsWork() {
blockchainQueries.matchingLogs(0, 2, logsQuery, () -> true);
verify(blockchain, times(3)).getBlockHeader(anyLong());
verify(blockchain, times(3)).getBlockHeader(testHash);
verify(blockchain, times(3)).getTxReceipts(testHash);
verify(blockchain, times(3)).getBlockBody(testHash);
verify(blockchain, times(3)).blockIsOnCanonicalChain(testHash);
verifyNoMoreInteractions(blockchain);
}
}

175
ethereum/api/src/test/java/org/hyperledger/besu/ethereum/api/query/cache/GoQuorumPrivateTransactionLogBloomCacherTest.java vendored
Unescape View File

@ -0,0 +1,175 @@
/*
* Copyright 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.
*
* SPDX-License-Identifier: Apache-2.0
*
*/
package org.hyperledger.besu.ethereum.api.query.cache;
import static org.assertj.core.api.Assertions.assertThat;
import static org.hyperledger.besu.ethereum.api.query.cache.TransactionLogBloomCacher.BLOOM_BITS_LENGTH;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import static org.mockito.Mockito.when;
import org.hyperledger.besu.ethereum.api.query.BlockchainQueries;
import org.hyperledger.besu.ethereum.api.query.LogsQuery;
import org.hyperledger.besu.ethereum.chain.MutableBlockchain;
import org.hyperledger.besu.ethereum.core.Address;
import org.hyperledger.besu.ethereum.core.BlockBody;
import org.hyperledger.besu.ethereum.core.BlockHeader;
import org.hyperledger.besu.ethereum.core.Difficulty;
import org.hyperledger.besu.ethereum.core.Hash;
import org.hyperledger.besu.ethereum.core.Log;
import org.hyperledger.besu.ethereum.core.LogsBloomFilter;
import org.hyperledger.besu.ethereum.eth.manager.EthScheduler;
import org.hyperledger.besu.ethereum.mainnet.MainnetBlockHeaderFunctions;
import org.hyperledger.besu.ethereum.worldstate.WorldStateArchive;
import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import org.apache.tuweni.bytes.Bytes;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
@SuppressWarnings("unused")
@RunWith(MockitoJUnitRunner.class)
public class GoQuorumPrivateTransactionLogBloomCacherTest {
private static final long NUMBER_3 = 3L;
private static LogsQuery logsQuery;
@Rule public TemporaryFolder cacheDir = new TemporaryFolder();
private Hash testBlockHeaderHash;
private static LogsBloomFilter testLogsBloomFilter;
@Mock MutableBlockchain blockchain;
@Mock EthScheduler scheduler;
@Mock WorldStateArchive worldStateArchive;
private TransactionLogBloomCacher transactionLogBloomCacher;
private BlockchainQueries blockchainQueries;
private static Address testAddress;
private static Bytes testMessage;
private BlockHeader fakeHeader;
@BeforeClass
public static void setupClass() {
testAddress = Address.fromHexString("0x123456");
testMessage = Bytes.fromHexString("0x9876");
final Log testLog = new Log(testAddress, testMessage, List.of());
testLogsBloomFilter = LogsBloomFilter.builder().insertLog(testLog).build();
logsQuery = new LogsQuery(List.of(testAddress), List.of());
}
@SuppressWarnings({"unchecked", "ReturnValueIgnored"})
@Before
public void setup() throws IOException {
fakeHeader = createBlock(NUMBER_3);
testBlockHeaderHash = fakeHeader.getHash();
transactionLogBloomCacher =
new TransactionLogBloomCacher(blockchain, cacheDir.getRoot().toPath(), scheduler);
blockchainQueries =
new BlockchainQueries(
blockchain,
worldStateArchive,
Optional.of(cacheDir.getRoot().toPath()),
Optional.of(scheduler));
}
@Test
public void shouldUpdateCacheWhenBlockAdded() throws IOException {
final BlockBody fakeBody = new BlockBody(Collections.emptyList(), Collections.emptyList());
when(blockchain.getBlockHeader(testBlockHeaderHash)).thenReturn(Optional.of(fakeHeader));
when(blockchain.getBlockHashByNumber(anyLong())).thenReturn(Optional.of(testBlockHeaderHash));
when(blockchain.getTxReceipts(any())).thenReturn(Optional.of(Collections.emptyList()));
when(blockchain.getBlockBody(any())).thenReturn(Optional.of(fakeBody));
final File logBloom = cacheDir.newFile("logBloom-0.cache");
createLogBloomCache(logBloom);
createBlock(3L);
assertThat(logBloom.length()).isEqualTo(BLOOM_BITS_LENGTH * 3);
when(blockchain.getBlockHeader(anyLong())).thenReturn(Optional.of(fakeHeader));
transactionLogBloomCacher.cacheLogsBloomForBlockHeader(
blockchain.getBlockHeader(NUMBER_3).get(), Optional.empty(), Optional.of(logBloom));
assertThat(logBloom.length()).isEqualTo(BLOOM_BITS_LENGTH * 4);
assertThat(cacheDir.getRoot().list().length).isEqualTo(1);
blockchainQueries.matchingLogs(NUMBER_3, 3, logsQuery, () -> true);
verify(blockchain, times(1)).getBlockHashByNumber(NUMBER_3);
verify(blockchain, times(1)).getBlockHeader(NUMBER_3);
verify(blockchain, times(1)).getBlockHeader(testBlockHeaderHash);
verify(blockchain, times(1)).getTxReceipts(testBlockHeaderHash);
verify(blockchain, times(1)).getBlockBody(testBlockHeaderHash);
verify(blockchain, times(1)).blockIsOnCanonicalChain(testBlockHeaderHash);
verifyNoMoreInteractions(blockchain);
}
private void createLogBloomCache(final File logBloom) throws IOException {
try (final RandomAccessFile randomAccessFile = new RandomAccessFile(logBloom, "rws")) {
randomAccessFile.write(testLogsBloomFilter.toArray());
randomAccessFile.write(testLogsBloomFilter.toArray());
randomAccessFile.write(testLogsBloomFilter.toArray());
}
}
private BlockHeader createBlock(final long number) {
final Log testLog = new Log(testAddress, testMessage, List.of());
final BlockHeader fakeHeader =
new BlockHeader(
Hash.EMPTY,
Hash.EMPTY,
Address.ZERO,
Hash.EMPTY,
Hash.EMPTY,
Hash.EMPTY,
LogsBloomFilter.empty(),
Difficulty.ZERO,
number,
0,
0,
0,
Bytes.EMPTY,
null,
Hash.EMPTY,
0,
new MainnetBlockHeaderFunctions(),
Optional.of(testLogsBloomFilter));
return fakeHeader;
}
}

2
ethereum/api/src/test/java/org/hyperledger/besu/ethereum/api/query/cache/TransactionLogBloomCacherTest.java vendored
Unescape View File

@ -231,7 +231,7 @@ public class TransactionLogBloomCacherTest {
private byte[] readLogBloomCache(final File logBloom, final long number) throws IOException {
try (final RandomAccessFile randomAccessFile = new RandomAccessFile(logBloom, "r")) {
randomAccessFile.seek(BLOOM_BITS_LENGTH * number);
byte[] retrievedLog = new byte[BLOOM_BITS_LENGTH];
final byte[] retrievedLog = new byte[BLOOM_BITS_LENGTH];
randomAccessFile.read(retrievedLog);
return retrievedLog;
}

Write Preview
Loading…
Cancel
Save