From d2c168945b62bbb8d6a186e1c255138f3f31a5fe Mon Sep 17 00:00:00 2001
From: Boris Nagaev <bnagaev@gmail.com>
Date: Mon, 28 Apr 2025 02:07:40 -0300
Subject: [PATCH] test/chainnotifier: send to specific spend reg

This is needed to have multiple spending registrations running and to send
a notification to a specific spending registration.
---
 test/chainnotifier_mock.go | 25 +++++++++++++++++--------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/test/chainnotifier_mock.go b/test/chainnotifier_mock.go
index 96e72f23..a4fae0e7 100644
--- a/test/chainnotifier_mock.go
+++ b/test/chainnotifier_mock.go
@@ -33,10 +33,11 @@ func (c *mockChainNotifier) RawClientWithMacAuth(
 
 // SpendRegistration contains registration details.
 type SpendRegistration struct {
-	Outpoint   *wire.OutPoint
-	PkScript   []byte
-	HeightHint int32
-	ErrChan    chan<- error
+	Outpoint     *wire.OutPoint
+	PkScript     []byte
+	HeightHint   int32
+	SpendChannel chan<- *chainntnfs.SpendDetail
+	ErrChan      chan<- error
 }
 
 // ConfRegistration contains registration details.
@@ -53,13 +54,15 @@ func (c *mockChainNotifier) RegisterSpendNtfn(ctx context.Context,
 	outpoint *wire.OutPoint, pkScript []byte, heightHint int32) (
 	chan *chainntnfs.SpendDetail, chan error, error) {
 
+	spendChan0 := make(chan *chainntnfs.SpendDetail)
 	spendErrChan := make(chan error, 1)
 
 	reg := &SpendRegistration{
-		HeightHint: heightHint,
-		Outpoint:   outpoint,
-		PkScript:   pkScript,
-		ErrChan:    spendErrChan,
+		HeightHint:   heightHint,
+		Outpoint:     outpoint,
+		PkScript:     pkScript,
+		SpendChannel: spendChan0,
+		ErrChan:      spendErrChan,
 	}
 
 	c.lnd.RegisterSpendChannel <- reg
@@ -78,6 +81,12 @@ func (c *mockChainNotifier) RegisterSpendNtfn(ctx context.Context,
 			case <-ctx.Done():
 			}
 
+		case m := <-spendChan0:
+			select {
+			case spendChan <- m:
+			case <-ctx.Done():
+			}
+
 		case err := <-spendErrChan:
 			select {
 			case errChan <- err: