lnd.xprv/autopilot/externalscoreattach_test.go

package autopilot_test

import (
	"testing"

	"github.com/btcsuite/btcd/btcec"
	"github.com/btcsuite/btcutil"
	"github.com/lightningnetwork/lnd/autopilot"
)

// randKey returns a random public key.
func randKey() (*btcec.PublicKey, error) {
	priv, err := btcec.NewPrivateKey(btcec.S256())
	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()

	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(
		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")
	}

}