From 37b8210f37038d2c98f4cd500734e7125cb48837 Mon Sep 17 00:00:00 2001
From: yyforyongyu <yong2452@gmail.com>
Date: Sat, 9 Nov 2024 15:18:40 +0800
Subject: [PATCH] itest: break down taproot tests

---
 itest/list_on_test.go     | 12 ++++++++++--
 itest/lnd_taproot_test.go | 17 ++++++++++++++---
 2 files changed, 24 insertions(+), 5 deletions(-)

diff --git a/itest/list_on_test.go b/itest/list_on_test.go
index b0146f16b..e8ae3fbe0 100644
--- a/itest/list_on_test.go
+++ b/itest/list_on_test.go
@@ -469,8 +469,16 @@ var allTestCases = []*lntest.TestCase{
 		TestFunc: testBumpForceCloseFee,
 	},
 	{
-		Name:     "taproot",
-		TestFunc: testTaproot,
+		Name:     "taproot spend",
+		TestFunc: testTaprootSpend,
+	},
+	{
+		Name:     "taproot musig2",
+		TestFunc: testTaprootMuSig2,
+	},
+	{
+		Name:     "taproot import scripts",
+		TestFunc: testTaprootImportScripts,
 	},
 	{
 		Name:     "simple taproot channel activation",
diff --git a/itest/lnd_taproot_test.go b/itest/lnd_taproot_test.go
index 2b68105fd..ddbfb6983 100644
--- a/itest/lnd_taproot_test.go
+++ b/itest/lnd_taproot_test.go
@@ -46,9 +46,9 @@ var (
 	))
 )
 
-// testTaproot ensures that the daemon can send to and spend from taproot (p2tr)
-// outputs.
-func testTaproot(ht *lntest.HarnessTest) {
+// testTaprootSpend ensures that the daemon can send to and spend from taproot
+// (p2tr) outputs.
+func testTaprootSpend(ht *lntest.HarnessTest) {
 	alice := ht.NewNode("Alice", nil)
 
 	testTaprootSendCoinsKeySpendBip86(ht, alice)
@@ -62,6 +62,12 @@ func testTaproot(ht *lntest.HarnessTest) {
 		ht, alice, txscript.SigHashSingle,
 	)
 	testTaprootSignOutputRawKeySpendRootHash(ht, alice)
+}
+
+// testTaprootMuSig2 ensures that the daemon can send to and spend from taproot
+// (p2tr) outputs using musig2.
+func testTaprootMuSig2(ht *lntest.HarnessTest) {
+	alice := ht.NewNodeWithCoins("Alice", nil)
 
 	muSig2Versions := []signrpc.MuSig2Version{
 		signrpc.MuSig2Version_MUSIG2_VERSION_V040,
@@ -74,6 +80,11 @@ func testTaproot(ht *lntest.HarnessTest) {
 		testTaprootMuSig2CombinedLeafKeySpend(ht, alice, version)
 		testMuSig2CombineKey(ht, alice, version)
 	}
+}
+
+// testTaprootImportScripts ensures that the daemon can import taproot scripts.
+func testTaprootImportScripts(ht *lntest.HarnessTest) {
+	alice := ht.NewNodeWithCoins("Alice", nil)
 
 	testTaprootImportTapscriptFullTree(ht, alice)
 	testTaprootImportTapscriptPartialReveal(ht, alice)