solana-program-library/stake-pool/py/tests/test_create_update_token_me...

68 lines
3.2 KiB
Python

import pytest
from stake_pool.actions import create_all, create_token_metadata, update_token_metadata
from stake_pool.state import Fee, StakePool
from solana.rpc.commitment import Confirmed
from solana.rpc.async_api import AsyncClient
from solana.keypair import Keypair
from stake_pool.constants import find_metadata_account
from solana.utils.helpers import decode_byte_string
@pytest.mark.asyncio
async def test_create_metadata_success(async_client: AsyncClient, payer: Keypair):
fee = Fee(numerator=1, denominator=1000)
referral_fee = 20
(stake_pool_address, _validator_list_address, _) = await create_all(async_client, payer, fee, referral_fee)
resp = await async_client.get_account_info(stake_pool_address, commitment=Confirmed)
data = resp['result']['value']['data']
stake_pool = StakePool.decode(data[0], data[1])
name = "test_name"
symbol = "SYM"
uri = "test_uri"
await create_token_metadata(async_client, payer, stake_pool_address, name, symbol, uri)
(metadata_program_address, _seed) = find_metadata_account(stake_pool.pool_mint)
resp = await async_client.get_account_info(metadata_program_address, commitment=Confirmed)
data = resp['result']['value']['data']
raw_data = decode_byte_string(data[0], data[1])
assert name == str(raw_data[69:101], "utf-8")[:len(name)]
assert symbol == str(raw_data[105:115], "utf-8")[:len(symbol)]
assert uri == str(raw_data[119:319], "utf-8")[:len(uri)]
@pytest.mark.asyncio
async def test_update_metadata_success(async_client: AsyncClient, payer: Keypair):
fee = Fee(numerator=1, denominator=1000)
referral_fee = 20
(stake_pool_address, _validator_list_address, _) = await create_all(async_client, payer, fee, referral_fee)
resp = await async_client.get_account_info(stake_pool_address, commitment=Confirmed)
data = resp['result']['value']['data']
stake_pool = StakePool.decode(data[0], data[1])
name = "test_name"
symbol = "SYM"
uri = "test_uri"
await create_token_metadata(async_client, payer, stake_pool_address, name, symbol, uri)
(metadata_program_address, _seed) = find_metadata_account(stake_pool.pool_mint)
resp = await async_client.get_account_info(metadata_program_address, commitment=Confirmed)
data = resp['result']['value']['data']
raw_data = decode_byte_string(data[0], data[1])
assert name == str(raw_data[69:101], "utf-8")[:len(name)]
assert symbol == str(raw_data[105:115], "utf-8")[:len(symbol)]
assert uri == str(raw_data[119:319], "utf-8")[:len(uri)]
updated_name = "updated_name"
updated_symbol = "USM"
updated_uri = "updated_uri"
await update_token_metadata(async_client, payer, stake_pool_address, updated_name, updated_symbol, updated_uri)
(metadata_program_address, _seed) = find_metadata_account(stake_pool.pool_mint)
resp = await async_client.get_account_info(metadata_program_address, commitment=Confirmed)
data = resp['result']['value']['data']
raw_data = decode_byte_string(data[0], data[1])
assert updated_name == str(raw_data[69:101], "utf-8")[:len(updated_name)]
assert updated_symbol == str(raw_data[105:115], "utf-8")[:len(updated_symbol)]
assert updated_uri == str(raw_data[119:319], "utf-8")[:len(updated_uri)]