68 lines
1.9 KiB
Go
68 lines
1.9 KiB
Go
|
package netann
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/btcsuite/btcd/chaincfg/chainhash"
|
||
|
|
"github.com/btcsuite/btcd/wire"
|
||
|
|
"github.com/btcsuite/btcutil"
|
||
|
|
"github.com/lightningnetwork/lnd/channeldb"
|
||
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCreateChanAnnouncement(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
key := [33]byte{0x1}
|
||
|
|
sig := lnwire.Sig{0x1}
|
||
|
|
features := lnwire.NewRawFeatureVector(lnwire.AnchorsRequired)
|
||
|
|
var featuresBuf bytes.Buffer
|
||
|
|
if err := features.Encode(&featuresBuf); err != nil {
|
||
|
|
t.Fatalf("unable to encode features: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
expChanAnn := &lnwire.ChannelAnnouncement{
|
||
|
|
ChainHash: chainhash.Hash{0x1},
|
||
|
|
ShortChannelID: lnwire.ShortChannelID{BlockHeight: 1},
|
||
|
|
NodeID1: key,
|
||
|
|
NodeID2: key,
|
||
|
|
NodeSig1: sig,
|
||
|
|
NodeSig2: sig,
|
||
|
|
BitcoinKey1: key,
|
||
|
|
BitcoinKey2: key,
|
||
|
|
BitcoinSig1: sig,
|
||
|
|
BitcoinSig2: sig,
|
||
|
|
Features: features,
|
||
|
|
ExtraOpaqueData: []byte{0x1},
|
||
|
|
}
|
||
|
|
|
||
|
|
chanProof := &channeldb.ChannelAuthProof{
|
||
|
|
NodeSig1Bytes: expChanAnn.NodeSig1.ToSignatureBytes(),
|
||
|
|
NodeSig2Bytes: expChanAnn.NodeSig2.ToSignatureBytes(),
|
||
|
|
BitcoinSig1Bytes: expChanAnn.BitcoinSig1.ToSignatureBytes(),
|
||
|
|
BitcoinSig2Bytes: expChanAnn.BitcoinSig2.ToSignatureBytes(),
|
||
|
|
}
|
||
|
|
chanInfo := &channeldb.ChannelEdgeInfo{
|
||
|
|
ChainHash: expChanAnn.ChainHash,
|
||
|
|
ChannelID: expChanAnn.ShortChannelID.ToUint64(),
|
||
|
|
ChannelPoint: wire.OutPoint{Index: 1},
|
||
|
|
Capacity: btcutil.SatoshiPerBitcoin,
|
||
|
|
NodeKey1Bytes: key,
|
||
|
|
NodeKey2Bytes: key,
|
||
|
|
BitcoinKey1Bytes: key,
|
||
|
|
BitcoinKey2Bytes: key,
|
||
|
|
Features: featuresBuf.Bytes(),
|
||
|
|
ExtraOpaqueData: expChanAnn.ExtraOpaqueData,
|
||
|
|
}
|
||
|
|
chanAnn, _, _, err := CreateChanAnnouncement(
|
||
|
|
chanProof, chanInfo, nil, nil,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("unable to create channel announcement: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Equal(t, chanAnn, expChanAnn)
|
||
|
|
}
|