mirror of
https://github.com/lightningnetwork/lnd.git
synced 2026-08-13 12:32:48 +02:00
8047149c6a
Migrate all btcd dependencies to the new per-package v2 modules (wire/v2, txscript/v2, chaincfg/v2, chainhash/v2, btcutil/v2, psbt/v2, btcec/v2) introduced by btcd v0.26.0, and pin the tagged ecosystem versions: btcwallet v0.17.0, neutrino v0.18.0 and lightning-onion v1.4.0. The bulk of the import rewrite was produced by the scripted diff from https://github.com/btcsuite/btcd/pull/2547 (followed by 'make rpc'). The address symbols that moved out of btcutil into the new address package are imported as btcaddr where a local "address" variable would otherwise shadow them. The go.mod/go.sum updates and the remaining manual compilation fixes are folded into this single commit so it builds on its own (the migration was previously split into a reproducible scripted-diff plus follow-ups, intended to be squashed on merge).
101 lines
2 KiB
Go
101 lines
2 KiB
Go
package autopilot_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcd/btcec/v2"
|
|
"github.com/btcsuite/btcd/btcutil/v2"
|
|
"github.com/lightningnetwork/lnd/autopilot"
|
|
)
|
|
|
|
// randKey returns a random public key.
|
|
func randKey() (*btcec.PublicKey, error) {
|
|
priv, err := btcec.NewPrivateKey()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return priv.PubKey(), nil
|
|
}
|
|
|
|
// TestSetNodeScores tests that the scores returned by the
|
|
// ExternalScoreAttachment correctly reflects the scores we set last.
|
|
func TestSetNodeScores(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := t.Context()
|
|
|
|
const name = "externalscore"
|
|
|
|
h := autopilot.NewExternalScoreAttachment()
|
|
|
|
// Create a list of random node IDs.
|
|
const numKeys = 20
|
|
var pubkeys []autopilot.NodeID
|
|
for i := 0; i < numKeys; i++ {
|
|
k, err := randKey()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
nID := autopilot.NewNodeID(k)
|
|
pubkeys = append(pubkeys, nID)
|
|
}
|
|
|
|
// Set the score of half of the nodes.
|
|
scores := make(map[autopilot.NodeID]float64)
|
|
for i := 0; i < numKeys/2; i++ {
|
|
nID := pubkeys[i]
|
|
scores[nID] = 0.05 * float64(i)
|
|
}
|
|
|
|
applied, err := h.SetNodeScores(name, scores)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !applied {
|
|
t.Fatalf("scores were not applied")
|
|
}
|
|
|
|
// Query all scores, half should be set, half should be zero.
|
|
q := make(map[autopilot.NodeID]struct{})
|
|
for _, nID := range pubkeys {
|
|
q[nID] = struct{}{}
|
|
}
|
|
resp, err := h.NodeScores(
|
|
ctx, nil, nil, btcutil.Amount(btcutil.SatoshiPerBitcoin), q,
|
|
)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
for i := 0; i < numKeys; i++ {
|
|
var expected float64
|
|
if i < numKeys/2 {
|
|
expected = 0.05 * float64(i)
|
|
}
|
|
nID := pubkeys[i]
|
|
|
|
var score float64
|
|
if s, ok := resp[nID]; ok {
|
|
score = s.Score
|
|
}
|
|
|
|
if score != expected {
|
|
t.Fatalf("expected score %v, got %v",
|
|
expected, score)
|
|
}
|
|
|
|
}
|
|
|
|
// Try to apply scores with bogus name, should not be applied.
|
|
applied, err = h.SetNodeScores("dummy", scores)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if applied {
|
|
t.Fatalf("scores were applied")
|
|
}
|
|
|
|
}
|