highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-07-12 14:12:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

wtclient+server: introduce tower client Manager

Browse Source 
Introduce a wtclient `Manager` which handles tower clients. It indexes
clients by the policy used. The policy field is thus removed from the
`Config` struct which configures the Manager and is instead added to a
new `towerClientCfg` which configures a specific client managed by the
manager. For now, only the `NewClient` method is added to the Manager.
It can be used to construct a new `TowerClient`. The Manager currently
does notthing with the clients added to it.
This commit is contained in:
Elle Mouton
2023-08-11 11:28:29 +02:00
parent 80684eccbd
commit ab0375e0c1
4 changed files with 204 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
watchtower/wtclient/client_test.go
View File

@ -395,15 +395,16 @@ func (c *mockChannel) getState(
}
type testHarness struct {
t *testing.T
cfg harnessCfg
signer *wtmock.MockSigner
capacity lnwire.MilliSatoshi
clientDB *wtdb.ClientDB
clientCfg *wtclient.Config
client wtclient.Client
server *serverHarness
net *mockNet
t *testing.T
cfg harnessCfg
signer *wtmock.MockSigner
capacity lnwire.MilliSatoshi
clientDB *wtdb.ClientDB
clientCfg *wtclient.Config
clientPolicy wtpolicy.Policy
client wtclient.Client
server *serverHarness
net *mockNet
blockEvents *mockBlockSub
height int32
@ -486,6 +487,7 @@ func newHarness(t *testing.T, cfg harnessCfg) *testHarness {
return &channeldb.ChannelCloseSummary{CloseHeight: height}, nil
}
h.clientPolicy = cfg.policy
h.clientCfg = &wtclient.Config{
Signer: signer,
SubscribeChannelEvents: func() (subscribe.Subscription, error) {
@ -497,7 +499,6 @@ func newHarness(t *testing.T, cfg harnessCfg) *testHarness {
DB: clientDB,
AuthDial: mockNet.AuthDial,
SecretKeyRing: wtmock.NewSecretKeyRing(),
Policy: cfg.policy,
NewAddress: func() ([]byte, error) {
return addrScript, nil
},
@ -559,7 +560,10 @@ func (h *testHarness) startClient() {
Address: towerTCPAddr,
}
h.client, err = wtclient.New(h.clientCfg)
m, err := wtclient.NewManager(h.clientCfg)
require.NoError(h.t, err)
h.client, err = m.NewClient(h.clientPolicy)
require.NoError(h.t, err)
require.NoError(h.t, h.client.Start())
require.NoError(h.t, h.client.AddTower(towerAddr))
@ -1452,9 +1456,7 @@ var clientTests = []clientTest{
// Assert that the server has updates for the clients
// most recent policy.
h.server.assertUpdatesForPolicy(
hints, h.clientCfg.Policy,
)
h.server.assertUpdatesForPolicy(hints, h.clientPolicy)
},
},
{
@ -1496,7 +1498,7 @@ var clientTests = []clientTest{
// Restart the client with a new policy, which will
// immediately try to overwrite the prior session with
// the old policy.
h.clientCfg.Policy.SweepFeeRate *= 2
h.clientPolicy.SweepFeeRate *= 2
h.startClient()
// Wait for all the updates to be populated in the
@ -1505,9 +1507,7 @@ var clientTests = []clientTest{
// Assert that the server has updates for the clients
// most recent policy.
h.server.assertUpdatesForPolicy(
hints, h.clientCfg.Policy,
)
h.server.assertUpdatesForPolicy(hints, h.clientPolicy)
},
},
{
@ -1549,10 +1549,10 @@ var clientTests = []clientTest{
// adjusting the MaxUpdates. The client should detect
// that the two policies have equivalent TxPolicies and
// continue using the first.
expPolicy := h.clientCfg.Policy
expPolicy := h.clientPolicy
// Restart the client with a new policy.
h.clientCfg.Policy.MaxUpdates = 20
h.clientPolicy.MaxUpdates = 20
h.startClient()
// Now, queue the second half of the retributions.