39c279b639
This commit adds a new feature to the network harness: test writers are now able to select arbitrary nodes, causing them to restart. This functionality will be useful in the future in order to test scenarios like persisting data across restarts, re-syncing after re-connections, reacting to the counter party broadcasting revoked states, etc.
76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/lightningnetwork/lnd/lnrpc"
|
|
"github.com/roasbeef/btcd/rpctest"
|
|
"github.com/roasbeef/btcrpcclient"
|
|
)
|
|
|
|
// TODO(roasbeef): randomize ports so can start multiple instance with each
|
|
// other, will need to catch up to rpctest upstream.
|
|
|
|
func TestNodeRestart(t *testing.T) {
|
|
// Create a new instance of the network harness in order to initialize the
|
|
// necessary state. We'll also need to create a temporary instance of
|
|
// rpctest due to initialization dependancies.
|
|
lndHarness, err := newNetworkHarness()
|
|
if err != nil {
|
|
t.Fatalf("unable to create lightning network harness: %v", err)
|
|
}
|
|
defer lndHarness.TearDownAll()
|
|
|
|
handlers := &btcrpcclient.NotificationHandlers{
|
|
OnTxAccepted: lndHarness.OnTxAccepted,
|
|
}
|
|
|
|
btcdHarness, err := rpctest.New(harnessNetParams, handlers, nil)
|
|
if err != nil {
|
|
t.Fatalf("unable to create mining node: %v", err)
|
|
}
|
|
defer btcdHarness.TearDown()
|
|
if err := btcdHarness.SetUp(true, 50); err != nil {
|
|
t.Fatalf("unable to set up mining node: %v", err)
|
|
}
|
|
if err := btcdHarness.Node.NotifyNewTransactions(false); err != nil {
|
|
t.Fatalf("unable to request transaction notifications: %v", err)
|
|
}
|
|
|
|
if err := lndHarness.InitializeSeedNodes(btcdHarness, nil); err != nil {
|
|
t.Fatalf("unable to initialize seed nodes: %v", err)
|
|
}
|
|
if err = lndHarness.SetUp(); err != nil {
|
|
t.Fatalf("unable to set up test lightning network: %v", err)
|
|
}
|
|
|
|
// With the harness set up, we can test the node restart method,
|
|
// asserting all data is properly persisted and recovered.
|
|
ctx := context.Background()
|
|
|
|
// First, we'll test restarting one of the initial seed nodes: Alice.
|
|
alice := lndHarness.Alice
|
|
aliceInfo, err := alice.GetInfo(ctx, &lnrpc.GetInfoRequest{})
|
|
if err != nil {
|
|
t.Fatalf("unable to query for alice's info: %v", err)
|
|
}
|
|
|
|
// With alice's node information stored, attempt to restart the node.
|
|
if lndHarness.RestartNode(alice); err != nil {
|
|
t.Fatalf("unable to resart node: %v", err)
|
|
}
|
|
|
|
// Query for alice's current information, it should be identical to
|
|
// what we received above.
|
|
aliceInfoRestart, err := alice.GetInfo(ctx, &lnrpc.GetInfoRequest{})
|
|
if err != nil {
|
|
t.Fatalf("unable to query for alice's info: %v", err)
|
|
}
|
|
|
|
if aliceInfo.IdentityPubkey != aliceInfoRestart.IdentityPubkey {
|
|
t.Fatalf("node info after restart doesn't match: %v vs %v",
|
|
aliceInfo.IdentityPubkey, aliceInfoRestart.IdentityPubkey)
|
|
}
|
|
}
|