From 204c160c4d01e5d12130a89347923e8d7d4f0f57 Mon Sep 17 00:00:00 2001
From: Antoine Poinsot <darosior@protonmail.com>
Date: Mon, 21 Aug 2023 16:49:07 +0200
Subject: [PATCH] tests: test the RPC interface for managing labels

---
 tests/test_rpc.py | 113 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)

diff --git a/tests/test_rpc.py b/tests/test_rpc.py
index 119b04c98..7a599945f 100644
--- a/tests/test_rpc.py
+++ b/tests/test_rpc.py
@@ -596,3 +596,116 @@ def test_create_recovery(lianad, bitcoind):
     assert len(reco_psbt.tx.vout) == 1
     assert int(0.39999 * COIN) < int(reco_psbt.tx.vout[0].nValue) < int(0.4 * COIN)
     sign_and_broadcast(lianad, bitcoind, reco_psbt, recovery=True)
+
+
+def test_labels(lianad, bitcoind):
+    """Test the creation and updating of labels."""
+    # We can set a label for an address.
+    addr = lianad.rpc.getnewaddress()["address"]
+    lianad.rpc.updatelabels({addr: "first-addr"})
+    assert lianad.rpc.getlabels([addr])["labels"] == {addr: "first-addr"}
+    # And also update it.
+    lianad.rpc.updatelabels({addr: "first-addr-1"})
+    assert lianad.rpc.getlabels([addr])["labels"] == {addr: "first-addr-1"}
+    # But we can't set a label larger than 100 characters
+    with pytest.raises(RpcError, match=".*must be less or equal than 100 characters"):
+        lianad.rpc.updatelabels({addr: "".join("a" for _ in range(101))})
+
+    # We can set a label for a coin.
+    sec_addr = lianad.rpc.getnewaddress()["address"]
+    txid = bitcoind.rpc.sendtoaddress(sec_addr, 1)
+    wait_for(lambda: len(lianad.rpc.listcoins()["coins"]) == 1)
+    coin = lianad.rpc.listcoins()["coins"][0]
+    lianad.rpc.updatelabels({coin["outpoint"]: "first-coin"})
+    assert lianad.rpc.getlabels([coin["outpoint"]])["labels"] == {
+        coin["outpoint"]: "first-coin"
+    }
+    # And also update it.
+    lianad.rpc.updatelabels({coin["outpoint"]: "first-coin-1"})
+    assert lianad.rpc.getlabels([coin["outpoint"]])["labels"] == {
+        coin["outpoint"]: "first-coin-1"
+    }
+    # Its address though has no label.
+    assert lianad.rpc.getlabels([sec_addr])["labels"] == {}
+    # But we can receive a coin to the address that has a label set, and query both.
+    sec_txid = bitcoind.rpc.sendtoaddress(addr, 1)
+    wait_for(lambda: len(lianad.rpc.listcoins()["coins"]) == 2)
+    sec_coin = next(
+        c for c in lianad.rpc.listcoins()["coins"] if sec_txid in c["outpoint"]
+    )
+    lianad.rpc.updatelabels({sec_coin["outpoint"]: "sec-coin"})
+    res = lianad.rpc.getlabels([sec_coin["outpoint"], addr])["labels"]
+    assert len(res) == 2
+    assert res[sec_coin["outpoint"]] == "sec-coin"
+    assert res[addr] == "first-addr-1"
+    # We can also query the labels for both coins, of course.
+    res = lianad.rpc.getlabels([coin["outpoint"], sec_coin["outpoint"]])["labels"]
+    assert len(res) == 2
+    assert res[coin["outpoint"]] == "first-coin-1"
+    assert res[sec_coin["outpoint"]] == "sec-coin"
+
+    # We can set, update and query labels for deposit transactions.
+    lianad.rpc.updatelabels({txid: "first-deposit"})
+    assert lianad.rpc.getlabels([txid, sec_txid])["labels"] == {txid: "first-deposit"}
+    lianad.rpc.updatelabels({txid: "first-deposit-1", sec_txid: "second-deposit"})
+    res = lianad.rpc.getlabels([txid, sec_txid])["labels"]
+    assert len(res) == 2
+    assert res[txid] == "first-deposit-1"
+    assert res[sec_txid] == "second-deposit"
+
+    # We can set and update a label for a spend transaction.
+    spend_txid = get_txid(spend_coins(lianad, bitcoind, [coin, sec_coin]))
+    lianad.rpc.updatelabels({spend_txid: "spend-tx"})
+    assert lianad.rpc.getlabels([spend_txid])["labels"] == {spend_txid: "spend-tx"}
+    lianad.rpc.updatelabels({spend_txid: "spend-tx-1"})
+    assert lianad.rpc.getlabels([spend_txid])["labels"] == {spend_txid: "spend-tx-1"}
+
+    # We can set labels for inexistent stuff, as long as the format of the item being
+    # labelled is valid.
+    inexistent_txid = "".join("0" for _ in range(64))
+    inexistent_outpoint = "".join("1" for _ in range(64)) + ":42"
+    random_address = bitcoind.rpc.getnewaddress()
+    lianad.rpc.updatelabels(
+        {
+            inexistent_txid: "inex_txid",
+            inexistent_outpoint: "inex_outpoint",
+            random_address: "bitcoind-addr",
+        }
+    )
+    res = lianad.rpc.getlabels([inexistent_txid, inexistent_outpoint, random_address])[
+        "labels"
+    ]
+    assert len(res) == 3
+    assert res[inexistent_txid] == "inex_txid"
+    assert res[inexistent_outpoint] == "inex_outpoint"
+    assert res[random_address] == "bitcoind-addr"
+
+    # We'll confirm everything, shouldn't affect any of the labels.
+    bitcoind.generate_block(1, wait_for_mempool=spend_txid)
+    wait_for(
+        lambda: bitcoind.rpc.getblockcount() == lianad.rpc.getinfo()["block_height"]
+    )
+    res = lianad.rpc.getlabels(
+        [
+            addr,
+            sec_addr,  # No label for this one.
+            txid,
+            sec_txid,
+            coin["outpoint"],
+            sec_coin["outpoint"],
+            spend_txid,
+            inexistent_txid,
+            inexistent_outpoint,
+            random_address,
+        ]
+    )["labels"]
+    assert len(res) == 9
+    assert res[sec_coin["outpoint"]] == "sec-coin"
+    assert res[addr] == "first-addr-1"
+    assert res[coin["outpoint"]] == "first-coin-1"
+    assert res[txid] == "first-deposit-1"
+    assert res[sec_txid] == "second-deposit"
+    assert res[spend_txid] == "spend-tx-1"
+    assert res[inexistent_txid] == "inex_txid"
+    assert res[inexistent_outpoint] == "inex_outpoint"
+    assert res[random_address] == "bitcoind-addr"