91 lines
1.7 KiB
Go
91 lines
1.7 KiB
Go
|
package htlcswitch
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"math/rand"
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/davecgh/go-spew/spew"
|
||
|
|
"github.com/lightningnetwork/lnd/lntypes"
|
||
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TestNetworkResultSerialization checks that NetworkResults are properly
|
||
|
|
// (de)serialized.
|
||
|
|
func TestNetworkResultSerialization(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
var preimage lntypes.Preimage
|
||
|
|
if _, err := rand.Read(preimage[:]); err != nil {
|
||
|
|
t.Fatalf("unable gen rand preimag: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var chanID lnwire.ChannelID
|
||
|
|
if _, err := rand.Read(chanID[:]); err != nil {
|
||
|
|
t.Fatalf("unable gen rand chanid: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var reason [256]byte
|
||
|
|
if _, err := rand.Read(reason[:]); err != nil {
|
||
|
|
t.Fatalf("unable gen rand reason: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
settle := &lnwire.UpdateFulfillHTLC{
|
||
|
|
ChanID: chanID,
|
||
|
|
ID: 2,
|
||
|
|
PaymentPreimage: preimage,
|
||
|
|
}
|
||
|
|
|
||
|
|
fail := &lnwire.UpdateFailHTLC{
|
||
|
|
ChanID: chanID,
|
||
|
|
ID: 1,
|
||
|
|
Reason: []byte{},
|
||
|
|
}
|
||
|
|
|
||
|
|
fail2 := &lnwire.UpdateFailHTLC{
|
||
|
|
ChanID: chanID,
|
||
|
|
ID: 1,
|
||
|
|
Reason: reason[:],
|
||
|
|
}
|
||
|
|
|
||
|
|
testCases := []*networkResult{
|
||
|
|
{
|
||
|
|
msg: settle,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
msg: fail,
|
||
|
|
unencrypted: false,
|
||
|
|
isResolution: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
msg: fail,
|
||
|
|
unencrypted: false,
|
||
|
|
isResolution: true,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
msg: fail2,
|
||
|
|
unencrypted: true,
|
||
|
|
isResolution: false,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, p := range testCases {
|
||
|
|
var buf bytes.Buffer
|
||
|
|
if err := serializeNetworkResult(&buf, p); err != nil {
|
||
|
|
t.Fatalf("serialize failed: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
r := bytes.NewReader(buf.Bytes())
|
||
|
|
p1, err := deserializeNetworkResult(r)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("unable to deserizlize: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if !reflect.DeepEqual(p, p1) {
|
||
|
|
t.Fatalf("not equal. %v vs %v", spew.Sdump(p),
|
||
|
|
spew.Sdump(p1))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|