itest: refactor testMaxChannelSize

lntest/itest/list_on_test.go

@@ -243,4 +243,8 @@ var allTestCasesTemp = []*lntemp.TestCase{
 		Name:     "hold invoice sender persistence",
 		TestFunc: testHoldInvoicePersistence,
 	},
+	{
+		Name:     "maximum channel size",
+		TestFunc: testMaxChannelSize,
+	},
 }

lntest/itest/lnd_max_channel_size_test.go

@@ -5,79 +5,50 @@ package itest
 
 import (
 	"fmt"
-	"strings"
 
 	"github.com/btcsuite/btcd/btcutil"
 	"github.com/lightningnetwork/lnd/funding"
-	"github.com/lightningnetwork/lnd/lntest"
+	"github.com/lightningnetwork/lnd/lntemp"
+	"github.com/lightningnetwork/lnd/lnwallet"
 )
 
-// testMaxChannelSize tests that lnd handles --maxchansize parameter
-// correctly. Wumbo nodes should enforce a default soft limit of 10 BTC by
-// default. This limit can be adjusted with --maxchansize config option
-func testMaxChannelSize(net *lntest.NetworkHarness, t *harnessTest) {
-	// We'll make two new nodes, both wumbo but with the default
-	// limit on maximum channel size (10 BTC)
-	wumboNode := net.NewNode(
-		t.t, "wumbo", []string{"--protocol.wumbo-channels"},
+// testMaxChannelSize tests that lnd handles --maxchansize parameter correctly.
+// Wumbo nodes should enforce a default soft limit of 10 BTC by default. This
+// limit can be adjusted with --maxchansize config option.
+func testMaxChannelSize(ht *lntemp.HarnessTest) {
+	// We'll make two new nodes, both wumbo but with the default limit on
+	// maximum channel size (10 BTC)
+	wumboNode := ht.NewNode(
+		"wumbo", []string{"--protocol.wumbo-channels"},
 	)
-	defer shutdownAndAssert(net, t, wumboNode)
-
-	wumboNode2 := net.NewNode(
-		t.t, "wumbo2", []string{"--protocol.wumbo-channels"},
+	wumboNode2 := ht.NewNode(
+		"wumbo2", []string{"--protocol.wumbo-channels"},
 	)
-	defer shutdownAndAssert(net, t, wumboNode2)
 
-	// We'll send 11 BTC to the wumbo node so it can test the wumbo soft limit.
-	net.SendCoins(t.t, 11*btcutil.SatoshiPerBitcoin, wumboNode)
+	// We'll send 11 BTC to the wumbo node so it can test the wumbo soft
+	// limit.
+	ht.FundCoins(11*btcutil.SatoshiPerBitcoin, wumboNode)
 
 	// Next we'll connect both nodes, then attempt to make a wumbo channel
 	// funding request, which should fail as it exceeds the default wumbo
 	// soft limit of 10 BTC.
-	net.EnsureConnected(t.t, wumboNode, wumboNode2)
+	ht.EnsureConnected(wumboNode, wumboNode2)
 
 	chanAmt := funding.MaxBtcFundingAmountWumbo + 1
-	_, err := net.OpenChannel(
-		wumboNode, wumboNode2, lntest.OpenChannelParams{
-			Amt: chanAmt,
-		},
+	// The test should show failure due to the channel exceeding our max
+	// size.
+	expectedErr := lnwallet.ErrChanTooLarge(
+		chanAmt, funding.MaxBtcFundingAmountWumbo,
 	)
-	if err == nil {
-		t.Fatalf("expected channel funding to fail as it exceeds 10 BTC limit")
-	}
-
-	// The test should show failure due to the channel exceeding our max size.
-	if !strings.Contains(err.Error(), "exceeds maximum chan size") {
-		t.Fatalf("channel should be rejected due to size, instead "+
-			"error was: %v", err)
-	}
-
-	// Next we'll create a non-wumbo node to verify that it enforces the
-	// BOLT-02 channel size limit and rejects our funding request.
-	miniNode := net.NewNode(t.t, "mini", nil)
-	defer shutdownAndAssert(net, t, miniNode)
-
-	net.EnsureConnected(t.t, wumboNode, miniNode)
-
-	_, err = net.OpenChannel(
-		wumboNode, miniNode, lntest.OpenChannelParams{
-			Amt: chanAmt,
-		},
+	ht.OpenChannelAssertErr(
+		wumboNode, wumboNode2,
+		lntemp.OpenChannelParams{Amt: chanAmt}, expectedErr,
 	)
-	if err == nil {
-		t.Fatalf("expected channel funding to fail as it exceeds 0.16 BTC limit")
-	}
 
-	// The test should show failure due to the channel exceeding our max size.
-	if !strings.Contains(err.Error(), "exceeds maximum chan size") {
-		t.Fatalf("channel should be rejected due to size, instead "+
-			"error was: %v", err)
-	}
-
-	// We'll now make another wumbo node with appropriate maximum channel size
-	// to accept our wumbo channel funding.
-	wumboNode3 := net.NewNode(
-		t.t, "wumbo3", []string{
+	// We'll now make another wumbo node with appropriate maximum channel
+	// size to accept our wumbo channel funding.
+	wumboNode3 := ht.NewNode(
+		"wumbo3", []string{
 			"--protocol.wumbo-channels",
 			fmt.Sprintf(
 				"--maxchansize=%v",
@@ -85,16 +56,12 @@ func testMaxChannelSize(net *lntest.NetworkHarness, t *harnessTest) {
 			),
 		},
 	)
-	defer shutdownAndAssert(net, t, wumboNode3)
 
 	// Creating a wumbo channel between these two nodes should succeed.
-	net.EnsureConnected(t.t, wumboNode, wumboNode3)
-	chanPoint := openChannelAndAssert(
-		t, net, wumboNode, wumboNode3,
-		lntest.OpenChannelParams{
-			Amt: chanAmt,
-		},
+	ht.EnsureConnected(wumboNode, wumboNode3)
+	chanPoint := ht.OpenChannel(
+		wumboNode, wumboNode3, lntemp.OpenChannelParams{Amt: chanAmt},
 	)
-	closeChannelAndAssert(t, net, wumboNode, chanPoint, false)
 
+	ht.CloseChannel(wumboNode, chanPoint)
 }

lntest/itest/lnd_test_list_on_test.go

@@ -186,10 +186,6 @@ var allTestCases = []*testCase{
 		name: "wumbo channels",
 		test: testWumboChannels,
 	},
-	{
-		name: "maximum channel size",
-		test: testMaxChannelSize,
-	},
 	{
 		name: "stateless init",
 		test: testStatelessInit,