itest: break down basic funding flow tests

itest/list_on_test.go

@@ -11,10 +11,6 @@ var allTestCases = []*lntest.TestCase{
 		Name:     "update channel status",
 		TestFunc: testUpdateChanStatus,
 	},
-	{
-		Name:     "basic funding flow",
-		TestFunc: testBasicChannelFunding,
-	},
 	{
 		Name:     "external channel funding",
 		TestFunc: testExternalFundingChanPoint,
@@ -681,4 +677,5 @@ func init() {
 	allTestCases = append(allTestCases, zeroConfPolicyTestCases...)
 	allTestCases = append(allTestCases, channelFeePolicyTestCases...)
 	allTestCases = append(allTestCases, walletImportAccountTestCases...)
+	allTestCases = append(allTestCases, basicFundingTestCases...)
 }

itest/lnd_funding_test.go

@@ -20,40 +20,137 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
-// testBasicChannelFunding performs a test exercising expected behavior from a
-// basic funding workflow. The test creates a new channel between Alice and
-// Bob, then immediately closes the channel after asserting some expected post
-// conditions. Finally, the chain itself is checked to ensure the closing
-// transaction was mined.
-func testBasicChannelFunding(ht *lntest.HarnessTest) {
-	// Run through the test with combinations of all the different
-	// commitment types.
-	allTypes := []lnrpc.CommitmentType{
+// basicFundingTestCases defines the test cases for the basic funding test.
+var basicFundingTestCases = []*lntest.TestCase{
+	{
+		Name:     "basic funding flow static key remote",
+		TestFunc: testBasicChannelFundingStaticRemote,
+	},
+	{
+		Name:     "basic funding flow anchor",
+		TestFunc: testBasicChannelFundingAnchor,
+	},
+	{
+		Name:     "basic funding flow simple taproot",
+		TestFunc: testBasicChannelFundingSimpleTaproot,
+	},
+}
+
+// allFundingTypes defines the channel types to test for the basic funding
+// test.
+var allFundingTypes = []lnrpc.CommitmentType{
 	lnrpc.CommitmentType_STATIC_REMOTE_KEY,
 	lnrpc.CommitmentType_ANCHORS,
 	lnrpc.CommitmentType_SIMPLE_TAPROOT,
-	}
+}
 
-	// testFunding is a function closure that takes Carol and Dave's
-	// commitment types and test the funding flow.
-	testFunding := func(ht *lntest.HarnessTest, carolCommitType,
+// testBasicChannelFundingStaticRemote performs a test exercising expected
+// behavior from a basic funding workflow. The test creates a new channel
+// between Carol and Dave, with Carol using the static remote key commitment
+// type, and Dave using allFundingTypes.
+func testBasicChannelFundingStaticRemote(ht *lntest.HarnessTest) {
+	carolCommitType := lnrpc.CommitmentType_STATIC_REMOTE_KEY
+
+	// We'll test all possible combinations of the feature bit presence
+	// that both nodes can signal for this new channel type. We'll make a
+	// new Carol+Dave for each test instance as well.
+	for _, daveCommitType := range allFundingTypes {
+		cc := carolCommitType
+		dc := daveCommitType
+
+		testName := fmt.Sprintf(
+			"carol_commit=%v,dave_commit=%v", cc, dc,
+		)
+
+		success := ht.Run(testName, func(t *testing.T) {
+			st := ht.Subtest(t)
+			runBasicFundingTest(st, cc, dc)
+		})
+
+		if !success {
+			break
+		}
+	}
+}
+
+// testBasicChannelFundingAnchor performs a test exercising expected behavior
+// from a basic funding workflow. The test creates a new channel between Carol
+// and Dave, with Carol using the anchor commitment type, and Dave using
+// allFundingTypes.
+func testBasicChannelFundingAnchor(ht *lntest.HarnessTest) {
+	carolCommitType := lnrpc.CommitmentType_ANCHORS
+
+	// We'll test all possible combinations of the feature bit presence
+	// that both nodes can signal for this new channel type. We'll make a
+	// new Carol+Dave for each test instance as well.
+	for _, daveCommitType := range allFundingTypes {
+		cc := carolCommitType
+		dc := daveCommitType
+
+		testName := fmt.Sprintf(
+			"carol_commit=%v,dave_commit=%v", cc, dc,
+		)
+
+		success := ht.Run(testName, func(t *testing.T) {
+			st := ht.Subtest(t)
+			runBasicFundingTest(st, cc, dc)
+		})
+
+		if !success {
+			break
+		}
+	}
+}
+
+// testBasicChannelFundingSimpleTaproot performs a test exercising expected
+// behavior from a basic funding workflow. The test creates a new channel
+// between Carol and Dave, with Carol using the simple taproot commitment type,
+// and Dave using allFundingTypes.
+func testBasicChannelFundingSimpleTaproot(ht *lntest.HarnessTest) {
+	carolCommitType := lnrpc.CommitmentType_SIMPLE_TAPROOT
+
+	// We'll test all possible combinations of the feature bit presence
+	// that both nodes can signal for this new channel type. We'll make a
+	// new Carol+Dave for each test instance as well.
+	for _, daveCommitType := range allFundingTypes {
+		cc := carolCommitType
+		dc := daveCommitType
+
+		testName := fmt.Sprintf(
+			"carol_commit=%v,dave_commit=%v", cc, dc,
+		)
+
+		success := ht.Run(testName, func(t *testing.T) {
+			st := ht.Subtest(t)
+			runBasicFundingTest(st, cc, dc)
+		})
+
+		if !success {
+			break
+		}
+	}
+}
+
+// runBasicFundingTest is a helper function that takes Carol and Dave's
+// commitment types and test the funding flow.
+func runBasicFundingTest(ht *lntest.HarnessTest, carolCommitType,
 	daveCommitType lnrpc.CommitmentType) {
 
-		// Based on the current tweak variable for Carol, we'll
-		// preferentially signal the legacy commitment format.  We do
-		// the same for Dave shortly below.
+	// Based on the current tweak variable for Carol, we'll preferentially
+	// signal the legacy commitment format.  We do the same for Dave
+	// shortly below.
 	carolArgs := lntest.NodeArgsForCommitType(carolCommitType)
 	carol := ht.NewNode("Carol", carolArgs)
 
-		// Each time, we'll send Carol a new set of coins in order to
-		// fund the channel.
+	// Each time, we'll send Carol a new set of coins in order to fund the
+	// channel.
 	ht.FundCoins(btcutil.SatoshiPerBitcoin, carol)
 
 	daveArgs := lntest.NodeArgsForCommitType(daveCommitType)
 	dave := ht.NewNode("Dave", daveArgs)
 
-		// Before we start the test, we'll ensure both sides are
-		// connected to the funding flow can properly be executed.
+	// Before we start the test, we'll ensure both sides are connected to
+	// the funding flow can properly be executed.
 	ht.EnsureConnected(carol, dave)
 
 	var privateChan bool
@@ -66,9 +163,8 @@ func testBasicChannelFunding(ht *lntest.HarnessTest) {
 		privateChan = true
 	}
 
-		// If carol wants taproot, but dave wants something
-		// else, then we'll assert that the channel negotiation
-		// attempt fails.
+	// If carol wants taproot, but dave wants something else, then we'll
+	// assert that the channel negotiation attempt fails.
 	if carolCommitType == lnrpc.CommitmentType_SIMPLE_TAPROOT &&
 		daveCommitType != lnrpc.CommitmentType_SIMPLE_TAPROOT {
 
@@ -90,15 +186,14 @@ func testBasicChannelFunding(ht *lntest.HarnessTest) {
 		ht, carol, dave, nil, privateChan, &carolCommitType,
 	)
 
-		// Both nodes should report the same commitment
-		// type.
+	// Both nodes should report the same commitment type.
 	chansCommitType := carolChan.CommitmentType
 	require.Equal(ht, chansCommitType, daveChan.CommitmentType,
 		"commit types don't match")
 
-		// Now check that the commitment type reported by both nodes is
-		// what we expect. It will be the minimum of the two nodes'
-		// preference, in the order Legacy, Tweakless, Anchors.
+	// Now check that the commitment type reported by both nodes is what we
+	// expect. It will be the minimum of the two nodes' preference, in the
+	// order Legacy, Tweakless, Anchors.
 	expType := carolCommitType
 
 	switch daveCommitType {
@@ -107,15 +202,14 @@ func testBasicChannelFunding(ht *lntest.HarnessTest) {
 
 	// Dave supports anchors, type will be what Carol supports.
 	case lnrpc.CommitmentType_ANCHORS:
-			// However if Alice wants taproot chans, then we
-			// downgrade to anchors as this is still using implicit
-			// negotiation.
+		// However if Alice wants taproot chans, then we downgrade to
+		// anchors as this is still using implicit negotiation.
 		if expType == lnrpc.CommitmentType_SIMPLE_TAPROOT {
 			expType = lnrpc.CommitmentType_ANCHORS
 		}
 
-		// Dave only supports tweakless, channel will be downgraded to
-		// this type if Carol supports anchors.
+	// Dave only supports tweakless, channel will be downgraded to this
+	// type if Carol supports anchors.
 	case lnrpc.CommitmentType_STATIC_REMOTE_KEY:
 		switch expType {
 		case lnrpc.CommitmentType_ANCHORS:
@@ -124,8 +218,8 @@ func testBasicChannelFunding(ht *lntest.HarnessTest) {
 			expType = lnrpc.CommitmentType_STATIC_REMOTE_KEY
 		}
 
-		// Dave only supports legacy type, channel will be downgraded
-		// to this type.
+	// Dave only supports legacy type, channel will be downgraded to this
+	// type.
 	case lnrpc.CommitmentType_LEGACY:
 		expType = lnrpc.CommitmentType_LEGACY
 
@@ -139,7 +233,7 @@ func testBasicChannelFunding(ht *lntest.HarnessTest) {
 		chansCommitType == lnrpc.CommitmentType_ANCHORS:
 
 	case expType == lnrpc.CommitmentType_STATIC_REMOTE_KEY &&
-			chansCommitType == lnrpc.CommitmentType_STATIC_REMOTE_KEY: //nolint:ll
+		chansCommitType == lnrpc.CommitmentType_STATIC_REMOTE_KEY:
 
 	case expType == lnrpc.CommitmentType_LEGACY &&
 		chansCommitType == lnrpc.CommitmentType_LEGACY:
@@ -148,34 +242,8 @@ func testBasicChannelFunding(ht *lntest.HarnessTest) {
 		chansCommitType == lnrpc.CommitmentType_SIMPLE_TAPROOT:
 
 	default:
-			ht.Fatalf("expected nodes to signal "+
-				"commit type %v, instead got "+
-				"%v", expType, chansCommitType)
-		}
-	}
-
-test:
-	// We'll test all possible combinations of the feature bit presence
-	// that both nodes can signal for this new channel type. We'll make a
-	// new Carol+Dave for each test instance as well.
-	for _, carolCommitType := range allTypes {
-		for _, daveCommitType := range allTypes {
-			cc := carolCommitType
-			dc := daveCommitType
-
-			testName := fmt.Sprintf(
-				"carol_commit=%v,dave_commit=%v", cc, dc,
-			)
-
-			success := ht.Run(testName, func(t *testing.T) {
-				st := ht.Subtest(t)
-				testFunding(st, cc, dc)
-			})
-
-			if !success {
-				break test
-			}
-		}
+		ht.Fatalf("expected nodes to signal commit type %v, instead "+
+			"got %v", expType, chansCommitType)
 	}
 }