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

Improve lnd test suite reliability (#4361)

Browse source
This commit is contained in:
benthecarman 2022-06-01 14:28:35 -05:00 • committed by GitHub
parent 4fc3b05ed5
commit 54c3f77f8e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
lnd-rpc/src/main/scala/org/bitcoins/lnd/rpc/LndRpcClient.scala
View file

@ -106,11 +106,15 @@ class LndRpcClient(val instance: LndInstance, binaryOpt: Option[File] = None)(
applier: CallCredentials.MetadataApplier
): Unit = {
appExecutor.execute(() => {
val metadata = new Metadata()
val key =
Metadata.Key.of(macaroonKey, Metadata.ASCII_STRING_MARSHALLER)
metadata.put(key, instance.macaroon)
applier(metadata)
// Wrap in a try, in case the macaroon hasn't been created yet.
Try {
val metadata = new Metadata()
val key =
Metadata.Key.of(macaroonKey, Metadata.ASCII_STRING_MARSHALLER)
metadata.put(key, instance.macaroon)
applier(metadata)
}
()
})
}
@ -141,6 +145,7 @@ class LndRpcClient(val instance: LndInstance, binaryOpt: Option[File] = None)(
lazy val signer: SignerClient = SignerClient(clientSettings)
lazy val router: RouterClient = RouterClient(clientSettings)
lazy val invoices: InvoicesClient = InvoicesClient(clientSettings)
lazy val stateClient: StateClient = StateClient(clientSettings)
def genSeed(): Future[GenSeedResponse] = {
logger.trace("lnd calling genseed")
@ -973,9 +978,16 @@ class LndRpcClient(val instance: LndInstance, binaryOpt: Option[File] = None)(
def isStarted: Future[Boolean] = {
val p = Promise[Boolean]()
Try(getInfo.onComplete {
case Success(_) =>
p.success(true)
Try(stateClient.getState(GetStateRequest()).onComplete {
case Success(state) =>
state.state match {
case WalletState.RPC_ACTIVE | WalletState.SERVER_ACTIVE =>
p.success(true)
case _: WalletState.Unrecognized |
WalletState.WAITING_TO_START | WalletState.UNLOCKED |
WalletState.LOCKED | WalletState.NON_EXISTING =>
p.success(false)
}
case Failure(_) =>
p.success(false)
})