itest: add test for zero-value utxo garbage collection

Author: darioAnongba

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,221 @@
+package itest
+
+import (
+	"context"
+
+	"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)
+
+	// Check Alice's UTXOs for tombstone script keys.
+	utxos, 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, utxos.ManagedUtxos, 1)
+
+	tombstoneOp := ""
+	for outpoint, utxo := range utxos.ManagedUtxos {
+		if !utxo.Swept {
+			tombstoneOp = outpoint
+			break
+		}
+	}
+
+	// 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 UTXOs again. The tombstone should have been swept.
+	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, 2)
+
+	// After the sweep, check that the first tombstone
+	// is now marked as swept.
+	sweptUtxo, ok := utxosAfterSend.ManagedUtxos[tombstoneOp]
+	require.True(t.t, ok)
+	require.True(t.t, sweptUtxo.Swept)
+
+	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,
+	)
+
+	// Second tombstone should be swept.
+	utxosAfterBurn, 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)
+
+	sweptTombstone2, ok := utxosAfterBurn.ManagedUtxos[tombstoneOp2]
+	require.True(t.t, ok)
+	require.True(t.t, sweptTombstone2.Swept)
+
+	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)
+
+	// Burn UTXO should be swept.
+	finalBurnUtxos, 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, finalBurnUtxos.ManagedUtxos, 1)
+
+	sweptBurnUtxo, ok := finalBurnUtxos.ManagedUtxos[burnOutpoint]
+	require.True(t.t, ok)
+	require.True(t.t, sweptBurnUtxo.Swept)
+}