invoices: update testSpontaneousAmpPayment for new AMP logic

Rather than checking the AMP invoice state itself, we'll instead check
the HTLC state.
This commit is contained in:
Olaoluwa Osuntokun
2021-10-14 18:46:04 +02:00
parent 3935725329
commit 31336f91b0

View File

@@ -176,7 +176,8 @@ func TestSettleInvoice(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
if inv.AmtPaid != amtPaid+amtPaid+600 { if inv.AmtPaid != amtPaid+amtPaid+600 {
t.Fatal("amount incorrect") t.Fatalf("amount incorrect: expected %v got %v",
amtPaid+amtPaid+600, inv.AmtPaid)
} }
// Try to cancel. // Try to cancel.
@@ -1641,7 +1642,12 @@ func testSpontaneousAmpPayment(
checkSettleSubscription := func() { checkSettleSubscription := func() {
t.Helper() t.Helper()
settledInvoice := <-allSubscriptions.SettledInvoices settledInvoice := <-allSubscriptions.SettledInvoices
require.Equal(t, settledInvoice.State, channeldb.ContractSettled)
// Since this is an AMP invoice, the invoice state never
// changes, but the AMP state should show that the setID has
// been settled.
htlcState := settledInvoice.AMPState[setID].State
require.Equal(t, htlcState, channeldb.HtlcStateSettled)
} }
// Asserts that no invoice is published on the SettledInvoices channel // Asserts that no invoice is published on the SettledInvoices channel