itest: add test for zero-value utxo garbage collection

Author: darioAnongba

itest/full_value_split_test.go

@@ -61,10 +61,11 @@ func testFullValueSend(t *harnessTest) {
 		1, 2,
 	)
 
-	// Alice should have one more zero-value tombstones in her wallet.
+	// After the second run, Alice's previous tombstons were swept. She now
+	// has 1 new tombstone UTXO from the last full-value send.
 	AssertBalances(
 		t.t, t.tapd, 0, WithScriptKeyType(asset.ScriptKeyTombstone),
-		WithNumUtxos(3), WithNumAnchorUtxos(3),
+		WithNumUtxos(1), WithNumAnchorUtxos(1),
 	)
 	AssertBalances(
 		t.t, secondTapd, mintedAsset.Amount+mintedGroupAsset.Amount,

itest/test_list_on_test.go

@@ -101,6 +101,10 @@ var allTestCases = []*testCase{
 		name: "min relay fee bump",
 		test: testMinRelayFeeBump,
 	},
+	{
+		name: "zero value anchor sweep",
+		test: testZeroValueAnchorSweep,
+	},
 	{
 		name: "restart receiver check balance",
 		test: testRestartReceiverCheckBalance,

itest/zero_value_anchor_test.go

@@ -0,0 +1,195 @@
+package itest
+
+import (
+	"context"
+
+	"github.com/lightninglabs/taproot-assets/asset"
+	"github.com/lightninglabs/taproot-assets/taprpc"
+	"github.com/lightninglabs/taproot-assets/taprpc/mintrpc"
+	"github.com/stretchr/testify/require"
+)
+
+// testZeroValueAnchorSweep tests that zero-value anchor outputs
+// are automatically swept when creating new on-chain transactions.
+func testZeroValueAnchorSweep(t *harnessTest) {
+	ctxb := context.Background()
+
+	// First, mint some simple asset.
+	rpcAssets := MintAssetsConfirmBatch(
+		t.t, t.lndHarness.Miner().Client, t.tapd,
+		[]*mintrpc.MintAssetRequest{simpleAssets[0]},
+	)
+	genInfo := rpcAssets[0].AssetGenesis
+	assetAmount := simpleAssets[0].Asset.Amount
+
+	// Create a second tapd node.
+	bobLnd := t.lndHarness.NewNodeWithCoins("Bob", nil)
+	secondTapd := setupTapdHarness(t.t, t, bobLnd, t.universeServer)
+	defer func() {
+		require.NoError(t.t, secondTapd.stop(!*noDelete))
+	}()
+
+	bobAddr, err := secondTapd.NewAddr(ctxb, &taprpc.NewAddrRequest{
+		AssetId:      genInfo.AssetId,
+		Amt:          assetAmount,
+		AssetVersion: rpcAssets[0].Version,
+	})
+	require.NoError(t.t, err)
+
+	// Send ALL assets to Bob, which should create a tombstone.
+	sendResp, _ := sendAssetsToAddr(t, t.tapd, bobAddr)
+
+	ConfirmAndAssertOutboundTransfer(
+		t.t, t.lndHarness.Miner().Client, t.tapd, sendResp,
+		genInfo.AssetId,
+		[]uint64{0, assetAmount}, 0, 1,
+	)
+	AssertNonInteractiveRecvComplete(t.t, secondTapd, 1)
+
+	// Alice should have 1 tombstone UTXO from the full-value send.
+	AssertBalances(
+		t.t, t.tapd, 0, WithScriptKeyType(asset.ScriptKeyTombstone),
+		WithNumUtxos(1), WithNumAnchorUtxos(1),
+	)
+
+	// Test 1: Send transaction sweeps tombstones.
+	rpcAssets2 := MintAssetsConfirmBatch(
+		t.t, t.lndHarness.Miner().Client, t.tapd,
+		[]*mintrpc.MintAssetRequest{simpleAssets[0]},
+	)
+	genInfo2 := rpcAssets2[0].AssetGenesis
+
+	// Send full amount of the new asset. This should sweep Alice's
+	// first tombstone and create a new one.
+	bobAddr2, err := secondTapd.NewAddr(ctxb, &taprpc.NewAddrRequest{
+		AssetId:      genInfo2.AssetId,
+		Amt:          assetAmount,
+		AssetVersion: rpcAssets2[0].Version,
+	})
+	require.NoError(t.t, err)
+
+	sendResp2, _ := sendAssetsToAddr(t, t.tapd, bobAddr2)
+
+	ConfirmAndAssertOutboundTransfer(
+		t.t, t.lndHarness.Miner().Client, t.tapd, sendResp2,
+		genInfo2.AssetId,
+		[]uint64{0, assetAmount}, 1, 2,
+	)
+	AssertNonInteractiveRecvComplete(t.t, secondTapd, 2)
+
+	// Check Alice's tombstone balance. The first tombstone should have been
+	// swept (spent on-chain as an input), and a new one created. We now
+	// have 1 tombstone UTXO (the new one from the second send).
+	AssertBalances(
+		t.t, t.tapd, 0, WithScriptKeyType(asset.ScriptKeyTombstone),
+		WithNumUtxos(1), WithNumAnchorUtxos(1),
+	)
+
+	// Get the new tombstone outpoint.
+	utxosAfterSend, err := t.tapd.ListUtxos(ctxb, &taprpc.ListUtxosRequest{
+		ScriptKeyType: &taprpc.ScriptKeyTypeQuery{
+			Type: &taprpc.ScriptKeyTypeQuery_ExplicitType{
+				ExplicitType: taprpc.
+					ScriptKeyType_SCRIPT_KEY_TOMBSTONE,
+			},
+		},
+	})
+	require.NoError(t.t, err)
+	require.Len(t.t, utxosAfterSend.ManagedUtxos, 1)
+
+	tombstoneOp2 := ""
+	for outpoint, utxo := range utxosAfterSend.ManagedUtxos {
+		if !utxo.Swept {
+			tombstoneOp2 = outpoint
+			break
+		}
+	}
+	require.NotEmpty(t.t, tombstoneOp2)
+
+	// Test 2: Burning transaction sweeps tombstones.
+	rpcAssets3 := MintAssetsConfirmBatch(
+		t.t, t.lndHarness.Miner().Client, t.tapd,
+		[]*mintrpc.MintAssetRequest{simpleAssets[0]},
+	)
+	genInfo3 := rpcAssets3[0].AssetGenesis
+
+	// Full burn the asset to create a zero-value burn UTXO
+	// and sweep the second tombstone.
+	burnResp, err := t.tapd.BurnAsset(ctxb, &taprpc.BurnAssetRequest{
+		Asset: &taprpc.BurnAssetRequest_AssetId{
+			AssetId: genInfo3.AssetId,
+		},
+		AmountToBurn:     assetAmount,
+		ConfirmationText: "assets will be destroyed",
+	})
+	require.NoError(t.t, err)
+
+	AssertAssetOutboundTransferWithOutputs(
+		t.t, t.lndHarness.Miner().Client, t.tapd, burnResp.BurnTransfer,
+		[][]byte{genInfo3.AssetId},
+		[]uint64{assetAmount}, 2, 3, 1, true,
+	)
+
+	// Alice should have 0 tombstones remaining and 1 burn UTXO.
+	AssertBalances(
+		t.t, t.tapd, 0, WithScriptKeyType(asset.ScriptKeyTombstone),
+		WithNumUtxos(0), WithNumAnchorUtxos(0),
+	)
+	AssertBalances(
+		t.t, t.tapd, assetAmount, WithScriptKeyType(asset.ScriptKeyBurn),
+		WithNumUtxos(1), WithNumAnchorUtxos(1),
+	)
+
+	// Get the burn UTXO outpoint for the next test.
+	burnUtxos, err := t.tapd.ListUtxos(ctxb, &taprpc.ListUtxosRequest{
+		ScriptKeyType: &taprpc.ScriptKeyTypeQuery{
+			Type: &taprpc.ScriptKeyTypeQuery_ExplicitType{
+				ExplicitType: taprpc.
+					ScriptKeyType_SCRIPT_KEY_BURN,
+			},
+		},
+	})
+	require.NoError(t.t, err)
+	require.Len(t.t, burnUtxos.ManagedUtxos, 1)
+
+	burnOutpoint := ""
+	for outpoint, utxo := range burnUtxos.ManagedUtxos {
+		if !utxo.Swept {
+			burnOutpoint = outpoint
+			break
+		}
+	}
+	require.NotEmpty(t.t, burnOutpoint)
+
+	// Test 3: Send transactions sweeps zero-value burns.
+	rpcAssets4 := MintAssetsConfirmBatch(
+		t.t, t.lndHarness.Miner().Client, t.tapd,
+		[]*mintrpc.MintAssetRequest{simpleAssets[0]},
+	)
+	genInfo4 := rpcAssets4[0].AssetGenesis
+
+	// Send partial amouunt. This should NOT create a tombstone output
+	// and sweep the burn UTXO.
+	partialAmount := assetAmount / 2
+	bobAddr3, err := secondTapd.NewAddr(ctxb, &taprpc.NewAddrRequest{
+		AssetId:      genInfo4.AssetId,
+		Amt:          partialAmount,
+		AssetVersion: rpcAssets4[0].Version,
+	})
+	require.NoError(t.t, err)
+
+	sendResp3, _ := sendAssetsToAddr(t, t.tapd, bobAddr3)
+
+	ConfirmAndAssertOutboundTransfer(
+		t.t, t.lndHarness.Miner().Client, t.tapd, sendResp3,
+		genInfo4.AssetId,
+		[]uint64{partialAmount, partialAmount}, 3, 4,
+	)
+	AssertNonInteractiveRecvComplete(t.t, secondTapd, 3)
+
+	// The burn UTXO should have been swept.
+	AssertBalances(
+		t.t, t.tapd, 0, WithScriptKeyType(asset.ScriptKeyBurn),
+		WithNumUtxos(0), WithNumAnchorUtxos(0),
+	)
+}