From 66c4240234e741edd12232de68f06f34e6acbc14 Mon Sep 17 00:00:00 2001 From: Conner Fromknecht Date: Tue, 26 Jun 2018 18:05:39 -0700 Subject: [PATCH] htlcswitch/link: add WaitForShutdown method for testing --- htlcswitch/link.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/htlcswitch/link.go b/htlcswitch/link.go index 9a933f04..2aeb9707 100644 --- a/htlcswitch/link.go +++ b/htlcswitch/link.go @@ -465,6 +465,12 @@ func (l *channelLink) Stop() { l.wg.Wait() } +// WaitForShutdown blocks until the link finishes shutting down, which includes +// termination of all dependent goroutines. +func (l *channelLink) WaitForShutdown() { + l.wg.Wait() +} + // EligibleToForward returns a bool indicating if the channel is able to // actively accept requests to forward HTLC's. We're able to forward HTLC's if // we know the remote party's next revocation point. Otherwise, we can't