add sync extrinsic unit test

Author: Roman

tests/unit_tests/extrinsics/test_registration.py

@@ -222,3 +222,124 @@ def test_burned_register_extrinsic(
         )
         # Assert
         assert result == expected_result, f"Test failed for test_id: {test_id}"
+
+
+def test_set_subnet_identity_extrinsic_is_success(mock_subtensor, mock_wallet, mocker):
+    """Verify that set_subnet_identity_extrinsic calls the correct functions and returns the correct result."""
+    # Preps
+    netuid = 123
+    subnet_name = "mock_subnet_name"
+    github_repo = "mock_github_repo"
+    subnet_contact = "mock_subnet_contact"
+    subnet_url = "mock_subnet_url"
+    discord = "mock_discord"
+    description = "mock_description"
+    additional = "mock_additional"
+
+    mocked_compose_call = mocker.patch.object(mock_subtensor.substrate, "compose_call")
+    mocked_sign_and_send_extrinsic = mocker.patch.object(
+        mock_subtensor, "sign_and_send_extrinsic", return_value=(True, "Success")
+    )
+
+    # Call
+    result = registration.set_subnet_identity_extrinsic(
+        subtensor=mock_subtensor,
+        wallet=mock_wallet,
+        netuid=netuid,
+        subnet_name=subnet_name,
+        github_repo=github_repo,
+        subnet_contact=subnet_contact,
+        subnet_url=subnet_url,
+        discord=discord,
+        description=description,
+        additional=additional,
+    )
+
+    # Asserts
+    mocked_compose_call.assert_called_once_with(
+        call_module="SubtensorModule",
+        call_function="set_subnet_identity",
+        call_params={
+            "hotkey": mock_wallet.hotkey.ss58_address,
+            "netuid": netuid,
+            "subnet_name": "mock_subnet_name",
+            "github_repo": "mock_github_repo",
+            "subnet_contact": "mock_subnet_contact",
+            "subnet_url": "mock_subnet_url",
+            "discord": "mock_discord",
+            "description": "mock_description",
+            "additional": "mock_additional",
+        },
+    )
+    mocked_sign_and_send_extrinsic.assert_called_once_with(
+        call=mocked_compose_call.return_value,
+        wallet=mock_wallet,
+        wait_for_inclusion=False,
+        wait_for_finalization=True,
+    )
+
+    assert result == (True, "Identities for subnet 123 are set.")
+
+
+def test_set_subnet_identity_extrinsic_is_failed(mock_subtensor, mock_wallet, mocker):
+    """Verify that set_subnet_identity_extrinsic calls the correct functions and returns False with bad result."""
+    # Preps
+    netuid = 123
+    subnet_name = "mock_subnet_name"
+    github_repo = "mock_github_repo"
+    subnet_contact = "mock_subnet_contact"
+    subnet_url = "mock_subnet_url"
+    discord = "mock_discord"
+    description = "mock_description"
+    additional = "mock_additional"
+
+    fake_error_message = "error message"
+
+    mocked_compose_call = mocker.patch.object(mock_subtensor.substrate, "compose_call")
+    mocked_sign_and_send_extrinsic = mocker.patch.object(
+        mock_subtensor,
+        "sign_and_send_extrinsic",
+        return_value=(False, fake_error_message),
+    )
+
+    # Call
+    result = registration.set_subnet_identity_extrinsic(
+        subtensor=mock_subtensor,
+        wallet=mock_wallet,
+        netuid=netuid,
+        subnet_name=subnet_name,
+        github_repo=github_repo,
+        subnet_contact=subnet_contact,
+        subnet_url=subnet_url,
+        discord=discord,
+        description=description,
+        additional=additional,
+    )
+
+    # Asserts
+    mocked_compose_call.assert_called_once_with(
+        call_module="SubtensorModule",
+        call_function="set_subnet_identity",
+        call_params={
+            "hotkey": mock_wallet.hotkey.ss58_address,
+            "netuid": netuid,
+            "subnet_name": "mock_subnet_name",
+            "github_repo": "mock_github_repo",
+            "subnet_contact": "mock_subnet_contact",
+            "subnet_url": "mock_subnet_url",
+            "discord": "mock_discord",
+            "description": "mock_description",
+            "additional": "mock_additional",
+        },
+    )
+    mocked_sign_and_send_extrinsic.assert_called_once_with(
+        call=mocked_compose_call.return_value,
+        wallet=mock_wallet,
+        wait_for_inclusion=False,
+        wait_for_finalization=True,
+    )
+
+    assert result == (
+        False,
+        f"Failed to set identity for subnet {netuid}: {fake_error_message}",
+    )