mirror of
https://github.com/lightningnetwork/lnd.git
synced 2024-11-19 09:53:54 +01:00
d109bd9298
This commit adds a new bucket to the database which is dedicated to storing data pertaining to p2p related reachability for direct channel counter parties. The data stored in this new bucket can be used within heuristics when deciding to unilaterally close a channel due to inactivity. Additionally, all known reachable IP addresses for a particular LinkNode are to be stored and updated within the database in order to facilitate the establishment of persistent connections to direct channel counter parties.
109 lines
3.4 KiB
Go
109 lines
3.4 KiB
Go
package channeldb
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/roasbeef/btcd/btcec"
|
|
"github.com/roasbeef/btcd/wire"
|
|
)
|
|
|
|
func TestLinkNodeEncodeDecode(t *testing.T) {
|
|
cdb, cleanUp, err := makeTestDB()
|
|
if err != nil {
|
|
t.Fatalf("uanble to make test database: %v", err)
|
|
}
|
|
defer cleanUp()
|
|
|
|
// First we'll create some initial data to use for populating our test
|
|
// LinkNode instances.
|
|
_, pub1 := btcec.PrivKeyFromBytes(btcec.S256(), key[:])
|
|
_, pub2 := btcec.PrivKeyFromBytes(btcec.S256(), rev[:])
|
|
addr1, err := net.ResolveTCPAddr("tcp", "10.0.0.1:9000")
|
|
if err != nil {
|
|
t.Fatalf("unable to create test addr: %v", err)
|
|
}
|
|
addr2, err := net.ResolveTCPAddr("tcp", "10.0.0.2:9000")
|
|
if err != nil {
|
|
t.Fatalf("unable to create test addr: %v", err)
|
|
}
|
|
|
|
// Create two fresh link node instances with the above dummy data, then
|
|
// fully sync both instances to disk.
|
|
node1 := cdb.NewLinkNode(wire.MainNet, pub1, addr1)
|
|
node2 := cdb.NewLinkNode(wire.TestNet3, pub2, addr2)
|
|
if err := node1.Sync(); err != nil {
|
|
t.Fatalf("unable to sync node: %v", err)
|
|
}
|
|
if err := node2.Sync(); err != nil {
|
|
t.Fatalf("unable to sync node: %v", err)
|
|
}
|
|
|
|
// Fetch all current link nodes from the database, they should exactly
|
|
// match the two created above.
|
|
originalNodes := []*LinkNode{node2, node1}
|
|
linkNodes, err := cdb.FetchAllLinkNodes()
|
|
if err != nil {
|
|
t.Fatalf("unable to fetch nodes: %v", err)
|
|
}
|
|
for i, node := range linkNodes {
|
|
if originalNodes[i].Network != node.Network {
|
|
t.Fatalf("node networks don't match: expected %v, got %v",
|
|
originalNodes[i].Network, node.Network)
|
|
}
|
|
|
|
originalPubkey := originalNodes[i].IdentityPub.SerializeCompressed()
|
|
dbPubkey := node.IdentityPub.SerializeCompressed()
|
|
if !bytes.Equal(originalPubkey, dbPubkey) {
|
|
t.Fatalf("node pubkeys don't match: expected %x, got %x",
|
|
originalPubkey, dbPubkey)
|
|
}
|
|
if originalNodes[i].LastSeen.Unix() != node.LastSeen.Unix() {
|
|
t.Fatalf("last seen timestamps don't match: expected %v got %v",
|
|
originalNodes[i].LastSeen.Unix(), node.LastSeen.Unix())
|
|
}
|
|
if !reflect.DeepEqual(originalNodes[i].Addresses,
|
|
node.Addresses) {
|
|
t.Fatalf("addresses don't match: expected %v, got %v",
|
|
originalNodes[i].Addresses, node.Addresses)
|
|
}
|
|
}
|
|
|
|
// Next, we'll excercise the methods to append additionall IP
|
|
// addresses, and also to update the last seen time.
|
|
if err := node1.UpdateLastSeen(time.Now()); err != nil {
|
|
t.Fatalf("unable to update last seen: %v", err)
|
|
}
|
|
if err := node1.AddAddress(addr2); err != nil {
|
|
t.Fatalf("unable to update addr: %v", err)
|
|
}
|
|
|
|
// Fetch the same node from the databse according to its public key.
|
|
node1DB, err := cdb.FetchLinkNode(pub1)
|
|
if err != nil {
|
|
t.Fatalf("unable to find node: %v", err)
|
|
}
|
|
|
|
// Both the last seen timestamp and the list of reachable addresses for
|
|
// the node should be updated.
|
|
if node1DB.LastSeen.Unix() != node1.LastSeen.Unix() {
|
|
t.Fatalf("last seen timestamps don't match: expected %v got %v",
|
|
node1.LastSeen.Unix(), node1DB.LastSeen.Unix())
|
|
}
|
|
if len(node1DB.Addresses) != 2 {
|
|
t.Fatalf("wrong length for node1 addrsses: expected %v, got %v",
|
|
2, len(node1DB.Addresses))
|
|
}
|
|
if node1DB.Addresses[0].String() != addr1.String() {
|
|
t.Fatalf("wrong address for node: expected %v, got %v",
|
|
addr1.String(), node1DB.Addresses[0].String())
|
|
}
|
|
if node1DB.Addresses[1].String() != addr2.String() {
|
|
t.Fatalf("wrong address for node: expected %v, got %v",
|
|
addr2.String(), node1DB.Addresses[1].String())
|
|
}
|
|
}
|