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
No known key found for this signature in database
GPG key ID: 3BBD59E99B280306

View file

@ -176,7 +176,8 @@ func TestSettleInvoice(t *testing.T) {
t.Fatal(err)
}
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.
@ -1641,7 +1642,12 @@ func testSpontaneousAmpPayment(
checkSettleSubscription := func() {
t.Helper()
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