mirrors/bitcoin-s
1
0
Fork 0
mirror of https://github.com/bitcoin-s/bitcoin-s.git synced 2025-03-13 11:35:40 +01:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #155 from Christewart/zmq_fix_test

Browse source 
Making sure ZMQSubscriberTest compiles
This commit is contained in:
Chris Stewart 2018-05-01 09:51:56 -04:00 • committed by GitHub
commit 2a39b9a326
2 changed files with 13 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
zmq/src/main/scala/org/bitcoins/zmq/ZMQSubscriber.scala
View file

@ -1,15 +1,10 @@
package org.bitcoins.zmq
import java.net.{ InetSocketAddress, Socket }
import java.net.InetSocketAddress
import org.bitcoins.core.crypto.{ DoubleSha256Digest, HashDigest }
import org.bitcoins.core.protocol.blockchain.Block
import org.bitcoins.core.protocol.transaction.Transaction
import org.bitcoins.core.util.BitcoinSLogger
import org.zeromq.{ ZMQ, ZMsg }
import scala.concurrent.{ ExecutionContext, Future }
/**
* This class is designed to consume a zmq stream from a cryptocurrency's daemon.
* An example of this is bitcoind. For information on how to setup your coin's conf

38
zmq/src/test/scala/org/bitcoins/zmq/ZMQSubscriberTest.scala
View file

@ -2,15 +2,9 @@ package org.bitcoins.zmq
import java.net.InetSocketAddress
import org.bitcoins.core.crypto.DoubleSha256Digest
import org.bitcoins.core.protocol.blockchain.Block
import org.bitcoins.core.protocol.transaction.Transaction
import org.bitcoins.core.util.BitcoinSUtil
import org.scalatest.{ FlatSpec, MustMatchers }
import org.slf4j.LoggerFactory
import scala.concurrent.ExecutionContext.Implicits.global
import scala.concurrent.duration.DurationInt
import scala.concurrent.{ Await, Future }
class ZMQSubscriberTest extends FlatSpec with MustMatchers {
private val logger = LoggerFactory.getLogger(this.getClass().toString)
"ZMQSubscriber" must "connect to a regtest instance of a daemon and stream txs/blocks from it" in {
@ -22,38 +16,30 @@ class ZMQSubscriberTest extends FlatSpec with MustMatchers {
val zmqSub = new ZMQSubscriber(socket, None, None, rawTxListener, rawBlockListener)
//stupid, doesn't test anything, for now. You need to look at log output to verify this is working
Await.result(zmqSub.start(), 100.seconds)
zmqSub.start()
Thread.sleep(100000) // 100 seconds
zmqSub.stop
}
val rawBlockListener: Option[Seq[Byte] => Future[Unit]] = Some {
val rawBlockListener: Option[Seq[Byte] => Unit] = Some {
{ bytes: Seq[Byte] =>
val block = Future(Block.fromBytes(bytes))
block.map { b =>
logger.debug(s"received block $b")
Future.successful(Unit)
}
val hex = BitcoinSUtil.encodeHex(bytes)
logger.debug(s"received raw block ${hex}")
}
}
val hashBlockListener: Option[Seq[Byte] => Future[Unit]] = Some {
val hashBlockListener: Option[Seq[Byte] => Unit] = Some {
{ bytes: Seq[Byte] =>
val hash = Future(DoubleSha256Digest.fromBytes(bytes))
hash.map { h =>
logger.debug(s"received block hash $h")
Future.successful(Unit)
}
val hex = BitcoinSUtil.encodeHex(bytes)
logger.debug(s"received raw block hash ${hex}")
}
}
val rawTxListener: Option[Seq[Byte] => Future[Unit]] = Some {
val rawTxListener: Option[Seq[Byte] => Unit] = Some {
{ bytes: Seq[Byte] =>
val txFuture = Future(Transaction.fromBytes(bytes))
txFuture.map { tx =>
logger.debug(s"received tx ${tx}")
Future.successful(Unit)
}
val hex = BitcoinSUtil.encodeHex(bytes)
logger.debug(s"received raw tx ${hex}")
}
}
}