chia-blockchain 2.4.4__py3-none-any.whl

chia/_tests/wallet/simple_sync/test_simple_sync_protocol.py

@@ -0,0 +1,719 @@
+from __future__ import annotations
+
+import asyncio
+from typing import List
+
+import pytest
+from clvm.casts import int_to_bytes
+from colorlog import getLogger
+
+from chia._tests.connection_utils import add_dummy_connection
+from chia._tests.util.setup_nodes import OldSimulatorsAndWallets
+from chia._tests.util.time_out_assert import time_out_assert
+from chia.consensus.block_rewards import calculate_base_farmer_reward, calculate_pool_reward
+from chia.protocols import wallet_protocol
+from chia.protocols.full_node_protocol import RespondTransaction
+from chia.protocols.protocol_message_types import ProtocolMessageTypes
+from chia.protocols.wallet_protocol import CoinStateUpdate, RespondToCoinUpdates
+from chia.server.outbound_message import Message, NodeType
+from chia.simulator.simulator_protocol import FarmNewBlockProtocol, ReorgProtocol
+from chia.types.blockchain_format.coin import Coin
+from chia.types.blockchain_format.sized_bytes import bytes32
+from chia.types.condition_opcodes import ConditionOpcode
+from chia.types.condition_with_args import ConditionWithArgs
+from chia.types.peer_info import PeerInfo
+from chia.util.ints import uint32, uint64
+from chia.wallet.util.tx_config import DEFAULT_TX_CONFIG
+from chia.wallet.wallet import Wallet
+
+log = getLogger(__name__)
+
+zero_ph = bytes32(32 * b"\0")
+
+
+async def get_all_messages_in_queue(queue: asyncio.Queue[Message]) -> List[Message]:
+    all_messages = []
+    await asyncio.sleep(2)
+    while not queue.empty():
+        message = await queue.get()
+        all_messages.append(message)
+    return all_messages
+
+
+@pytest.mark.anyio
+async def test_subscribe_for_ph(simulator_and_wallet: OldSimulatorsAndWallets, self_hostname: str) -> None:
+    num_blocks = 4
+    full_nodes, wallets, _ = simulator_and_wallet
+    full_node_api = full_nodes[0]
+    wallet_node, server_2 = wallets[0]
+    fn_server = full_node_api.full_node.server
+
+    await server_2.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    incoming_queue, peer_id = await add_dummy_connection(fn_server, self_hostname, 12312, NodeType.WALLET)
+
+    junk_ph = bytes32(32 * b"\a")
+    fake_wallet_peer = fn_server.all_connections[peer_id]
+    msg = wallet_protocol.RegisterForPhUpdates([zero_ph], uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, fake_wallet_peer)
+
+    assert msg_response.type == ProtocolMessageTypes.respond_to_ph_update.value
+    data_response = RespondToCoinUpdates.from_bytes(msg_response.data)
+    assert data_response.coin_states == []
+
+    # Farm few more with reward
+    for i in range(num_blocks):
+        if i == num_blocks - 1:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(junk_ph))
+        else:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    msg = wallet_protocol.RegisterForPhUpdates([zero_ph], uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, fake_wallet_peer)
+    assert msg_response.type == ProtocolMessageTypes.respond_to_ph_update.value
+    data_response = RespondToCoinUpdates.from_bytes(msg_response.data)
+    # we have already subscribed to this puzzle hash, it will be ignored
+    # we still receive the updates (see below)
+    assert data_response.coin_states == []
+
+    # Farm more rewards to check the incoming queue for the updates
+    for i in range(num_blocks):
+        if i == num_blocks - 1:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(junk_ph))
+        else:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+
+    zero_coin = await full_node_api.full_node.coin_store.get_coin_states_by_puzzle_hashes(True, {zero_ph})
+    all_zero_coin = set(zero_coin)
+    notified_zero_coins = set()
+
+    for message in all_messages:
+        if message.type == ProtocolMessageTypes.coin_state_update.value:
+            coin_state_update = CoinStateUpdate.from_bytes(message.data)
+            assert len(coin_state_update.items) == 2  # 2 per height farmer / pool reward
+            for coin_state in coin_state_update.items:
+                notified_zero_coins.add(coin_state)
+
+    assert all_zero_coin == notified_zero_coins
+
+    # Test subscribing to more coins
+    one_ph = bytes32(32 * b"\1")
+    msg = wallet_protocol.RegisterForPhUpdates([one_ph], uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, fake_wallet_peer)
+    peak = full_node_api.full_node.blockchain.get_peak()
+
+    for i in range(num_blocks):
+        if i == num_blocks - 1:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(junk_ph))
+        else:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    for i in range(num_blocks):
+        if i == num_blocks - 1:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(one_ph))
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(junk_ph))
+        else:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(one_ph))
+
+    assert peak is not None
+    zero_coins = await full_node_api.full_node.coin_store.get_coin_states_by_puzzle_hashes(
+        True, {zero_ph}, uint32(peak.height + 1)
+    )
+    one_coins = await full_node_api.full_node.coin_store.get_coin_states_by_puzzle_hashes(True, {one_ph})
+
+    all_coins = set(zero_coins)
+    all_coins.update(one_coins)
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+
+    notified_all_coins = set()
+
+    for message in all_messages:
+        if message.type == ProtocolMessageTypes.coin_state_update.value:
+            coin_state_update = CoinStateUpdate.from_bytes(message.data)
+            assert len(coin_state_update.items) == 2  # 2 per height farmer / pool reward
+            for coin_state in coin_state_update.items:
+                notified_all_coins.add(coin_state)
+
+    assert all_coins == notified_all_coins
+
+    wallet = wallet_node.wallet_state_manager.wallets[uint32(1)]
+    assert isinstance(wallet, Wallet)
+    puzzle_hash = await wallet.get_new_puzzlehash()
+
+    for i in range(num_blocks):
+        if i == num_blocks - 1:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(puzzle_hash))
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(junk_ph))
+        else:
+            await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(puzzle_hash))
+
+    funds = sum(
+        calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks + 1)
+    )
+    fn_amount = sum(
+        cr.coin.amount
+        for cr in await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hash(False, puzzle_hash)
+    )
+
+    await time_out_assert(20, wallet.get_confirmed_balance, funds)
+    assert funds == fn_amount
+
+    msg_1 = wallet_protocol.RegisterForPhUpdates([puzzle_hash], uint32(0))
+    msg_response_1 = await full_node_api.register_interest_in_puzzle_hash(msg_1, fake_wallet_peer)
+    assert msg_response_1.type == ProtocolMessageTypes.respond_to_ph_update.value
+    data_response_1 = RespondToCoinUpdates.from_bytes(msg_response_1.data)
+    assert len(data_response_1.coin_states) == 2 * num_blocks  # 2 per height farmer / pool reward
+
+    await full_node_api.wait_for_wallet_synced(wallet_node=wallet_node, timeout=20)
+
+    async with wallet.wallet_state_manager.new_action_scope(DEFAULT_TX_CONFIG, push=True) as action_scope:
+        await wallet.generate_signed_transaction(uint64(10), puzzle_hash, action_scope, uint64(0))
+    [tx_record] = action_scope.side_effects.transactions
+    assert tx_record.spend_bundle is not None
+    assert len(tx_record.spend_bundle.removals()) == 1
+    spent_coin = tx_record.spend_bundle.removals()[0]
+    assert spent_coin.puzzle_hash == puzzle_hash
+
+    await full_node_api.process_transaction_records(records=[tx_record])
+
+    # Let's make sure the wallet can handle a non ephemeral launcher
+    from chia.wallet.puzzles.singleton_top_layer import SINGLETON_LAUNCHER_HASH
+
+    await full_node_api.wait_for_wallet_synced(wallet_node=wallet_node, timeout=20)
+
+    async with wallet.wallet_state_manager.new_action_scope(DEFAULT_TX_CONFIG, push=True) as action_scope:
+        await wallet.generate_signed_transaction(uint64(10), SINGLETON_LAUNCHER_HASH, action_scope, uint64(0))
+
+    await full_node_api.process_transaction_records(records=action_scope.side_effects.transactions)
+
+    await full_node_api.wait_for_wallet_synced(wallet_node=wallet_node, timeout=20)
+
+    # Send a transaction to make sure the wallet is still running
+    async with wallet.wallet_state_manager.new_action_scope(DEFAULT_TX_CONFIG, push=True) as action_scope:
+        await wallet.generate_signed_transaction(uint64(10), junk_ph, action_scope, uint64(0))
+
+    await full_node_api.process_transaction_records(records=action_scope.side_effects.transactions)
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+
+    notified_state = None
+
+    for message in all_messages:
+        if message.type == ProtocolMessageTypes.coin_state_update.value:
+            coin_state_update = CoinStateUpdate.from_bytes(message.data)
+            for coin_state in coin_state_update.items:
+                if coin_state.coin.name() == spent_coin.name():
+                    notified_state = coin_state
+
+    assert notified_state is not None
+    assert notified_state.coin == spent_coin
+    assert notified_state.spent_height is not None
+
+
+@pytest.mark.anyio
+async def test_subscribe_for_coin_id(simulator_and_wallet: OldSimulatorsAndWallets, self_hostname: str) -> None:
+    num_blocks = 4
+    full_nodes, wallets, _ = simulator_and_wallet
+    full_node_api = full_nodes[0]
+    wallet_node, server_2 = wallets[0]
+    fn_server = full_node_api.full_node.server
+    standard_wallet = wallet_node.wallet_state_manager.wallets[uint32(1)]
+    assert isinstance(standard_wallet, Wallet)
+    puzzle_hash = await standard_wallet.get_new_puzzlehash()
+
+    await server_2.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    incoming_queue, peer_id = await add_dummy_connection(fn_server, self_hostname, 12312, NodeType.WALLET)
+
+    fake_wallet_peer = fn_server.all_connections[peer_id]
+
+    # Farm to create a coin that we'll track
+    for _ in range(num_blocks):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(puzzle_hash))
+
+    funds = sum(
+        calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks)
+    )
+
+    await time_out_assert(20, standard_wallet.get_confirmed_balance, funds)
+
+    my_coins = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hash(True, puzzle_hash)
+    coin_to_spend = my_coins[0].coin
+
+    msg = wallet_protocol.RegisterForCoinUpdates([coin_to_spend.name()], uint32(0))
+    msg_response = await full_node_api.register_interest_in_coin(msg, fake_wallet_peer)
+    assert msg_response is not None
+    assert msg_response.type == ProtocolMessageTypes.respond_to_coin_update.value
+    data_response = RespondToCoinUpdates.from_bytes(msg_response.data)
+    assert data_response.coin_states[0].coin == coin_to_spend
+
+    coins = set()
+    coins.add(coin_to_spend)
+    async with standard_wallet.wallet_state_manager.new_action_scope(DEFAULT_TX_CONFIG, push=True) as action_scope:
+        await standard_wallet.generate_signed_transaction(uint64(10), puzzle_hash, action_scope, uint64(0), coins=coins)
+
+    await full_node_api.process_transaction_records(records=action_scope.side_effects.transactions)
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+
+    notified_coins = set()
+    for message in all_messages:
+        if message.type == ProtocolMessageTypes.coin_state_update.value:
+            coin_state_update = CoinStateUpdate.from_bytes(message.data)
+            for coin_state in coin_state_update.items:
+                notified_coins.add(coin_state.coin)
+                assert coin_state.spent_height is not None
+
+    assert notified_coins == coins
+
+    # Test getting notification for coin that is about to be created
+    await full_node_api.wait_for_wallet_synced(wallet_node=wallet_node, timeout=20)
+
+    async with standard_wallet.wallet_state_manager.new_action_scope(DEFAULT_TX_CONFIG, push=False) as action_scope:
+        await standard_wallet.generate_signed_transaction(uint64(10), puzzle_hash, action_scope, uint64(0))
+
+    [tx_record] = action_scope.side_effects.transactions
+
+    added_target = None
+    assert tx_record.spend_bundle is not None
+    for coin in tx_record.spend_bundle.additions():
+        if coin.puzzle_hash == puzzle_hash:
+            added_target = coin
+
+    assert added_target is not None
+
+    msg = wallet_protocol.RegisterForCoinUpdates([added_target.name()], uint32(0))
+    msg_response = await full_node_api.register_interest_in_coin(msg, fake_wallet_peer)
+    assert msg_response is not None
+    assert msg_response.type == ProtocolMessageTypes.respond_to_coin_update.value
+    data_response = RespondToCoinUpdates.from_bytes(msg_response.data)
+    assert len(data_response.coin_states) == 0
+
+    [tx_record] = await standard_wallet.wallet_state_manager.add_pending_transactions([tx_record])
+
+    await full_node_api.process_transaction_records(records=[tx_record])
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+
+    notified_state = None
+
+    for message in all_messages:
+        if message.type == ProtocolMessageTypes.coin_state_update.value:
+            coin_state_update = CoinStateUpdate.from_bytes(message.data)
+            for coin_state in coin_state_update.items:
+                if coin_state.coin.name() == added_target.name():
+                    notified_state = coin_state
+
+    assert notified_state is not None
+    assert notified_state.coin == added_target
+    assert notified_state.spent_height is None
+
+
+@pytest.mark.anyio
+@pytest.mark.standard_block_tools
+async def test_subscribe_for_ph_reorg(simulator_and_wallet: OldSimulatorsAndWallets, self_hostname: str) -> None:
+    num_blocks = 4
+    long_blocks = 20
+    full_nodes, wallets, _ = simulator_and_wallet
+    full_node_api = full_nodes[0]
+    wallet_node, server_2 = wallets[0]
+    fn_server = full_node_api.full_node.server
+    standard_wallet = wallet_node.wallet_state_manager.wallets[uint32(1)]
+    assert isinstance(standard_wallet, Wallet)
+    puzzle_hash = await standard_wallet.get_new_puzzlehash()
+
+    await server_2.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    incoming_queue, peer_id = await add_dummy_connection(fn_server, self_hostname, 12312, NodeType.WALLET)
+
+    fake_wallet_peer = fn_server.all_connections[peer_id]
+
+    # Farm to create a coin that we'll track
+    for _ in range(num_blocks):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    for _ in range(long_blocks):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    msg = wallet_protocol.RegisterForPhUpdates([puzzle_hash], uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, fake_wallet_peer)
+    assert msg_response is not None
+    await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(puzzle_hash))
+
+    for _ in range(num_blocks):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    expected_height = uint32(long_blocks + 2 * num_blocks + 1)
+    await time_out_assert(20, full_node_api.full_node.blockchain.get_peak_height, expected_height)
+
+    coin_records = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hash(True, puzzle_hash)
+    assert len(coin_records) > 0
+    fork_height = uint32(expected_height - num_blocks - 5)
+    req = ReorgProtocol(fork_height, uint32(expected_height + 5), zero_ph, None)
+    await full_node_api.reorg_from_index_to_new_index(req)
+
+    coin_records = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hash(True, puzzle_hash)
+    assert coin_records == []
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+
+    coin_update_messages: List[CoinStateUpdate] = []
+    for message in all_messages:
+        if message.type == ProtocolMessageTypes.coin_state_update.value:
+            coin_state_update = CoinStateUpdate.from_bytes(message.data)
+            coin_update_messages.append(coin_state_update)
+
+    # First state is creation, second one is a reorg
+    assert len(coin_update_messages) == 2
+    first = coin_update_messages[0]
+
+    assert len(first.items) == 2
+    first_state_coin_1 = first.items[0]
+    assert first_state_coin_1.spent_height is None
+    assert first_state_coin_1.created_height is not None
+    first_state_coin_2 = first.items[1]
+    assert first_state_coin_2.spent_height is None
+    assert first_state_coin_2.created_height is not None
+
+    second = coin_update_messages[1]
+    assert second.fork_height == fork_height
+    assert len(second.items) == 2
+    second_state_coin_1 = second.items[0]
+    assert second_state_coin_1.spent_height is None
+    assert second_state_coin_1.created_height is None
+    second_state_coin_2 = second.items[1]
+    assert second_state_coin_2.spent_height is None
+    assert second_state_coin_2.created_height is None
+
+
+@pytest.mark.anyio
+async def test_subscribe_for_coin_id_reorg(simulator_and_wallet: OldSimulatorsAndWallets, self_hostname: str) -> None:
+    num_blocks = 4
+    long_blocks = 20
+    full_nodes, wallets, _ = simulator_and_wallet
+    full_node_api = full_nodes[0]
+    wallet_node, server_2 = wallets[0]
+    fn_server = full_node_api.full_node.server
+    standard_wallet = wallet_node.wallet_state_manager.wallets[uint32(1)]
+    assert isinstance(standard_wallet, Wallet)
+    puzzle_hash = await standard_wallet.get_new_puzzlehash()
+
+    await server_2.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    incoming_queue, peer_id = await add_dummy_connection(fn_server, self_hostname, 12312, NodeType.WALLET)
+
+    fake_wallet_peer = fn_server.all_connections[peer_id]
+
+    # Farm to create a coin that we'll track
+    for _ in range(num_blocks):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    for _ in range(long_blocks):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(puzzle_hash))
+
+    for _ in range(num_blocks):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(zero_ph))
+
+    expected_height = uint32(long_blocks + 2 * num_blocks + 1)
+    await time_out_assert(20, full_node_api.full_node.blockchain.get_peak_height, expected_height)
+
+    coin_records = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hash(True, puzzle_hash)
+    assert len(coin_records) > 0
+
+    for coin_rec in coin_records:
+        msg = wallet_protocol.RegisterForCoinUpdates([coin_rec.name], uint32(0))
+        msg_response = await full_node_api.register_interest_in_coin(msg, fake_wallet_peer)
+        assert msg_response is not None
+
+    fork_height = uint32(expected_height - num_blocks - 5)
+    req = ReorgProtocol(fork_height, uint32(expected_height + 5), zero_ph, None)
+    await full_node_api.reorg_from_index_to_new_index(req)
+
+    coin_records = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hash(True, puzzle_hash)
+    assert coin_records == []
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+
+    coin_update_messages: List[CoinStateUpdate] = []
+    for message in all_messages:
+        if message.type == ProtocolMessageTypes.coin_state_update.value:
+            coin_state_update = CoinStateUpdate.from_bytes(message.data)
+            coin_update_messages.append(coin_state_update)
+
+    assert len(coin_update_messages) == 1
+    update = coin_update_messages[0]
+    coin_states = update.items
+    assert len(coin_states) == 2
+    first_coin = coin_states[0]
+    assert first_coin.spent_height is None
+    assert first_coin.created_height is None
+    second_coin = coin_states[1]
+    assert second_coin.spent_height is None
+    assert second_coin.created_height is None
+
+
+@pytest.mark.anyio
+async def test_subscribe_for_hint(simulator_and_wallet: OldSimulatorsAndWallets, self_hostname: str) -> None:
+    num_blocks = 4
+    full_nodes, wallets, bt = simulator_and_wallet
+    full_node_api = full_nodes[0]
+    wallet_node, server_2 = wallets[0]
+    fn_server = full_node_api.full_node.server
+
+    await server_2.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    incoming_queue, peer_id = await add_dummy_connection(fn_server, self_hostname, 12312, NodeType.WALLET)
+
+    wt = bt.get_pool_wallet_tool()
+    ph = wt.get_new_puzzlehash()
+    for _ in range(num_blocks):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph))
+
+    await asyncio.sleep(6)
+    coins = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hashes(False, [ph])
+    coin_spent = coins[0].coin
+    hint_puzzle_hash = 32 * b"\2"
+    amount = uint64(1)
+    amount_bin = int_to_bytes(1)
+    hint = bytes32(32 * b"\5")
+
+    fake_wallet_peer = fn_server.all_connections[peer_id]
+    msg = wallet_protocol.RegisterForPhUpdates([hint], uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, fake_wallet_peer)
+    assert msg_response.type == ProtocolMessageTypes.respond_to_ph_update.value
+    data_response = RespondToCoinUpdates.from_bytes(msg_response.data)
+    assert len(data_response.coin_states) == 0
+
+    condition_dict = {
+        ConditionOpcode.CREATE_COIN: [
+            ConditionWithArgs(ConditionOpcode.CREATE_COIN, [hint_puzzle_hash, amount_bin, hint])
+        ]
+    }
+    await full_node_api.wait_for_wallet_synced(wallet_node=wallet_node, timeout=20)
+
+    tx = wt.generate_signed_transaction(uint64(10), wt.get_new_puzzlehash(), coin_spent, condition_dic=condition_dict)
+    await full_node_api.respond_transaction(RespondTransaction(tx), fake_wallet_peer)
+
+    await full_node_api.process_spend_bundles(bundles=[tx])
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+
+    notified_state = None
+
+    for message in all_messages:
+        if message.type == ProtocolMessageTypes.coin_state_update.value:
+            coin_state_update = CoinStateUpdate.from_bytes(message.data)
+            notified_state = coin_state_update
+            break
+
+    assert notified_state is not None
+    assert notified_state.items[0].coin == Coin(coin_spent.name(), hint_puzzle_hash, amount)
+
+    msg = wallet_protocol.RegisterForPhUpdates([hint], uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, fake_wallet_peer)
+    assert msg_response.type == ProtocolMessageTypes.respond_to_ph_update.value
+    response = RespondToCoinUpdates.from_bytes(msg_response.data)
+    # we have already subscribed to this puzzle hash. The full node will
+    # ignore the duplicate
+    assert response.coin_states == []
+
+
+@pytest.mark.anyio
+async def test_subscribe_for_puzzle_hash_coin_hint_duplicates(
+    simulator_and_wallet: OldSimulatorsAndWallets, self_hostname: str
+) -> None:
+    [full_node_api], [[_, wallet_server]], bt = simulator_and_wallet
+    full_node_server = full_node_api.full_node.server
+
+    await wallet_server.start_client(PeerInfo(self_hostname, full_node_server.get_port()), None)
+
+    wt = bt.get_pool_wallet_tool()
+    ph = wt.get_new_puzzlehash()
+    await full_node_api.farm_blocks_to_puzzlehash(4, ph)
+    coins = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hashes(False, [ph])
+    wallet_connection = full_node_server.all_connections[wallet_server.node_id]
+
+    # Create a coin which is hinted with its own destination puzzle hash
+    tx = wt.generate_signed_transaction(
+        uint64(10),
+        wt.get_new_puzzlehash(),
+        coins[0].coin,
+        condition_dic={
+            ConditionOpcode.CREATE_COIN: [ConditionWithArgs(ConditionOpcode.CREATE_COIN, [ph, int_to_bytes(1), ph])]
+        },
+    )
+    await full_node_api.respond_transaction(RespondTransaction(tx), wallet_connection)
+    await full_node_api.process_spend_bundles(bundles=[tx])
+    # Query the coin states and make sure it doesn't contain duplicated entries
+    msg = wallet_protocol.RegisterForPhUpdates([ph], uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, wallet_connection)
+    assert msg_response.type == ProtocolMessageTypes.respond_to_ph_update.value
+    response = RespondToCoinUpdates.from_bytes(msg_response.data)
+    assert len(response.coin_states) > 0
+    assert len(set(response.coin_states)) == len(response.coin_states)
+
+
+@pytest.mark.anyio
+@pytest.mark.standard_block_tools
+async def test_subscribe_for_hint_long_sync(
+    wallet_two_node_simulator: OldSimulatorsAndWallets, self_hostname: str
+) -> None:
+    full_nodes, wallets, bt = wallet_two_node_simulator
+    full_node_api = full_nodes[0]
+    full_node_api_1 = full_nodes[1]
+    wallet_node, server_2 = wallets[0]
+    fn_server = full_node_api.full_node.server
+    fn_server_1 = full_node_api_1.full_node.server
+
+    await server_2.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    incoming_queue, peer_id = await add_dummy_connection(fn_server, self_hostname, 12312, NodeType.WALLET)
+    incoming_queue_1, peer_id_1 = await add_dummy_connection(fn_server_1, self_hostname, 12313, NodeType.WALLET)
+
+    wt = bt.get_pool_wallet_tool()
+    ph = wt.get_new_puzzlehash()
+    for _ in range(2):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph))
+
+    coins = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hashes(False, [ph])
+    coin_spent = coins[0].coin
+    hint_puzzle_hash = 32 * b"\2"
+    amount = uint64(1)
+    amount_bin = int_to_bytes(1)
+    hint = bytes32(32 * b"\5")
+
+    fake_wallet_peer = fn_server.all_connections[peer_id]
+    fake_wallet_peer_1 = fn_server_1.all_connections[peer_id_1]
+    msg = wallet_protocol.RegisterForPhUpdates([hint], uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, fake_wallet_peer)
+    await full_node_api_1.register_interest_in_puzzle_hash(msg, fake_wallet_peer_1)
+
+    assert msg_response.type == ProtocolMessageTypes.respond_to_ph_update.value
+    data_response = RespondToCoinUpdates.from_bytes(msg_response.data)
+    assert len(data_response.coin_states) == 0
+
+    condition_dict = {
+        ConditionOpcode.CREATE_COIN: [
+            ConditionWithArgs(ConditionOpcode.CREATE_COIN, [hint_puzzle_hash, amount_bin, hint])
+        ]
+    }
+    await full_node_api.wait_for_wallet_synced(wallet_node=wallet_node, timeout=20)
+
+    tx = wt.generate_signed_transaction(uint64(10), wt.get_new_puzzlehash(), coin_spent, condition_dic=condition_dict)
+    await full_node_api.respond_transaction(RespondTransaction(tx), fake_wallet_peer)
+
+    await full_node_api.process_spend_bundles(bundles=[tx])
+
+    # Create more blocks than recent "short_sync_blocks_behind_threshold" so that node enters batch
+    blocks_to_farm = full_node_api.full_node.config.get("short_sync_blocks_behind_threshold", 100)
+    for _ in range(blocks_to_farm):
+        await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph))
+
+    node1_height = full_node_api_1.full_node.blockchain.get_peak_height()
+    assert node1_height is None
+
+    await fn_server_1.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    node0_height = full_node_api.full_node.blockchain.get_peak_height()
+    await time_out_assert(60, full_node_api_1.full_node.blockchain.get_peak_height, node0_height)
+
+    all_messages = await get_all_messages_in_queue(incoming_queue)
+    all_messages_1 = await get_all_messages_in_queue(incoming_queue_1)
+
+    def check_messages_for_hint(messages: List[Message]) -> None:
+        notified_state = None
+
+        for message in messages:
+            if message.type == ProtocolMessageTypes.coin_state_update.value:
+                coin_state_update = CoinStateUpdate.from_bytes(message.data)
+                notified_state = coin_state_update
+                break
+
+        assert notified_state is not None
+        assert notified_state.items[0].coin == Coin(coin_spent.name(), hint_puzzle_hash, amount)
+
+    check_messages_for_hint(all_messages)
+    check_messages_for_hint(all_messages_1)
+
+
+@pytest.mark.anyio
+async def test_ph_subscribe_limits(simulator_and_wallet: OldSimulatorsAndWallets, self_hostname: str) -> None:
+    full_nodes, wallets, _ = simulator_and_wallet
+    full_node_api = full_nodes[0]
+    _, server_2 = wallets[0]
+    fn_server = full_node_api.full_node.server
+    await server_2.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    con = list(fn_server.all_connections.values())[0]
+    phs = []
+    phs.append(bytes32(32 * b"\0"))
+    phs.append(bytes32(32 * b"\1"))
+    phs.append(bytes32(32 * b"\2"))
+    phs.append(bytes32(32 * b"\3"))
+    phs.append(bytes32(32 * b"\4"))
+    phs.append(bytes32(32 * b"\5"))
+    phs.append(bytes32(32 * b"\6"))
+    full_node_api.full_node.config["max_subscribe_items"] = 2
+    assert full_node_api.is_trusted(con) is False
+    msg = wallet_protocol.RegisterForPhUpdates(phs, uint32(0))
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, con)
+    assert msg_response.type == ProtocolMessageTypes.respond_to_ph_update.value
+    s = full_node_api.full_node.subscriptions
+    assert s.puzzle_subscription_count() == 2
+    assert s.has_puzzle_subscription(phs[0])
+    assert s.has_puzzle_subscription(phs[1])
+    assert not s.has_puzzle_subscription(phs[2])
+    assert not s.has_puzzle_subscription(phs[3])
+    full_node_api.full_node.config["trusted_max_subscribe_items"] = 4
+    full_node_api.full_node.config["trusted_peers"] = {server_2.node_id.hex(): server_2.node_id.hex()}
+    assert full_node_api.is_trusted(con) is True
+    msg_response = await full_node_api.register_interest_in_puzzle_hash(msg, con)
+    assert msg_response.type == ProtocolMessageTypes.respond_to_ph_update.value
+    assert s.puzzle_subscription_count() == 4
+    assert s.has_puzzle_subscription(phs[0])
+    assert s.has_puzzle_subscription(phs[1])
+    assert s.has_puzzle_subscription(phs[2])
+    assert s.has_puzzle_subscription(phs[3])
+    assert not s.has_puzzle_subscription(phs[4])
+    assert not s.has_puzzle_subscription(phs[5])
+
+
+@pytest.mark.anyio
+async def test_coin_subscribe_limits(simulator_and_wallet: OldSimulatorsAndWallets, self_hostname: str) -> None:
+    full_nodes, wallets, _ = simulator_and_wallet
+    full_node_api = full_nodes[0]
+    _, server_2 = wallets[0]
+    fn_server = full_node_api.full_node.server
+    await server_2.start_client(PeerInfo(self_hostname, fn_server.get_port()), None)
+    con = list(fn_server.all_connections.values())[0]
+    coins = []
+    coins.append(bytes32(32 * b"\0"))
+    coins.append(bytes32(32 * b"\1"))
+    coins.append(bytes32(32 * b"\2"))
+    coins.append(bytes32(32 * b"\3"))
+    coins.append(bytes32(32 * b"\4"))
+    coins.append(bytes32(32 * b"\5"))
+    coins.append(bytes32(32 * b"\6"))
+    full_node_api.full_node.config["max_subscribe_items"] = 2
+    assert full_node_api.is_trusted(con) is False
+    msg = wallet_protocol.RegisterForCoinUpdates(coins, uint32(0))
+    msg_response = await full_node_api.register_interest_in_coin(msg, con)
+    assert msg_response.type == ProtocolMessageTypes.respond_to_coin_update.value
+    s = full_node_api.full_node.subscriptions
+    assert s.coin_subscription_count() == 2
+    assert s.has_coin_subscription(coins[0])
+    assert s.has_coin_subscription(coins[1])
+    assert not s.has_coin_subscription(coins[2])
+    assert not s.has_coin_subscription(coins[3])
+    full_node_api.full_node.config["trusted_max_subscribe_items"] = 4
+    full_node_api.full_node.config["trusted_peers"] = {server_2.node_id.hex(): server_2.node_id.hex()}
+    assert full_node_api.is_trusted(con) is True
+    msg_response = await full_node_api.register_interest_in_coin(msg, con)
+    assert msg_response.type == ProtocolMessageTypes.respond_to_coin_update.value
+    assert s.coin_subscription_count() == 4
+    assert s.has_coin_subscription(coins[0])
+    assert s.has_coin_subscription(coins[1])
+    assert s.has_coin_subscription(coins[2])
+    assert s.has_coin_subscription(coins[3])
+    assert not s.has_coin_subscription(coins[4])
+    assert not s.has_coin_subscription(coins[5])