cyberia/research/events/backend/cyber_valley/shaman_verification/test_contract_service.py

python
ฯ€ 0.0% 161 words ยท 1 min 
"""Tests for ContractService to verify role granting works correctly."""

import os

import pytest
from web3 import Web3

from .contract_service import ContractService

# Test address
TEST_SHAMAN_ADDRESS = "0xA84036A18ecd8f4F3D21ca7f85BEcC033571b15e"

pytestmark = pytest.mark.skipif(
    os.environ.get("RUN_CONTRACT_SERVICE_TESTS") != "1",
    reason=(
        "Requires a running Ethereum JSON-RPC node and deployed contracts. "
        "Set RUN_CONTRACT_SERVICE_TESTS=1 to enable."
    ),
)


@pytest.fixture
def contract_service() -> ContractService:
    """Create a ContractService instance."""
    return ContractService()


@pytest.fixture
def checksum_address() -> str:
    """Convert test address to checksum format."""
    return Web3.to_checksum_address(TEST_SHAMAN_ADDRESS)


def test_backend_has_role(contract_service: ContractService) -> None:
    """Test that backend EOA has BACKEND_ROLE."""
    has_role = contract_service.check_backend_has_role()
    assert has_role, "Backend EOA should have BACKEND_ROLE"


def test_grant_verified_shaman_role(
    contract_service: ContractService, checksum_address: str
) -> None:
    """Test granting VERIFIED_SHAMAN_ROLE to an address."""
    # First, ensure the address doesn't have the role
    verified_shaman_role = (
        contract_service.contract.functions.VERIFIED_SHAMAN_ROLE().call()
    )

    # Grant the role
    success, error = contract_service.grant_verified_shaman_role(checksum_address)
    assert success, f"Failed to grant role: {error}"

    # Verify the role was granted
    has_role_after = contract_service.contract.functions.hasRole(
        verified_shaman_role, checksum_address
    ).call()
    assert has_role_after, "Address should have VERIFIED_SHAMAN_ROLE after granting"

Graph