multi: allow mock aux signer to customize sig jobs

This commit is contained in:
Jonathan Harvey-Buschel authored and Oliver Gugger committed 2024-10-17 17:33:14 +02:00
1 parent 395a761eb6
commit b3f953b1c4
5 files changed
+32 -12

No files matched your search

+5 -4
View File
@@ -3479,8 +3479,9 @@ func TestChanSyncOweCommitmentAuxSigner(t *testing.T) {
aliceChannel, bobChannel, err := CreateTestChannels(t, chanType)
require.NoError(t, err, "unable to create test channels")
// We'll now manually attach an aux signer to Alice's channel.
auxSigner := &MockAuxSigner{}
// We'll now manually attach an aux signer to Alice's channel. We'll
// set each aux sig job to receive an instant response.
auxSigner := NewAuxSignerMock(EmptyMockJobHandler)
aliceChannel.auxSigner = fn.Some[AuxSigner](auxSigner)
var fakeOnionBlob [lnwire.OnionPacketSize]byte
@@ -3504,8 +3505,8 @@ func TestChanSyncOweCommitmentAuxSigner(t *testing.T) {
_, err = aliceChannel.AddHTLC(h, nil)
require.NoError(t, err, "unable to recv bob's htlc: %v", err)
// We'll set up the mock to expect calls to PackSigs and also
// SubmitSubmitSecondLevelSigBatch.
// We'll set up the mock aux signer to expect calls to PackSigs and also
// SubmitSecondLevelSigBatch.
var sigBlobBuf bytes.Buffer
sigBlob := testSigBlob{
BlobInt: tlv.NewPrimitiveRecord[tlv.TlvType65634, uint16](5),