package lnwire import ( "bytes" "reflect" "testing" "github.com/roasbeef/btcd/btcec" "github.com/roasbeef/btcd/chaincfg/chainhash" ) func TestChannelAnnoucementEncodeDecode(t *testing.T) { ca := &ChannelAnnouncement{ FirstNodeSig: someSig, SecondNodeSig: someSig, ChannelID: someChannelID, FirstBitcoinSig: someSig, SecondBitcoinSig: someSig, FirstNodeID: pubKey, SecondNodeID: pubKey, FirstBitcoinKey: pubKey, SecondBitcoinKey: pubKey, } // Next encode the CA message into an empty bytes buffer. var b bytes.Buffer if err := ca.Encode(&b, 0); err != nil { t.Fatalf("unable to encode ChannelAnnouncement: %v", err) } // Ensure the max payload estimate is correct. serializedLength := uint32(b.Len()) if serializedLength != ca.MaxPayloadLength(0) { t.Fatalf("payload length estimate is incorrect: expected %v "+ "got %v", serializedLength, ca.MaxPayloadLength(0)) } // Deserialize the encoded CA message into a new empty struct. ca2 := &ChannelAnnouncement{} if err := ca2.Decode(&b, 0); err != nil { t.Fatalf("unable to decode ChannelAnnouncement: %v", err) } // Assert equality of the two instances. if !reflect.DeepEqual(ca, ca2) { t.Fatalf("encode/decode error messages don't match %#v vs %#v", ca, ca2) } } func TestChannelAnnoucementValidation(t *testing.T) { getKeys := func(s string) (*btcec.PrivateKey, *btcec.PublicKey) { return btcec.PrivKeyFromBytes(btcec.S256(), []byte(s)) } firstNodePrivKey, firstNodePubKey := getKeys("node-id-1") secondNodePrivKey, secondNodePubKey := getKeys("node-id-2") firstBitcoinPrivKey, firstBitcoinPubKey := getKeys("bitcoin-key-1") secondBitcoinPrivKey, secondBitcoinPubKey := getKeys("bitcoin-key-2") hash := chainhash.DoubleHashB(firstNodePubKey.SerializeCompressed()) firstBitcoinSig, _ := firstBitcoinPrivKey.Sign(hash) hash = chainhash.DoubleHashB(secondNodePubKey.SerializeCompressed()) secondBitcoinSig, _ := secondBitcoinPrivKey.Sign(hash) ca := &ChannelAnnouncement{ ChannelID: someChannelID, FirstBitcoinSig: firstBitcoinSig, SecondBitcoinSig: secondBitcoinSig, FirstNodeID: firstNodePubKey, SecondNodeID: secondNodePubKey, FirstBitcoinKey: firstBitcoinPubKey, SecondBitcoinKey: secondBitcoinPubKey, } dataToSign, _ := ca.DataToSign() hash = chainhash.DoubleHashB(dataToSign) firstNodeSign, _ := firstNodePrivKey.Sign(hash) ca.FirstNodeSig = firstNodeSign secondNodeSign, _ := secondNodePrivKey.Sign(hash) ca.SecondNodeSig = secondNodeSign if err := ca.Validate(); err != nil { t.Fatal(err) } }