From 0686329062550520ace617535f4a52d421e981dd Mon Sep 17 00:00:00 2001 From: carla Date: Wed, 14 Apr 2021 09:19:22 +0200 Subject: [PATCH] itest/test: add settle/canceled assertions for subscribe single --- lntest/itest/lnd_hold_persistence_test.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/lntest/itest/lnd_hold_persistence_test.go b/lntest/itest/lnd_hold_persistence_test.go index 069d4e0a..387b363f 100644 --- a/lntest/itest/lnd_hold_persistence_test.go +++ b/lntest/itest/lnd_hold_persistence_test.go @@ -14,6 +14,7 @@ import ( "github.com/lightningnetwork/lnd/lntest" "github.com/lightningnetwork/lnd/lntest/wait" "github.com/lightningnetwork/lnd/lntypes" + "github.com/stretchr/testify/require" ) // testHoldInvoicePersistence tests that a sender to a hold-invoice, can be @@ -302,22 +303,33 @@ func testHoldInvoicePersistence(net *lntest.NetworkHarness, t *harnessTest) { // Settle invoices half the invoices, cancel the rest. for i, preimage := range preimages { + var expectedState lnrpc.Invoice_InvoiceState + ctxt, _ = context.WithTimeout(ctxb, defaultTimeout) if i%2 == 0 { settle := &invoicesrpc.SettleInvoiceMsg{ Preimage: preimage[:], } _, err = carol.SettleInvoice(ctxt, settle) + + expectedState = lnrpc.Invoice_SETTLED } else { hash := preimage.Hash() settle := &invoicesrpc.CancelInvoiceMsg{ PaymentHash: hash[:], } _, err = carol.CancelInvoice(ctxt, settle) + + expectedState = lnrpc.Invoice_CANCELED } if err != nil { t.Fatalf("unable to cancel/settle invoice: %v", err) } + + stream := invoiceStreams[i] + invoice, err := stream.Recv() + require.NoError(t.t, err) + require.Equal(t.t, expectedState, invoice.State) } // Make sure we get the expected status update.