tests

Author: andreea-popescu-reef

bittensor/core/async_subtensor.py

@@ -32,6 +32,7 @@
 from bittensor.core.chain_data.chain_identity import ChainIdentity
 from bittensor.core.chain_data.delegate_info import DelegatedInfo
 from bittensor.core.chain_data.utils import (
+    decode_block,
     decode_metadata,
     decode_revealed_commitment,
     decode_revealed_commitment_with_hotkey,
@@ -55,6 +56,7 @@
     root_register_extrinsic,
 )
 from bittensor.core.extrinsics.asyncex.serving import (
+    get_last_bonds_reset,
     publish_metadata,
     get_metadata,
 )
@@ -1108,6 +1110,31 @@ async def get_commitment(
         except TypeError:
             return ""
 
+    async def get_last_commitment_bonds_reset_block(
+        self, netuid: int, uid: int
+    ) -> Optional[int]:
+        """
+        Retrieves the last block number when the bonds reset were triggered by publish_metadata for a specific neuron.
+        Arguments:
+            netuid (int): The unique identifier of the subnetwork.
+            uid (int): The unique identifier of the neuron.
+        Returns:
+            Optional[int]: The block number when the bonds were last reset, or None if not found.
+        """
+
+        metagraph = await self.metagraph(netuid)
+        try:
+            hotkey = metagraph.hotkeys[uid]  # type: ignore
+        except IndexError:
+            logging.error(
+                "Your uid is not in the hotkeys. Please double-check your UID."
+            )
+            return None
+        block = await get_last_bonds_reset(self, netuid, hotkey)
+        if block is None:
+            return None
+        return await decode_block(block)
+
     async def get_all_commitments(
         self,
         netuid: int,

bittensor/core/chain_data/utils.py

@@ -148,8 +148,7 @@ def decode_block(data: bytes) -> int:
     Returns:
         int: The decoded block.
     """
-
-    return int.from_bytes(block)
+    return int(data.value) if isinstance(data, ScaleBytes) else data
 
 
 def decode_revealed_commitment(encoded_data) -> tuple[int, str]:

bittensor/core/extrinsics/asyncex/serving.py

@@ -320,3 +320,21 @@ async def get_metadata(
             reuse_block_hash=reuse_block,
         )
     return commit_data
+
+
+async def get_last_bonds_reset(
+    subtensor: "AsyncSubtensor",
+    netuid: int,
+    hotkey: str,
+    block: Optional[int] = None,
+) -> bytes:
+    """Fetches the last bonds reset triggered at commitment from the blockchain for a given hotkey and netuid."""
+    async with subtensor.substrate:
+        block_hash = await subtensor.determine_block_hash(block)
+        block = subtensor.substrate.query(
+            module="Commitments",
+            storage_function="LastBondsReset",
+            params=[netuid, hotkey],
+            block_hash=block_hash,
+        )
+    return block

bittensor/core/subtensor_api/commitments.py

@@ -11,6 +11,9 @@ def __init__(self, subtensor: Union["_Subtensor", "_AsyncSubtensor"]):
         self.get_all_commitments = subtensor.get_all_commitments
         self.get_all_revealed_commitments = subtensor.get_all_revealed_commitments
         self.get_commitment = subtensor.get_commitment
+        self.get_last_commitment_bonds_reset_block = (
+            subtensor.get_last_commitment_bonds_reset_block
+        )
         self.get_current_weight_commit_info = subtensor.get_current_weight_commit_info
         self.get_revealed_commitment = subtensor.get_revealed_commitment
         self.get_revealed_commitment_by_hotkey = (

bittensor/core/subtensor_api/utils.py

@@ -39,6 +39,9 @@ def add_legacy_methods(subtensor: "SubtensorApi"):
     subtensor.get_children = subtensor._subtensor.get_children
     subtensor.get_children_pending = subtensor._subtensor.get_children_pending
     subtensor.get_commitment = subtensor._subtensor.get_commitment
+    subtensor.get_last_commitment_bonds_reset_block = (
+        subtensor._subtensor.get_last_commitment_bonds_reset_block
+    )
     subtensor.get_current_block = subtensor._subtensor.get_current_block
     subtensor.get_current_weight_commit_info = (
         subtensor._subtensor.get_current_weight_commit_info

tests/unit_tests/test_subtensor.py

@@ -1813,6 +1813,33 @@ def test_get_commitment(subtensor, mocker):
     assert result == expected_result
 
 
+def test_get_last_commitment_bonds_reset_block(subtensor, mocker):
+    """Successful get_last_commitment_bonds_reset_block call."""
+    # Preps
+    fake_netuid = 1
+    fake_uid = 2
+    fake_hotkey = "hotkey"
+    expected_result = 3
+
+    mocked_get_last_bonds_reset = mocker.patch.object(
+        subtensor_module, "get_last_bonds_reset"
+    )
+    mocked_get_last_bonds_reset.return_value = expected_result
+
+    mocked_metagraph = mocker.MagicMock()
+    subtensor.metagraph = mocked_metagraph
+    mocked_metagraph.return_value.hotkeys = {fake_uid: fake_hotkey}
+
+    # Call
+    result = subtensor.get_last_commitment_bonds_reset_block(
+        netuid=fake_netuid, uid=fake_uid
+    )
+
+    # Assertions
+    mocked_get_last_bonds_reset.assert_called_once()
+    assert result == expected_result
+
+
 def test_min_allowed_weights(subtensor, mocker):
     """Successful min_allowed_weights call."""
     fake_netuid = 1