tapdb: add test db helper func for server addresses and proof leaves

This commit adds a helper method for inserting server addresses into a
unit test db. It also adds a helper method for inserting a proof leaf
into a unit test db.

These helper methods will be used in a subsequent commit.

Author: ffranr

tapdb/sqlutils_test.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"context"
 	"database/sql"
+	"fmt"
 	"math/rand"
 	"testing"
 	"time"
@@ -15,6 +16,7 @@ import (
 	"github.com/lightninglabs/taproot-assets/internal/test"
 	"github.com/lightninglabs/taproot-assets/proof"
 	"github.com/lightninglabs/taproot-assets/tapdb/sqlc"
+	"github.com/lightninglabs/taproot-assets/universe"
 	"github.com/lightningnetwork/lnd/clock"
 	"github.com/stretchr/testify/require"
 )
@@ -158,6 +160,67 @@ func (d *DbHandler) AddRandomAssetProof(t *testing.T) (*asset.Asset,
 	return testAsset, annotatedProof
 }
 
+// AddUniProofLeaf generates a universe proof leaf and inserts it into the test
+// database.
+func (d *DbHandler) AddUniProofLeaf(t *testing.T, testAsset *asset.Asset,
+	annotatedProof *proof.AnnotatedProof) *universe.Proof {
+
+	ctx := context.Background()
+
+	// Insert proof into the multiverse/universe store. This step will
+	// populate the universe root and universe leaves tables.
+	uniId := universe.NewUniIDFromAsset(*testAsset)
+
+	leafKey := universe.LeafKey{
+		OutPoint:  annotatedProof.AssetSnapshot.OutPoint,
+		ScriptKey: &testAsset.ScriptKey,
+	}
+
+	leaf := universe.Leaf{
+		GenesisWithGroup: universe.GenesisWithGroup{
+			Genesis:  testAsset.Genesis,
+			GroupKey: testAsset.GroupKey,
+		},
+		RawProof: annotatedProof.Blob,
+		Asset:    testAsset,
+		Amt:      testAsset.Amount,
+	}
+
+	uniProof, err := d.MultiverseStore.UpsertProofLeaf(
+		ctx, uniId, leafKey, &leaf, nil,
+	)
+	require.NoError(t, err)
+
+	return uniProof
+}
+
+// AddRandomServerAddrs is a helper function that will create server addresses
+// and add them to the database.
+func (d *DbHandler) AddRandomServerAddrs(t *testing.T,
+	numServers int) []universe.ServerAddr {
+
+	var (
+		ctx   = context.Background()
+		fedDB = d.UniverseFederationStore
+	)
+
+	addrs := make([]universe.ServerAddr, 0, numServers)
+	for i := 0; i < numServers; i++ {
+		portOffset := i + 10_000
+		hostStr := fmt.Sprintf("localhost:%v", portOffset)
+
+		addr := universe.NewServerAddr(int64(i+1), hostStr)
+		addrs = append(addrs, addr)
+	}
+
+	// With the set of addrs created, we'll now insert them all into the
+	// database.
+	err := fedDB.AddServers(ctx, addrs...)
+	require.NoError(t, err)
+
+	return addrs
+}
+
 // NewDbHandle creates a new store and query handle to the test database.
 func NewDbHandle(t *testing.T) *DbHandler {
 	// Create a new test database.

tapdb/universe_federation_test.go

@@ -3,7 +3,6 @@ package tapdb
 import (
 	"context"
 	"database/sql"
-	"fmt"
 	"testing"
 	"time"
 
@@ -35,10 +34,12 @@ func newTestFederationDb(t *testing.T,
 func TestUniverseFederationCRUD(t *testing.T) {
 	t.Parallel()
 
-	testClock := clock.NewTestClock(time.Now())
-	fedDB, _ := newTestFederationDb(t, testClock)
+	var (
+		ctx = context.Background()
 
-	ctx := context.Background()
+		db    = NewDbHandle(t)
+		fedDB = db.UniverseFederationStore
+	)
 
 	// If we try to list the set of servers without any added, we should
 	// get the error we expect.
@@ -47,19 +48,7 @@ func TestUniverseFederationCRUD(t *testing.T) {
 	require.Empty(t, dbServers)
 
 	// Next, we'll try to add a new series of servers to the DB.
-	const numServers = 10
-	addrs := make([]universe.ServerAddr, 0, numServers)
-	for i := int64(0); i < numServers; i++ {
-		portOffset := i + 10_000
-		hostStr := fmt.Sprintf("localhost:%v", portOffset)
-
-		addrs = append(addrs, universe.NewServerAddr(i+1, hostStr))
-	}
-
-	// With the set of addrs created, we'll now insert them all into the
-	// database.
-	err = fedDB.AddServers(ctx, addrs...)
-	require.NoError(t, err)
+	addrs := db.AddRandomServerAddrs(t, 10)
 
 	// If we try to insert them all again, then we should get an error as
 	// we ensure the host names are unique.

universe/interface.go

@@ -84,6 +84,26 @@ func (i *Identifier) StringForLog() string {
 		i.String(), i.AssetID[:], groupKey, i.ProofType)
 }
 
+// NewUniIDFromAsset creates a new universe ID from an asset.
+func NewUniIDFromAsset(a asset.Asset) Identifier {
+	proofType := ProofTypeTransfer
+	if a.IsGenesisAsset() {
+		proofType = ProofTypeIssuance
+	}
+
+	if a.GroupKey != nil {
+		return Identifier{
+			GroupKey:  &a.GroupKey.GroupPubKey,
+			ProofType: proofType,
+		}
+	}
+
+	return Identifier{
+		AssetID:   a.ID(),
+		ProofType: proofType,
+	}
+}
+
 // NewUniIDFromRawArgs creates a new universe ID from the raw arguments. The
 // asset ID bytes and group key bytes are mutually exclusive. If the group key
 // bytes are set, then the asset ID bytes will be ignored.