akovalenko/lnd.xprv
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

lnwallet: extend TestStateUpdatePersistence to assert proper htlc counters

Browse Source
This commit is contained in:
Olaoluwa Osuntokun 2017-11-10 14:30:58 -08:00
parent 32651e9311
commit 08c7fd9b4d
No known key found for this signature in database
GPG Key ID: 964EA263DD637C21
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lnwallet/channel_test.go
View File

@ -1470,8 +1470,6 @@ func TestStateUpdatePersistence(t *testing.T) {
// The state update logs of the new channels and the old channels // The state update logs of the new channels and the old channels
// should now be identical other than the height the HTLCs were added. // should now be identical other than the height the HTLCs were added.
//
// TODO(roasbeef): check for HTLC index as well!!!!!
if aliceChannel.localUpdateLog.logIndex != if aliceChannel.localUpdateLog.logIndex !=
aliceChannelNew.localUpdateLog.logIndex { aliceChannelNew.localUpdateLog.logIndex {
t.Fatalf("alice log counter: expected %v, got %v", t.Fatalf("alice log counter: expected %v, got %v",
@ -1624,6 +1622,25 @@ func TestStateUpdatePersistence(t *testing.T) {
t.Fatalf("expected %v bob satoshis sent, got %v", t.Fatalf("expected %v bob satoshis sent, got %v",
htlcAmt*3, bobChannel.channelState.TotalMSatReceived) htlcAmt*3, bobChannel.channelState.TotalMSatReceived)
} }
// As a final test, we'll ensure that the HTLC counters for both sides
// has been persisted properly. If we instruct Alice to add a new HTLC,
// it should have an index of 3. If we instruct Bob to do the
// same, it should have an index of 1.
aliceHtlcIndex, err := aliceChannel.AddHTLC(bobh)
if err != nil {
t.Fatalf("unable to add htlc: %v", err)
}
if aliceHtlcIndex != 3 {
t.Fatalf("wrong htlc index: expected %v, got %v", 3, aliceHtlcIndex)
}
bobHtlcIndex, err := bobChannel.AddHTLC(bobh)
if err != nil {
t.Fatalf("unable to add htlc: %v", err)
}
if bobHtlcIndex != 1 {
t.Fatalf("wrong htlc index: expected %v, got %v", 1, aliceHtlcIndex)
}
} }
func TestCancelHTLC(t *testing.T) { func TestCancelHTLC(t *testing.T) {