mirrors/bitcoin-s
1
0
Fork 0
mirror of https://github.com/bitcoin-s/bitcoin-s.git synced 2025-02-23 14:50:42 +01:00
Code Issues Projects Releases Packages Wiki Activity

Attempt to batch checking of headers in chainTest to avoid reject exe… (#1826)

Browse source 
* Attempt to batch checking of headers in chainTest to avoid reject execution exceptions in the scala slick database queue

* Fix 2.12 compile issue
This commit is contained in:
Chris Stewart 2020-08-15 07:23:26 -05:00 • committed by GitHub
parent 2445003503
commit 48e3cb8e14
1 changed files with 35 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
chain-test/src/test/scala/org/bitcoins/chain/pow/BitcoinPowTest.scala
View file

@ -3,17 +3,19 @@ package org.bitcoins.chain.pow
import akka.actor.ActorSystem
import org.bitcoins.chain.blockchain.Blockchain
import org.bitcoins.chain.config.ChainAppConfig
import org.bitcoins.chain.models.BlockHeaderDAO
import org.bitcoins.core.protocol.blockchain.{
MainNetChainParams,
TestNetChainParams
}
import org.bitcoins.core.util.FutureUtil
import org.bitcoins.testkit.chain.fixture.{ChainFixture, ChainFixtureTag}
import org.bitcoins.testkit.chain.{
ChainDbUnitTest,
ChainTestUtil,
ChainUnitTest
}
import org.scalatest.FutureOutcome
import org.scalatest.{Assertion, FutureOutcome}
import scala.concurrent.Future
@ -62,24 +64,17 @@ class BitcoinPowTest extends ChainDbUnitTest {
case ChainFixture.PopulatedBlockHeaderDAO(blockHeaderDAO) =>
val iterations = 4200
// We must start after the first POW change to avoid looking for a block we don't have
val assertionFs =
(ChainUnitTest.FIRST_POW_CHANGE + 1 until ChainUnitTest.FIRST_POW_CHANGE + 1 + iterations)
.map { height =>
val blockF = blockHeaderDAO.getAtHeight(height).map(_.head)
val blockchainF =
blockF.flatMap(b => blockHeaderDAO.getBlockchainFrom(b))
val nextBlockF = blockHeaderDAO.getAtHeight(height + 1).map(_.head)
val iterator =
(ChainUnitTest.FIRST_POW_CHANGE + 1)
.until(ChainUnitTest.FIRST_POW_CHANGE + 1 + iterations)
.toVector
val assertionFs: Future[Assertion] = FutureUtil
.batchExecute(elements = iterator,
f = batchCheckHeaderPOW(_: Vector[Int], blockHeaderDAO),
init = succeed,
batchSize = 1000)
for {
blockchain <- blockchainF
nextTip <- nextBlockF
nextNBits =
Pow.getNetworkWorkRequired(nextTip.blockHeader, blockchain)
} yield assert(nextNBits == nextTip.nBits)
}
val seqF = Future.sequence(assertionFs)
seqF.map(_ => succeed)
assertionFs
}
it must "getBlockProof correctly for the testnet genesis block" inFixtured {
@ -101,4 +96,26 @@ class BitcoinPowTest extends ChainDbUnitTest {
assert(proof == BigInt(4295032833L))
}
}
/** Helper method to check headers proof of work in batches */
private def batchCheckHeaderPOW(
iterator: Vector[Int],
blockHeaderDAO: BlockHeaderDAO): Future[Assertion] = {
val nestedAssertions: Vector[Future[Assertion]] = {
iterator.map { height =>
val blockF = blockHeaderDAO.getAtHeight(height + 1).map(_.head)
val blockchainF =
blockF.flatMap(b => blockHeaderDAO.getBlockchainFrom(b))
for {
blockchain <- blockchainF
nextTip = blockchain.head
chain = Blockchain.fromHeaders(blockchain.tail.toVector)
nextNBits = Pow.getNetworkWorkRequired(nextTip.blockHeader, chain)
} yield assert(nextNBits == nextTip.nBits)
}
}
Future
.sequence(nestedAssertions)
.map(_ => succeed)
}
}