Some API changes for Eclair test kit

This commit is contained in:
rorp 2019-07-10 14:04:54 -07:00
parent 30e6d7030f
commit 684607f510

View file

@ -11,10 +11,11 @@ import org.bitcoins.core.protocol.ln.channel.{ChannelId, ChannelState, FundedCha
import org.bitcoins.core.protocol.ln.currency.MilliSatoshis import org.bitcoins.core.protocol.ln.currency.MilliSatoshis
import org.bitcoins.core.protocol.ln.node.NodeId import org.bitcoins.core.protocol.ln.node.NodeId
import org.bitcoins.core.util.BitcoinSLogger import org.bitcoins.core.util.BitcoinSLogger
import org.bitcoins.eclair.rpc.api.EclairApi
import org.bitcoins.eclair.rpc.client.EclairRpcClient import org.bitcoins.eclair.rpc.client.EclairRpcClient
import org.bitcoins.eclair.rpc.config.EclairInstance import org.bitcoins.eclair.rpc.config.EclairInstance
import org.bitcoins.eclair.rpc.json.{PaymentId, PaymentStatus} import org.bitcoins.eclair.rpc.json.{PaymentId, PaymentStatus}
import org.bitcoins.rpc.client.common.{BitcoindRpcClient} import org.bitcoins.rpc.client.common.BitcoindRpcClient
import org.bitcoins.rpc.config.BitcoindInstance import org.bitcoins.rpc.config.BitcoindInstance
import org.bitcoins.rpc.util.RpcUtil import org.bitcoins.rpc.util.RpcUtil
import org.bitcoins.testkit.async.TestAsyncUtil import org.bitcoins.testkit.async.TestAsyncUtil
@ -159,7 +160,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
if (bitcoindRpcOpt.isDefined) { if (bitcoindRpcOpt.isDefined) {
Future.successful(bitcoindRpcOpt.get) Future.successful(bitcoindRpcOpt.get)
} else { } else {
BitcoindRpcTestUtil.startedBitcoindRpcClient() EclairRpcTestUtil.startedBitcoindRpcClient()
} }
} }
@ -193,18 +194,18 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
* @param client * @param client
* @param chanId * @param chanId
*/ */
def awaitUntilChannelNormal(client: EclairRpcClient, chanId: ChannelId)( def awaitUntilChannelNormal(client: EclairApi, chanId: ChannelId)(
implicit system: ActorSystem): Future[Unit] = { implicit system: ActorSystem): Future[Unit] = {
awaitUntilChannelState(client, chanId, ChannelState.NORMAL) awaitUntilChannelState(client, chanId, ChannelState.NORMAL)
} }
def awaitUntilChannelClosing(client: EclairRpcClient, chanId: ChannelId)( def awaitUntilChannelClosing(client: EclairApi, chanId: ChannelId)(
implicit system: ActorSystem): Future[Unit] = { implicit system: ActorSystem): Future[Unit] = {
awaitUntilChannelState(client, chanId, ChannelState.CLOSING) awaitUntilChannelState(client, chanId, ChannelState.CLOSING)
} }
private def awaitUntilChannelState( private def awaitUntilChannelState(
client: EclairRpcClient, client: EclairApi,
chanId: ChannelId, chanId: ChannelId,
state: ChannelState)(implicit system: ActorSystem): Future[Unit] = { state: ChannelState)(implicit system: ActorSystem): Future[Unit] = {
logger.debug(s"Awaiting ${chanId} to enter ${state} state") logger.debug(s"Awaiting ${chanId} to enter ${state} state")
@ -224,7 +225,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
} }
def awaitUntilPaymentSucceeded( def awaitUntilPaymentSucceeded(
client: EclairRpcClient, client: EclairApi,
paymentId: PaymentId, paymentId: PaymentId,
duration: FiniteDuration = 1.second, duration: FiniteDuration = 1.second,
maxTries: Int = 50, maxTries: Int = 50,
@ -234,7 +235,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
} }
def awaitUntilPaymentFailed( def awaitUntilPaymentFailed(
client: EclairRpcClient, client: EclairApi,
paymentId: PaymentId, paymentId: PaymentId,
duration: FiniteDuration = 1.second, duration: FiniteDuration = 1.second,
maxTries: Int = 50, maxTries: Int = 50,
@ -244,7 +245,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
} }
def awaitUntilPaymentPending( def awaitUntilPaymentPending(
client: EclairRpcClient, client: EclairApi,
paymentId: PaymentId, paymentId: PaymentId,
duration: FiniteDuration = 1.second, duration: FiniteDuration = 1.second,
maxTries: Int = 50, maxTries: Int = 50,
@ -254,7 +255,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
} }
private def awaitUntilPaymentStatus( private def awaitUntilPaymentStatus(
client: EclairRpcClient, client: EclairApi,
paymentId: PaymentId, paymentId: PaymentId,
state: PaymentStatus, state: PaymentStatus,
duration: FiniteDuration, duration: FiniteDuration,
@ -292,7 +293,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
if (bitcoindRpcClient.isDefined) { if (bitcoindRpcClient.isDefined) {
Future.successful(bitcoindRpcClient.get) Future.successful(bitcoindRpcClient.get)
} else { } else {
BitcoindRpcTestUtil.startedBitcoindRpcClient() EclairRpcTestUtil.startedBitcoindRpcClient()
} }
} }
@ -413,7 +414,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
if (bitcoindRpcClientOpt.isDefined) { if (bitcoindRpcClientOpt.isDefined) {
Future.successful(bitcoindRpcClientOpt.get) Future.successful(bitcoindRpcClientOpt.get)
} else { } else {
BitcoindRpcTestUtil.startedBitcoindRpcClient() EclairRpcTestUtil.startedBitcoindRpcClient()
} }
} }
@ -448,7 +449,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
connectedLnF connectedLnF
} }
def connectLNNodes(client: EclairRpcClient, otherClient: EclairRpcClient)( def connectLNNodes(client: EclairApi, otherClient: EclairApi)(
implicit implicit
system: ActorSystem): Future[Unit] = { system: ActorSystem): Future[Unit] = {
implicit val dispatcher = system.dispatcher implicit val dispatcher = system.dispatcher
@ -487,8 +488,8 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
* <i>something<i/>. * <i>something<i/>.
*/ */
def sendPayments( def sendPayments(
c1: EclairRpcClient, c1: EclairApi,
c2: EclairRpcClient, c2: EclairApi,
numPayments: Int = 5)( numPayments: Int = 5)(
implicit ec: ExecutionContext): Future[Vector[PaymentId]] = { implicit ec: ExecutionContext): Future[Vector[PaymentId]] = {
val payments = (1 to numPayments) val payments = (1 to numPayments)
@ -574,7 +575,7 @@ trait EclairRpcTestUtil extends BitcoinSLogger {
openedF openedF
} }
def awaitChannelOpened(client1: EclairRpcClient, chanId: ChannelId)( def awaitChannelOpened(client1: EclairApi, chanId: ChannelId)(
implicit system: ActorSystem): Future[Unit] = { implicit system: ActorSystem): Future[Unit] = {
EclairRpcTestUtil.awaitUntilChannelNormal(client1, chanId) EclairRpcTestUtil.awaitUntilChannelNormal(client1, chanId)
} }