lntemp+itest: refactor testSignOutputRaw

lntemp/harness.go

@@ -1944,3 +1944,26 @@ func (h *HarnessTest) ReceiveChannelEvent(
 
 	return nil
 }
+
+// GetOutputIndex returns the output index of the given address in the given
+// transaction.
+func (h *HarnessTest) GetOutputIndex(txid *chainhash.Hash, addr string) int {
+	// We'll then extract the raw transaction from the mempool in order to
+	// determine the index of the p2tr output.
+	tx := h.Miner.GetRawTransaction(txid)
+
+	p2trOutputIndex := -1
+	for i, txOut := range tx.MsgTx().TxOut {
+		_, addrs, _, err := txscript.ExtractPkScriptAddrs(
+			txOut.PkScript, h.Miner.ActiveNet,
+		)
+		require.NoError(h, err)
+
+		if addrs[0].String() == addr {
+			p2trOutputIndex = i
+		}
+	}
+	require.Greater(h, p2trOutputIndex, -1)
+
+	return p2trOutputIndex
+}

lntemp/harness_assertion.go

@@ -1,6 +1,7 @@
 package lntemp
 
 import (
+	"bytes"
 	"context"
 	"crypto/rand"
 	"encoding/hex"
@@ -2133,3 +2134,36 @@ func (h *HarnessTest) AssertInvoiceEqual(a, b *lnrpc.Invoice) {
 		require.Equal(h, htlcA.Amp, htlcB.Amp)
 	}
 }
+
+// AssertUTXOInWallet asserts that a given UTXO can be found in the node's
+// wallet.
+func (h *HarnessTest) AssertUTXOInWallet(hn *node.HarnessNode,
+	op *lnrpc.OutPoint, account string) {
+
+	err := wait.NoError(func() error {
+		req := &walletrpc.ListUnspentRequest{
+			Account: account,
+		}
+		resp := hn.RPC.ListUnspent(req)
+
+		err := fmt.Errorf("tx with hash %x not found", op.TxidBytes)
+		for _, utxo := range resp.Utxos {
+			if !bytes.Equal(utxo.Outpoint.TxidBytes, op.TxidBytes) {
+				continue
+			}
+
+			err = fmt.Errorf("tx with output index %v not found",
+				op.OutputIndex)
+			if utxo.Outpoint.OutputIndex != op.OutputIndex {
+				continue
+			}
+
+			return nil
+		}
+
+		return err
+	}, DefaultTimeout)
+
+	require.NoErrorf(h, err, "outpoint %v not found in %s's wallet",
+		op, hn.Name())
+}

lntest/itest/list_on_test.go

@@ -445,4 +445,8 @@ var allTestCasesTemp = []*lntemp.TestCase{
 		Name:     "derive shared key",
 		TestFunc: testDeriveSharedKey,
 	},
+	{
+		Name:     "sign output raw",
+		TestFunc: testSignOutputRaw,
+	},
 }

lntest/itest/lnd_signer_test.go

@@ -197,43 +197,35 @@ func runDeriveSharedKey(ht *lntemp.HarnessTest, alice *node.HarnessNode) {
 
 // testSignOutputRaw makes sure that the SignOutputRaw RPC can be used with all
 // custom ways of specifying the signing key in the key descriptor/locator.
-func testSignOutputRaw(net *lntest.NetworkHarness, t *harnessTest) {
-	runSignOutputRaw(t, net, net.Alice)
+func testSignOutputRaw(ht *lntemp.HarnessTest) {
+	runSignOutputRaw(ht, ht.Alice)
 }
 
 // runSignOutputRaw makes sure that the SignOutputRaw RPC can be used with all
 // custom ways of specifying the signing key in the key descriptor/locator.
-func runSignOutputRaw(t *harnessTest, net *lntest.NetworkHarness,
-	alice *lntest.HarnessNode) {
-
-	ctxb := context.Background()
-	ctxt, cancel := context.WithTimeout(ctxb, defaultTimeout)
-	defer cancel()
-
+func runSignOutputRaw(ht *lntemp.HarnessTest, alice *node.HarnessNode) {
 	// For the next step, we need a public key. Let's use a special family
 	// for this. We want this to be an index of zero.
 	const testCustomKeyFamily = 44
-	keyDesc, err := alice.WalletKitClient.DeriveNextKey(
-		ctxt, &walletrpc.KeyReq{
-			KeyFamily: testCustomKeyFamily,
-		},
-	)
-	require.NoError(t.t, err)
-	require.Equal(t.t, int32(0), keyDesc.KeyLoc.KeyIndex)
+	req := &walletrpc.KeyReq{
+		KeyFamily: testCustomKeyFamily,
+	}
+	keyDesc := alice.RPC.DeriveNextKey(req)
+	require.Equal(ht, int32(0), keyDesc.KeyLoc.KeyIndex)
 
 	targetPubKey, err := btcec.ParsePubKey(keyDesc.RawKeyBytes)
-	require.NoError(t.t, err)
+	require.NoError(ht, err)
 
 	// First, try with a key descriptor that only sets the public key.
 	assertSignOutputRaw(
-		t, net, alice, targetPubKey, &signrpc.KeyDescriptor{
+		ht, alice, targetPubKey, &signrpc.KeyDescriptor{
 			RawKeyBytes: keyDesc.RawKeyBytes,
 		},
 	)
 
 	// Now try again, this time only with the (0 index!) key locator.
 	assertSignOutputRaw(
-		t, net, alice, targetPubKey, &signrpc.KeyDescriptor{
+		ht, alice, targetPubKey, &signrpc.KeyDescriptor{
 			KeyLoc: &signrpc.KeyLocator{
 				KeyFamily: keyDesc.KeyLoc.KeyFamily,
 				KeyIndex:  keyDesc.KeyLoc.KeyIndex,
@@ -243,29 +235,25 @@ func runSignOutputRaw(t *harnessTest, net *lntest.NetworkHarness,
 
 	// And now test everything again with a new key where we know the index
 	// is not 0.
-	ctxt, cancel = context.WithTimeout(ctxb, defaultTimeout)
-	defer cancel()
-	keyDesc, err = alice.WalletKitClient.DeriveNextKey(
-		ctxt, &walletrpc.KeyReq{
-			KeyFamily: testCustomKeyFamily,
-		},
-	)
-	require.NoError(t.t, err)
-	require.Equal(t.t, int32(1), keyDesc.KeyLoc.KeyIndex)
+	req = &walletrpc.KeyReq{
+		KeyFamily: testCustomKeyFamily,
+	}
+	keyDesc = alice.RPC.DeriveNextKey(req)
+	require.Equal(ht, int32(1), keyDesc.KeyLoc.KeyIndex)
 
 	targetPubKey, err = btcec.ParsePubKey(keyDesc.RawKeyBytes)
-	require.NoError(t.t, err)
+	require.NoError(ht, err)
 
 	// First, try with a key descriptor that only sets the public key.
 	assertSignOutputRaw(
-		t, net, alice, targetPubKey, &signrpc.KeyDescriptor{
+		ht, alice, targetPubKey, &signrpc.KeyDescriptor{
 			RawKeyBytes: keyDesc.RawKeyBytes,
 		},
 	)
 
 	// Now try again, this time only with the key locator.
 	assertSignOutputRaw(
-		t, net, alice, targetPubKey, &signrpc.KeyDescriptor{
+		ht, alice, targetPubKey, &signrpc.KeyDescriptor{
 			KeyLoc: &signrpc.KeyLocator{
 				KeyFamily: keyDesc.KeyLoc.KeyFamily,
 				KeyIndex:  keyDesc.KeyLoc.KeyIndex,
@@ -277,53 +265,44 @@ func runSignOutputRaw(t *harnessTest, net *lntest.NetworkHarness,
 // assertSignOutputRaw sends coins to a p2wkh address derived from the given
 // target public key and then tries to spend that output again by invoking the
 // SignOutputRaw RPC with the key descriptor provided.
-func assertSignOutputRaw(t *harnessTest, net *lntest.NetworkHarness,
-	alice *lntest.HarnessNode, targetPubKey *btcec.PublicKey,
+func assertSignOutputRaw(ht *lntemp.HarnessTest,
+	alice *node.HarnessNode, targetPubKey *btcec.PublicKey,
 	keyDesc *signrpc.KeyDescriptor) {
 
-	ctxb := context.Background()
-	ctxt, cancel := context.WithTimeout(ctxb, defaultTimeout)
-	defer cancel()
-
 	pubKeyHash := btcutil.Hash160(targetPubKey.SerializeCompressed())
 	targetAddr, err := btcutil.NewAddressWitnessPubKeyHash(
 		pubKeyHash, harnessNetParams,
 	)
-	require.NoError(t.t, err)
+	require.NoError(ht, err)
 	targetScript, err := txscript.PayToAddrScript(targetAddr)
-	require.NoError(t.t, err)
+	require.NoError(ht, err)
 
 	// Send some coins to the generated p2wpkh address.
-	_, err = alice.SendCoins(ctxt, &lnrpc.SendCoinsRequest{
+	req := &lnrpc.SendCoinsRequest{
 		Addr:   targetAddr.String(),
 		Amount: 800_000,
-	})
-	require.NoError(t.t, err)
+	}
+	alice.RPC.SendCoins(req)
 
 	// Wait until the TX is found in the mempool.
-	txid, err := waitForTxInMempool(net.Miner.Client, minerMempoolTimeout)
-	require.NoError(t.t, err)
+	txid := ht.Miner.AssertNumTxsInMempool(1)[0]
 
-	targetOutputIndex := getOutputIndex(
-		t, net.Miner, txid, targetAddr.String(),
-	)
+	targetOutputIndex := ht.GetOutputIndex(txid, targetAddr.String())
 
 	// Clear the mempool.
-	mineBlocks(t, net, 1, 1)
+	ht.MineBlocksAndAssertNumTxes(1, 1)
 
 	// Try to spend the output now to a new p2wkh address.
-	p2wkhResp, err := alice.NewAddress(ctxt, &lnrpc.NewAddressRequest{
-		Type: lnrpc.AddressType_WITNESS_PUBKEY_HASH,
-	})
-	require.NoError(t.t, err)
+	addrReq := &lnrpc.NewAddressRequest{Type: AddrTypeWitnessPubkeyHash}
+	p2wkhResp := alice.RPC.NewAddress(addrReq)
 
 	p2wkhAdrr, err := btcutil.DecodeAddress(
 		p2wkhResp.Address, harnessNetParams,
 	)
-	require.NoError(t.t, err)
+	require.NoError(ht, err)
 
 	p2wkhPkScript, err := txscript.PayToAddrScript(p2wkhAdrr)
-	require.NoError(t.t, err)
+	require.NoError(ht, err)
 
 	tx := wire.NewMsgTx(2)
 	tx.TxIn = []*wire.TxIn{{
@@ -339,24 +318,22 @@ func assertSignOutputRaw(t *harnessTest, net *lntest.NetworkHarness,
 	}}
 
 	var buf bytes.Buffer
-	require.NoError(t.t, tx.Serialize(&buf))
+	require.NoError(ht, tx.Serialize(&buf))
 
-	signResp, err := alice.SignerClient.SignOutputRaw(
-		ctxt, &signrpc.SignReq{
-			RawTxBytes: buf.Bytes(),
-			SignDescs: []*signrpc.SignDescriptor{{
-				Output: &signrpc.TxOut{
-					PkScript: targetScript,
-					Value:    800_000,
-				},
-				InputIndex:    0,
-				KeyDesc:       keyDesc,
-				Sighash:       uint32(txscript.SigHashAll),
-				WitnessScript: targetScript,
-			}},
-		},
-	)
-	require.NoError(t.t, err)
+	signReq := &signrpc.SignReq{
+		RawTxBytes: buf.Bytes(),
+		SignDescs: []*signrpc.SignDescriptor{{
+			Output: &signrpc.TxOut{
+				PkScript: targetScript,
+				Value:    800_000,
+			},
+			InputIndex:    0,
+			KeyDesc:       keyDesc,
+			Sighash:       uint32(txscript.SigHashAll),
+			WitnessScript: targetScript,
+		}},
+	}
+	signResp := alice.RPC.SignOutputRaw(signReq)
 
 	tx.TxIn[0].Witness = wire.TxWitness{
 		append(signResp.RawSigs[0], byte(txscript.SigHashAll)),
@@ -364,30 +341,25 @@ func assertSignOutputRaw(t *harnessTest, net *lntest.NetworkHarness,
 	}
 
 	buf.Reset()
-	require.NoError(t.t, tx.Serialize(&buf))
+	require.NoError(ht, tx.Serialize(&buf))
 
-	_, err = alice.WalletKitClient.PublishTransaction(
-		ctxt, &walletrpc.Transaction{
-			TxHex: buf.Bytes(),
-		},
-	)
-	require.NoError(t.t, err)
+	alice.RPC.PublishTransaction(&walletrpc.Transaction{
+		TxHex: buf.Bytes(),
+	})
 
 	// Wait until the spending tx is found.
-	txid, err = waitForTxInMempool(net.Miner.Client, minerMempoolTimeout)
-	require.NoError(t.t, err)
-	p2wkhOutputIndex := getOutputIndex(
-		t, net.Miner, txid, p2wkhAdrr.String(),
-	)
+	txid = ht.Miner.AssertNumTxsInMempool(1)[0]
+	p2wkhOutputIndex := ht.GetOutputIndex(txid, p2wkhAdrr.String())
+
 	op := &lnrpc.OutPoint{
 		TxidBytes:   txid[:],
 		OutputIndex: uint32(p2wkhOutputIndex),
 	}
-	assertWalletUnspent(t, alice, op, "")
+	ht.AssertUTXOInWallet(alice, op, "")
 
-	// Mine another block to clean up the mempool and to make sure the spend
-	// tx is actually included in a block.
-	mineBlocks(t, net, 1, 1)
+	// Mine another block to clean up the mempool and to make sure the
+	// spend tx is actually included in a block.
+	ht.MineBlocksAndAssertNumTxes(1, 1)
 }
 
 // deriveCustomizedKey uses the family and index to derive a public key from

lntest/itest/lnd_test_list_on_test.go

@@ -4,10 +4,6 @@
 package itest
 
 var allTestCases = []*testCase{
-	{
-		name: "sign output raw",
-		test: testSignOutputRaw,
-	},
 	{
 		name: "sign verify message",
 		test: testSignVerifyMessage,