From b54e623b8e8379baf23d9ad8d0b83af6c8f10835 Mon Sep 17 00:00:00 2001
From: Eunovo <eunovo9@gmail.com>
Date: Sat, 21 Sep 2024 17:29:00 +0000
Subject: [PATCH] wallet/spkman_tests: Add GetLabelledSPDestination unit test

---
 src/wallet/test/scriptpubkeyman_tests.cpp | 48 +++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/src/wallet/test/scriptpubkeyman_tests.cpp b/src/wallet/test/scriptpubkeyman_tests.cpp
index 15bff04221d23..45c1a75d7d4a7 100644
--- a/src/wallet/test/scriptpubkeyman_tests.cpp
+++ b/src/wallet/test/scriptpubkeyman_tests.cpp
@@ -2,13 +2,21 @@
 // Distributed under the MIT software license, see the accompanying
 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
+#include <addresstype.h>
+#include <coins.h>
+#include <common/bip352.h>
 #include <key.h>
+#include <primitives/transaction.h>
 #include <test/util/setup_common.h>
 #include <script/solver.h>
+#include <uint256.h>
 #include <wallet/scriptpubkeyman.h>
 #include <wallet/wallet.h>
+#include <wallet/walletutil.h>
 #include <wallet/test/util.h>
 
+#include <variant>
+
 #include <boost/test/unit_test.hpp>
 
 namespace wallet {
@@ -40,5 +48,45 @@ BOOST_AUTO_TEST_CASE(CanProvide)
     BOOST_CHECK(keyman.CanProvide(p2sh_script, data));
 }
 
+// Test SilentPaymentDescriptorScriptPubKeyMan::GetLabelledSPDestination
+// returns the correct label for the given Taproot spk
+BOOST_AUTO_TEST_CASE(GetLabelledSPDestination)
+{
+    CWallet wallet(m_node.chain.get(), "", CreateMockableWalletDatabase());
+    wallet.SetWalletFlag(WALLET_FLAG_DESCRIPTORS);
+    wallet.SetWalletFlag(WALLET_FLAG_SILENT_PAYMENTS);
+    {
+        LOCK(wallet.cs_wallet);
+        wallet.SetupDescriptorScriptPubKeyMans();
+    }
+    SilentPaymentDescriptorScriptPubKeyMan* sp_spk_man = *wallet.GetSilentPaymentsSPKMs().begin();
+    uint64_t index;
+    auto labelled_dest = sp_spk_man->GetNewLabelledDestination(index);
+
+    CKey private_key = GenerateRandomKey();
+    CPubKey pubkey = private_key.GetPubKey();
+    COutPoint prevout(Txid::FromUint256(*uint256::FromHex("daec98e9311f843277e8c59f9dccb42efd4d40881bff215ae82e5724fa754c50")), 0);
+    WitnessV0KeyHash p2wpkh(pubkey);
+    CScript scriptPubKey = GetScriptForDestination(p2wpkh);
+    CScript scriptSig;
+    auto sp_dest = std::get_if<V0SilentPaymentDestination>(&labelled_dest.value());
+    auto destinations = BIP352::GenerateSilentPaymentTaprootDestinations(
+        std::map<size_t, V0SilentPaymentDestination>{{0, *sp_dest}},
+        std::vector{private_key},
+        std::vector<KeyPair>(),
+        prevout
+    );
+    auto taproot_dest = destinations.value()[0];
+
+    CTxIn tx_in(prevout, scriptSig);
+    tx_in.scriptWitness.stack.push_back(std::vector<unsigned char>(pubkey.begin(), pubkey.end()));
+    auto public_data = BIP352::GetSilentPaymentsPublicData(
+        std::vector{tx_in},
+        std::map<COutPoint, Coin>{{prevout, Coin(CTxOut(10000, scriptPubKey), 1, false)}}
+    );
+    auto retrieved_labelled_dest = sp_spk_man->GetLabelledSPDestination(taproot_dest, *public_data);
+    BOOST_CHECK(*labelled_dest == *retrieved_labelled_dest);
+}
+
 BOOST_AUTO_TEST_SUITE_END()
 } // namespace wallet