fixup! itest: add test to check set/query for federation sync config

itest/universe_test.go

@@ -6,6 +6,7 @@ import (
 	"crypto/sha256"
 	"encoding/hex"
 	"fmt"
+	"github.com/lightninglabs/taproot-assets/universe"
 	"io"
 	prand "math/rand"
 
@@ -511,7 +512,12 @@ func testUniverseFederation(t *harnessTest) {
 // testFederationSyncConfig tests that we can properly set and query the
 // federation sync config.
 func testFederationSyncConfig(t *harnessTest) {
-	ctxb := context.Background()
+	var (
+		ctx = context.Background()
+
+		excludeAllProofs = universe.SyncProofTypesNone
+		issuanceOnly     = universe.SyncProofTypesIssuance
+	)
 
 	// Generate a random asset ID in order to generate a universe ID.
 	rand := prand.New(prand.NewSource(1))
@@ -522,36 +528,40 @@ func testFederationSyncConfig(t *harnessTest) {
 
 	// Set both the general and a universe specific federation sync configs.
 	_, err := t.tapd.UniverseClient.SetFederationSyncConfig(
-		ctxb, &unirpc.SetFederationSyncConfigRequest{
+		ctx, &unirpc.SetFederationSyncConfigRequest{
 			GeneralSyncConfig: &unirpc.GeneralFederationSyncConfig{
-				OnlySyncIssuanceProofs: true,
+				ProofTypes: issuanceOnly.String(),
 			},
 			AssetSyncConfig: []*unirpc.AssetFederationSyncConfig{
 				{
-					Id:      uniIdRpc,
-					Exclude: true,
+					Id:         uniIdRpc,
+					ProofTypes: excludeAllProofs.String(),
 				},
 			},
 		},
 	)
 	require.NoError(t.t, err)
 
-	resp, err := t.tapd.UniverseClient.QueryFederationSyncConfig(ctxb, &unirpc.QueryFederationSyncConfigRequest{
-		Id: []*unirpc.ID{
-			uniIdRpc,
-		},
-	})
+	resp, err := t.tapd.UniverseClient.QueryFederationSyncConfig(
+		ctx, &unirpc.QueryFederationSyncConfigRequest{
+			Id: []*unirpc.ID{
+				uniIdRpc,
+			},
+		})
 	require.NoError(t.t, err)
 
 	// In general, we expect to only sync issuance proofs.
-	require.True(t.t, resp.GeneralSyncConfig.OnlySyncIssuanceProofs)
+	require.Equal(
+		t.t, resp.GeneralSyncConfig.ProofTypes, issuanceOnly.String(),
+	)
 
 	// Ensure that the universe specific config is set as expected.
 	require.Equal(t.t, len(resp.AssetSyncConfigs), 1)
 
 	assetSyncConfig := resp.AssetSyncConfigs[0]
 
 	require.Equal(t.t, uniIdRpc.Id, assetSyncConfig.Id.Id)
-	require.True(t.t, assetSyncConfig.Exclude)
-	require.False(t.t, assetSyncConfig.OnlySyncIssuanceProofs)
+	require.Equal(
+		t.t, assetSyncConfig.ProofTypes, excludeAllProofs.String(),
+	)
 }