From 913c9f9ae9d7ff7f7b5ba6f245f5041e41958e17 Mon Sep 17 00:00:00 2001
From: Jesse de Wit <witdejesse@hotmail.com>
Date: Mon, 19 Dec 2022 13:20:54 +0100
Subject: [PATCH] adds a test for probing with probing-01:

---
 itest/lspd_test.go    |  4 +++
 itest/probing_test.go | 74 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 78 insertions(+)
 create mode 100644 itest/probing_test.go

diff --git a/itest/lspd_test.go b/itest/lspd_test.go
index 6360539..4c69ad4 100644
--- a/itest/lspd_test.go
+++ b/itest/lspd_test.go
@@ -93,4 +93,8 @@ var allTestCases = []*testCase{
 		name: "testRegularForward",
 		test: testRegularForward,
 	},
+	{
+		name: "testProbing",
+		test: testProbing,
+	},
 }
diff --git a/itest/probing_test.go b/itest/probing_test.go
new file mode 100644
index 0000000..a11818e
--- /dev/null
+++ b/itest/probing_test.go
@@ -0,0 +1,74 @@
+package itest
+
+import (
+	"crypto/sha256"
+	"log"
+	"time"
+
+	"github.com/breez/lntest"
+	lspd "github.com/breez/lspd/rpc"
+	"github.com/stretchr/testify/assert"
+)
+
+func testProbing(p *testParams) {
+	alice := lntest.NewClnNode(p.h, p.m, "Alice")
+	alice.Start()
+	alice.Fund(10000000)
+	p.lsp.LightningNode().Fund(10000000)
+
+	log.Print("Opening channel between Alice and the lsp")
+	channel := alice.OpenChannel(p.lsp.LightningNode(), &lntest.OpenChannelOptions{
+		AmountSat: publicChanAmount,
+	})
+	channelId := alice.WaitForChannelReady(channel)
+
+	log.Printf("Adding bob's invoices")
+	outerAmountMsat := uint64(2100000)
+	innerAmountMsat, lspAmountMsat := calculateInnerAmountMsat(p.lsp, outerAmountMsat)
+	description := "Please pay me"
+	innerInvoice, outerInvoice := GenerateInvoices(p.BreezClient(),
+		generateInvoicesRequest{
+			innerAmountMsat: innerAmountMsat,
+			outerAmountMsat: outerAmountMsat,
+			description:     description,
+			lsp:             p.lsp,
+		})
+
+	log.Print("Connecting bob to lspd")
+	p.BreezClient().Node().ConnectPeer(p.lsp.LightningNode())
+
+	log.Printf("Registering payment with lsp")
+	RegisterPayment(p.lsp, &lspd.PaymentInformation{
+		PaymentHash:        innerInvoice.paymentHash,
+		PaymentSecret:      innerInvoice.paymentSecret,
+		Destination:        p.BreezClient().Node().NodeId(),
+		IncomingAmountMsat: int64(outerAmountMsat),
+		OutgoingAmountMsat: int64(lspAmountMsat),
+	})
+
+	// TODO: Fix race waiting for htlc interceptor.
+	log.Printf("Waiting %v to allow htlc interceptor to activate.", htlcInterceptorDelay)
+	<-time.After(htlcInterceptorDelay)
+
+	h := sha256.New()
+	_, _ = h.Write([]byte("probing-01:"))
+	_, _ = h.Write(outerInvoice.paymentHash)
+	fakePaymentHash := h.Sum(nil)
+
+	log.Printf("Alice paying with fake payment hash with Bob online %x", fakePaymentHash)
+	route := constructRoute(p.lsp.LightningNode(), p.BreezClient().Node(), channelId, lntest.NewShortChanIDFromString("1x0x0"), outerAmountMsat)
+	_, err := alice.PayViaRoute(outerAmountMsat, fakePaymentHash, outerInvoice.paymentSecret, route)
+
+	// Expect temporary channel failure if the peer is online
+	assert.Contains(p.t, err.Error(), "WIRE_TEMPORARY_CHANNEL_FAILURE")
+
+	// Kill the mobile client
+	log.Printf("Stopping breez client")
+	p.BreezClient().Stop()
+
+	log.Printf("Alice paying with fake payment hash with Bob offline %x", fakePaymentHash)
+	_, err = alice.PayViaRoute(outerAmountMsat, fakePaymentHash, outerInvoice.paymentSecret, route)
+
+	// Expect incorrect or unknown payment details if the peer is offline
+	assert.Contains(p.t, err.Error(), "WIRE_INCORRECT_OR_UNKNOWN_PAYMENT_DETAILS")
+}