highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-05-30 01:30:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

routing/chainview: refactor TestFilteredChainView

Browse Source 
So each test has its own miner and chainView.
This commit is contained in:
yyforyongyu 2025-02-25 22:12:25 +08:00
parent 4bfcfea2ee
commit 56fa3aae34
No known key found for this signature in database
GPG Key ID: 9BCD95C4FF296868
1 changed files with 45 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
routing/chainview/interface_test.go
View File

@ -160,9 +160,40 @@ func assertFilteredBlock(t *testing.T, fb *FilteredBlock, expectedHeight int32,
}
func testFilterBlockNotifications(node *rpctest.Harness,
chainView FilteredChainView, chainViewInit chainViewInitFunc,
t *testing.T) {
// setupMinerAndChainView creates a miner node and initialize a chain view using
// this miner node. It returns the miner and the chain view.
func setupMinerAndChainView(t *testing.T, chainViewInit chainViewInitFunc) (
*rpctest.Harness, FilteredChainView) {
// Initialize the harness around a btcd node which will serve as our
// dedicated miner to generate blocks, cause re-orgs, etc. We'll set up
// this node with a chain length of 125, so we have plenty of BTC to
// play around with.
miner := unittest.NewMiner(
t, netParams, []string{"--txindex"}, true, 25,
)
rpcConfig := miner.RPCConfig()
_, bestHeight, err := miner.Client.GetBestBlock()
require.NoError(t, err)
chainView, err := chainViewInit(t, rpcConfig, miner, bestHeight)
require.NoError(t, err)
err = chainView.Start()
require.NoError(t, err)
t.Cleanup(func() {
require.NoError(t, chainView.Stop())
})
return miner, chainView
}
func testFilterBlockNotifications(t *testing.T,
chainViewInit chainViewInitFunc) {
node, chainView := setupMinerAndChainView(t, chainViewInit)
// To start the test, we'll create to fresh outputs paying to the
// private key that we generated above.
@ -270,9 +301,8 @@ func testFilterBlockNotifications(node *rpctest.Harness,
}
}
func testUpdateFilterBackTrack(node *rpctest.Harness,
chainView FilteredChainView, chainViewInit chainViewInitFunc,
t *testing.T) {
func testUpdateFilterBackTrack(t *testing.T, chainViewInit chainViewInitFunc) {
node, chainView := setupMinerAndChainView(t, chainViewInit)
// To start, we'll create a fresh output paying to the height generated
// above.
@ -345,8 +375,8 @@ func testUpdateFilterBackTrack(node *rpctest.Harness,
}
}
func testFilterSingleBlock(node *rpctest.Harness, chainView FilteredChainView,
chainViewInit chainViewInitFunc, t *testing.T) {
func testFilterSingleBlock(t *testing.T, chainViewInit chainViewInitFunc) {
node, chainView := setupMinerAndChainView(t, chainViewInit)
// In this test, we'll test the manual filtration of blocks, which can
// be used by clients to manually rescan their sub-set of the UTXO set.
@ -445,9 +475,10 @@ func testFilterSingleBlock(node *rpctest.Harness, chainView FilteredChainView,
// testFilterBlockDisconnected triggers a reorg all the way back to genesis,
// and a small 5 block reorg, ensuring the chainView notifies about
// disconnected and connected blocks in the order we expect.
func testFilterBlockDisconnected(node *rpctest.Harness,
chainView FilteredChainView, chainViewInit chainViewInitFunc,
t *testing.T) {
func testFilterBlockDisconnected(t *testing.T,
chainViewInit chainViewInitFunc) {
node, _ := setupMinerAndChainView(t, chainViewInit)
// Create a node that has a shorter chain than the main chain, so we
// can trigger a reorg.
@ -636,8 +667,7 @@ type chainViewInitFunc func(t *testing.T, rpcInfo rpcclient.ConnConfig,
type testCase struct {
name string
test func(*rpctest.Harness, FilteredChainView, chainViewInitFunc,
*testing.T)
test func(*testing.T, chainViewInitFunc)
}
var chainViewTests = []testCase{
@ -793,41 +823,17 @@ var interfaceImpls = []struct {
}
func TestFilteredChainView(t *testing.T) {
// Initialize the harness around a btcd node which will serve as our
// dedicated miner to generate blocks, cause re-orgs, etc. We'll set up
// this node with a chain length of 125, so we have plenty of BTC to
// play around with.
miner := unittest.NewMiner(
t, netParams, []string{"--txindex"}, true, 25,
)
rpcConfig := miner.RPCConfig()
for _, chainViewImpl := range interfaceImpls {
t.Logf("Testing '%v' implementation of FilteredChainView",
chainViewImpl.name)
_, bestHeight, err := miner.Client.GetBestBlock()
if err != nil {
t.Fatalf("error getting best block: %v", err)
}
chainView, err := chainViewImpl.chainViewInit(
t, rpcConfig, miner, bestHeight,
)
if err != nil {
t.Fatalf("unable to make chain view: %v", err)
}
if err := chainView.Start(); err != nil {
t.Fatalf("unable to start chain view: %v", err)
}
for _, chainViewTest := range chainViewTests {
testName := fmt.Sprintf("%v: %v", chainViewImpl.name,
chainViewTest.name)
success := t.Run(testName, func(t *testing.T) {
chainViewTest.test(
miner, chainView,
chainViewImpl.chainViewInit, t,
t, chainViewImpl.chainViewInit,
)
})
@ -836,8 +842,5 @@ func TestFilteredChainView(t *testing.T) {
}
}
if err := chainView.Stop(); err != nil {
t.Fatalf("unable to stop chain view: %v", err)
}
}
}