Merge release/1.4.0's 0f49e5d into main

This commit is contained in:
Amine Khaldi 2022-06-21 15:46:31 +01:00
commit dd1a07965a
No known key found for this signature in database
GPG Key ID: B1C074FFC904E2D9
2 changed files with 639 additions and 39 deletions

View File

@ -29,9 +29,9 @@ from chia.wallet.nft_wallet.nft_puzzles import (
get_metadata_and_phs,
)
from chia.wallet.nft_wallet.uncurry_nft import UncurriedNFT
from chia.wallet.outer_puzzles import AssetType, construct_puzzle, match_puzzle
from chia.wallet.outer_puzzles import AssetType, construct_puzzle, match_puzzle, solve_puzzle
from chia.wallet.payment import Payment
from chia.wallet.puzzle_drivers import PuzzleInfo
from chia.wallet.puzzle_drivers import PuzzleInfo, Solver
from chia.wallet.puzzles.p2_delegated_puzzle_or_hidden_puzzle import (
DEFAULT_HIDDEN_PUZZLE_HASH,
calculate_synthetic_secret_key,
@ -939,7 +939,7 @@ class NFTWallet:
wallet = wallet_state_manager.main_wallet
else:
# cat offer
wallet = await wallet_state_manager.get_wallet_for_asset_id(offered_asset_id)
wallet = await wallet_state_manager.get_wallet_for_asset_id(offered_asset_id.hex())
if wallet.type() == WalletType.STANDARD_WALLET:
coin_amount_needed: int = offered_amount + royalty_amount + fee
@ -988,15 +988,36 @@ class NFTWallet:
for coin in txn.additions():
if coin.amount == royalty_amount:
royalty_coin = coin
parent_spend = txn.coin_spends[0]
break
assert royalty_coin
# make the royalty payment solution
# ((nft_launcher_id . ((ROYALTY_ADDRESS, royalty_amount, (ROYALTY_ADDRESS)))))
royalty_sol = Program.to([[requested_asset_id, [royalty_address, royalty_amount, [royalty_address]]]])
inner_royalty_sol = Program.to([[requested_asset_id, [royalty_address, royalty_amount, [royalty_address]]]])
if offered_asset_id is None:
offer_puzzle: Program = OFFER_MOD
royalty_sol = inner_royalty_sol
else:
offer_puzzle = construct_puzzle(driver_dict[offered_asset_id], OFFER_MOD)
# adapt royalty_sol to work with cat puzzle
royalty_coin_hex = (
"0x"
+ royalty_coin.parent_coin_info.hex()
+ royalty_coin.puzzle_hash.hex()
+ bytes(royalty_coin.amount).hex()
)
parent_spend_hex: str = "0x" + bytes(parent_spend).hex()
solver = Solver(
{
"coin": royalty_coin_hex,
"parent_spend": parent_spend_hex,
"siblings": "(" + royalty_coin_hex + ")",
"sibling_spends": "(" + parent_spend_hex + ")",
"sibling_puzzles": "()",
"sibling_solutions": "()",
}
)
royalty_sol = solve_puzzle(driver_dict[offered_asset_id], solver, OFFER_MOD, inner_royalty_sol)
royalty_spend = SpendBundle([CoinSpend(royalty_coin, offer_puzzle, royalty_sol)], G2Element())
total_spend_bundle = SpendBundle.aggregate([txn_spend_bundle, royalty_spend])

View File

@ -1,7 +1,6 @@
import asyncio
import logging
# from secrets import token_bytes
from secrets import token_bytes
from typing import Any, Optional
import pytest
@ -19,6 +18,7 @@ from chia.types.peer_info import PeerInfo
# from chia.util.bech32m import encode_puzzle_hash
# from chia.util.byte_types import hexstr_to_bytes
from chia.util.ints import uint16, uint32, uint64
from chia.wallet.cat_wallet.cat_wallet import CATWallet
from chia.wallet.did_wallet.did_wallet import DIDWallet
from chia.wallet.nft_wallet.nft_wallet import NFTWallet
from chia.wallet.outer_puzzles import create_asset_id, match_puzzle
@ -49,7 +49,7 @@ async def tx_in_pool(mempool: MempoolManager, tx_id: bytes32) -> bool:
@pytest.mark.asyncio
# @pytest.mark.skip
async def test_nft_offer_sell_nft(two_wallet_nodes: Any, trusted: Any) -> None:
num_blocks = 2
num_blocks = 5
full_nodes, wallets = two_wallet_nodes
full_node_api: FullNodeSimulator = full_nodes[0]
full_node_server = full_node_api.server
@ -60,7 +60,7 @@ async def test_nft_offer_sell_nft(two_wallet_nodes: Any, trusted: Any) -> None:
ph_maker = await wallet_maker.get_new_puzzlehash()
ph_taker = await wallet_taker.get_new_puzzlehash()
# token_ph = bytes32(token_bytes())
ph_token = bytes32(token_bytes())
if trusted:
wallet_node_maker.config["trusted_peers"] = {
@ -79,20 +79,16 @@ async def test_nft_offer_sell_nft(two_wallet_nodes: Any, trusted: Any) -> None:
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_taker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
funds = sum(
[calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks - 1)]
[calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks)]
)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_maker.get_confirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_confirmed_balance, funds)
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
did_wallet_maker: DIDWallet = await DIDWallet.create_new_did_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, uint64(1)
@ -105,13 +101,17 @@ async def test_nft_offer_sell_nft(two_wallet_nodes: Any, trusted: Any) -> None:
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, spend_bundle.name())
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(15, wallet_maker.get_pending_change_balance, 0)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, funds - 1)
await time_out_assert(10, wallet_maker.get_confirmed_balance, funds - 1)
hex_did_id = did_wallet_maker.get_my_DID()
did_id = bytes32.fromhex(hex_did_id)
target_puzhash = ph_maker
royalty_puzhash = ph_maker
royalty_percentage = uint16(200)
royalty_basis_pts = uint16(200)
nft_wallet_maker = await NFTWallet.create_new_nft_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, name="NFT WALLET DID 1", did_id=did_id
@ -122,13 +122,12 @@ async def test_nft_offer_sell_nft(two_wallet_nodes: Any, trusted: Any) -> None:
("h", "0xD4584AD463139FA8C0D9F68F4B59F185"),
]
)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, 5999999999999)
await time_out_assert(10, wallet_maker.get_confirmed_balance, 5999999999999)
sb = await nft_wallet_maker.generate_new_nft(
metadata,
target_puzhash,
royalty_puzhash,
royalty_percentage,
royalty_basis_pts,
did_id,
)
assert sb
@ -137,7 +136,7 @@ async def test_nft_offer_sell_nft(two_wallet_nodes: Any, trusted: Any) -> None:
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, sb.name())
for i in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(bytes32([0] * 32)))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(10, len, 1, nft_wallet_maker.my_nft_coins)
@ -172,8 +171,9 @@ async def test_nft_offer_sell_nft(two_wallet_nodes: Any, trusted: Any) -> None:
assert error is None
assert trade_make is not None
taker_fee = 1
success, trade_take, error = await trade_manager_taker.respond_to_offer(
Offer.from_bytes(trade_make.offer), fee=uint64(1)
Offer.from_bytes(trade_make.offer), fee=uint64(taker_fee)
)
await asyncio.sleep(1)
assert error is None
@ -181,11 +181,18 @@ async def test_nft_offer_sell_nft(two_wallet_nodes: Any, trusted: Any) -> None:
assert trade_take is not None
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(5, len, 0, nft_wallet_maker.my_nft_coins)
await time_out_assert(5, len, 1, nft_wallet_taker.my_nft_coins)
# assert payments and royalties
expected_royalty = uint64(xch_requested * royalty_basis_pts / 10000)
expected_maker_balance = funds - 2 - maker_fee + xch_requested + expected_royalty
expected_taker_balance = funds - taker_fee - xch_requested - expected_royalty
await time_out_assert(10, wallet_maker.get_confirmed_balance, expected_maker_balance)
await time_out_assert(10, wallet_taker.get_confirmed_balance, expected_taker_balance)
@pytest.mark.parametrize(
"trusted",
@ -205,7 +212,7 @@ async def test_nft_offer_request_nft(two_wallet_nodes: Any, trusted: Any) -> Non
ph_maker = await wallet_maker.get_new_puzzlehash()
ph_taker = await wallet_taker.get_new_puzzlehash()
# token_ph = bytes32(token_bytes())
ph_token = bytes32(token_bytes())
if trusted:
wallet_node_maker.config["trusted_peers"] = {
@ -224,37 +231,37 @@ async def test_nft_offer_request_nft(two_wallet_nodes: Any, trusted: Any) -> Non
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_taker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
funds = sum(
[calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks - 1)]
[calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks)]
)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_maker.get_confirmed_balance, funds)
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await time_out_assert(10, wallet_taker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_confirmed_balance, funds)
did_wallet_taker: DIDWallet = await DIDWallet.create_new_did_wallet(
wallet_node_taker.wallet_state_manager, wallet_taker, uint64(1)
)
spend_bundle_list = await wallet_node_taker.wallet_state_manager.tx_store.get_unconfirmed_for_wallet(
wallet_maker.id()
wallet_taker.id()
)
spend_bundle = spend_bundle_list[0].spend_bundle
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, spend_bundle.name())
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_taker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(15, wallet_taker.get_pending_change_balance, 0)
hex_did_id = did_wallet_taker.get_my_DID()
did_id = bytes32.fromhex(hex_did_id)
target_puzhash = ph_taker
royalty_puzhash = ph_taker
royalty_percentage = uint16(200)
royalty_basis_pts = uint16(200)
nft_wallet_taker = await NFTWallet.create_new_nft_wallet(
wallet_node_taker.wallet_state_manager, wallet_taker, name="NFT WALLET DID TAKER", did_id=did_id
@ -265,13 +272,14 @@ async def test_nft_offer_request_nft(two_wallet_nodes: Any, trusted: Any) -> Non
("h", "0xD4584AD463139FA8C0D9F68F4B59F185"),
]
)
await time_out_assert(10, wallet_taker.get_unconfirmed_balance, 1999999999999)
await time_out_assert(10, wallet_taker.get_confirmed_balance, 1999999999999)
await time_out_assert(10, wallet_taker.get_unconfirmed_balance, funds - 1)
await time_out_assert(10, wallet_taker.get_confirmed_balance, funds - 1)
sb = await nft_wallet_taker.generate_new_nft(
metadata,
target_puzhash,
royalty_puzhash,
royalty_percentage,
royalty_basis_pts,
did_id,
)
assert sb
@ -280,7 +288,7 @@ async def test_nft_offer_request_nft(two_wallet_nodes: Any, trusted: Any) -> Non
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, sb.name())
for i in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(10, len, 1, nft_wallet_taker.my_nft_coins)
@ -304,20 +312,194 @@ async def test_nft_offer_request_nft(two_wallet_nodes: Any, trusted: Any) -> Non
assert isinstance(nft_to_request_info, PuzzleInfo)
nft_to_request_asset_id = create_asset_id(nft_to_request_info)
xch_offered = 1000
offer_fee = 10
maker_fee = 10
driver_dict = {nft_to_request_asset_id: nft_to_request_info}
offer_dict = {nft_to_request_asset_id: 1, wallet_maker.id(): -xch_offered}
success, trade_make, error = await trade_manager_maker.create_offer_for_ids(offer_dict, driver_dict, fee=offer_fee)
success, trade_make, error = await trade_manager_maker.create_offer_for_ids(offer_dict, driver_dict, fee=maker_fee)
await asyncio.sleep(1)
assert success is True
assert error is None
assert trade_make is not None
taker_fee = 1
success, trade_take, error = await trade_manager_taker.respond_to_offer(
Offer.from_bytes(trade_make.offer), fee=uint64(1)
Offer.from_bytes(trade_make.offer), fee=uint64(taker_fee)
)
await asyncio.sleep(1)
assert error is None
assert success is True
assert trade_take is not None
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(5, len, 1, nft_wallet_maker.my_nft_coins)
await time_out_assert(5, len, 0, nft_wallet_taker.my_nft_coins)
# assert payments and royalties
expected_royalty = uint64(xch_offered * royalty_basis_pts / 10000)
expected_maker_balance = funds - maker_fee - xch_offered - expected_royalty
expected_taker_balance = funds - 2 - taker_fee + xch_offered + expected_royalty
await time_out_assert(10, wallet_maker.get_confirmed_balance, expected_maker_balance)
await time_out_assert(10, wallet_taker.get_confirmed_balance, expected_taker_balance)
@pytest.mark.parametrize(
"trusted",
[True],
)
@pytest.mark.asyncio
# @pytest.mark.skip
async def test_nft_offer_sell_did_to_did(two_wallet_nodes: Any, trusted: Any) -> None:
num_blocks = 2
full_nodes, wallets = two_wallet_nodes
full_node_api: FullNodeSimulator = full_nodes[0]
full_node_server = full_node_api.server
wallet_node_maker, server_0 = wallets[0]
wallet_node_taker, server_1 = wallets[1]
wallet_maker = wallet_node_maker.wallet_state_manager.main_wallet
wallet_taker = wallet_node_taker.wallet_state_manager.main_wallet
ph_maker = await wallet_maker.get_new_puzzlehash()
ph_taker = await wallet_taker.get_new_puzzlehash()
ph_token = bytes32(token_bytes())
if trusted:
wallet_node_maker.config["trusted_peers"] = {
full_node_api.full_node.server.node_id.hex(): full_node_api.full_node.server.node_id.hex()
}
wallet_node_taker.config["trusted_peers"] = {
full_node_api.full_node.server.node_id.hex(): full_node_api.full_node.server.node_id.hex()
}
else:
wallet_node_maker.config["trusted_peers"] = {}
wallet_node_taker.config["trusted_peers"] = {}
await server_0.start_client(PeerInfo("localhost", uint16(full_node_server._port)), None)
await server_1.start_client(PeerInfo("localhost", uint16(full_node_server._port)), None)
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_taker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
funds = sum(
[calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks)]
)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_maker.get_confirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_confirmed_balance, funds)
did_wallet_maker: DIDWallet = await DIDWallet.create_new_did_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, uint64(1)
)
spend_bundle_list = await wallet_node_maker.wallet_state_manager.tx_store.get_unconfirmed_for_wallet(
wallet_maker.id()
)
spend_bundle = spend_bundle_list[0].spend_bundle
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, spend_bundle.name())
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(15, wallet_maker.get_pending_change_balance, 0)
hex_did_id = did_wallet_maker.get_my_DID()
did_id = bytes32.fromhex(hex_did_id)
target_puzhash = ph_maker
royalty_puzhash = ph_maker
royalty_basis_pts = uint16(200)
nft_wallet_maker = await NFTWallet.create_new_nft_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, name="NFT WALLET DID 1", did_id=did_id
)
metadata = Program.to(
[
("u", ["https://www.chia.net/img/branding/chia-logo.svg"]),
("h", "0xD4584AD463139FA8C0D9F68F4B59F185"),
]
)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, funds - 1)
await time_out_assert(10, wallet_maker.get_confirmed_balance, funds - 1)
sb = await nft_wallet_maker.generate_new_nft(
metadata,
target_puzhash,
royalty_puzhash,
royalty_basis_pts,
did_id,
)
assert sb
# ensure hints are generated
assert compute_memos(sb)
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, sb.name())
for i in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(bytes32([0] * 32)))
await time_out_assert(10, len, 1, nft_wallet_maker.my_nft_coins)
# TAKER SETUP - WITH DID
did_wallet_taker: DIDWallet = await DIDWallet.create_new_did_wallet(
wallet_node_taker.wallet_state_manager, wallet_taker, uint64(1)
)
spend_bundle_list_taker = await wallet_node_taker.wallet_state_manager.tx_store.get_unconfirmed_for_wallet(
wallet_taker.id()
)
spend_bundle_taker = spend_bundle_list_taker[0].spend_bundle
await time_out_assert_not_none(
5, full_node_api.full_node.mempool_manager.get_spendbundle, spend_bundle_taker.name()
)
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(15, wallet_taker.get_pending_change_balance, 0)
hex_did_id_taker = did_wallet_taker.get_my_DID()
did_id_taker = bytes32.fromhex(hex_did_id_taker)
nft_wallet_taker = await NFTWallet.create_new_nft_wallet(
wallet_node_taker.wallet_state_manager, wallet_taker, name="NFT WALLET TAKER", did_id=did_id_taker
)
# maker create offer: NFT for xch
trade_manager_maker = wallet_maker.wallet_state_manager.trade_manager
trade_manager_taker = wallet_taker.wallet_state_manager.trade_manager
coins_maker = nft_wallet_maker.my_nft_coins
assert len(coins_maker) == 1
coins_taker = nft_wallet_taker.my_nft_coins
assert len(coins_taker) == 0
nft_to_offer = coins_maker[0]
nft_to_offer_info: Optional[PuzzleInfo] = match_puzzle(nft_to_offer.full_puzzle)
nft_to_offer_asset_id: bytes32 = create_asset_id(nft_to_offer_info) # type: ignore
xch_requested = 1000
maker_fee = uint64(433)
offer_did_nft_for_xch = {nft_to_offer_asset_id: -1, wallet_maker.id(): xch_requested}
success, trade_make, error = await trade_manager_maker.create_offer_for_ids(
offer_did_nft_for_xch, {}, fee=maker_fee
)
await asyncio.sleep(1)
assert success is True
assert error is None
assert trade_make is not None
taker_fee = 1
success, trade_take, error = await trade_manager_taker.respond_to_offer(
Offer.from_bytes(trade_make.offer), fee=uint64(taker_fee)
)
await asyncio.sleep(1)
assert error is None
@ -327,5 +509,402 @@ async def test_nft_offer_request_nft(two_wallet_nodes: Any, trusted: Any) -> Non
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await time_out_assert(5, len, 0, nft_wallet_maker.my_nft_coins)
# assert nnew nft wallet is created for taker
await time_out_assert(5, len, 4, wallet_taker.wallet_state_manager.wallets)
await time_out_assert(5, len, 1, wallet_taker.wallet_state_manager.wallets[4].my_nft_coins)
assert wallet_taker.wallet_state_manager.wallets[4].my_nft_coins[0].nft_id == nft_to_offer_asset_id
# assert payments and royalties
expected_royalty = uint64(xch_requested * royalty_basis_pts / 10000)
expected_maker_balance = funds - 2 - maker_fee + xch_requested + expected_royalty
expected_taker_balance = funds - 1 - taker_fee - xch_requested - expected_royalty
await time_out_assert(10, wallet_maker.get_confirmed_balance, expected_maker_balance)
await time_out_assert(10, wallet_taker.get_confirmed_balance, expected_taker_balance)
@pytest.mark.parametrize(
"trusted",
[True],
)
@pytest.mark.asyncio
# @pytest.mark.skip
async def test_nft_offer_sell_nft_for_cat(two_wallet_nodes: Any, trusted: Any) -> None:
num_blocks = 5
full_nodes, wallets = two_wallet_nodes
full_node_api: FullNodeSimulator = full_nodes[0]
full_node_server = full_node_api.server
wallet_node_maker, server_0 = wallets[0]
wallet_node_taker, server_1 = wallets[1]
wallet_maker = wallet_node_maker.wallet_state_manager.main_wallet
wallet_taker = wallet_node_taker.wallet_state_manager.main_wallet
ph_maker = await wallet_maker.get_new_puzzlehash()
ph_taker = await wallet_taker.get_new_puzzlehash()
ph_token = bytes32(token_bytes())
if trusted:
wallet_node_maker.config["trusted_peers"] = {
full_node_api.full_node.server.node_id.hex(): full_node_api.full_node.server.node_id.hex()
}
wallet_node_taker.config["trusted_peers"] = {
full_node_api.full_node.server.node_id.hex(): full_node_api.full_node.server.node_id.hex()
}
else:
wallet_node_maker.config["trusted_peers"] = {}
wallet_node_taker.config["trusted_peers"] = {}
await server_0.start_client(PeerInfo("localhost", uint16(full_node_server._port)), None)
await server_1.start_client(PeerInfo("localhost", uint16(full_node_server._port)), None)
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_taker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
funds = sum(
[calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks)]
)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_maker.get_confirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_confirmed_balance, funds)
did_wallet_maker: DIDWallet = await DIDWallet.create_new_did_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, uint64(1)
)
spend_bundle_list = await wallet_node_maker.wallet_state_manager.tx_store.get_unconfirmed_for_wallet(
wallet_maker.id()
)
spend_bundle = spend_bundle_list[0].spend_bundle
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, spend_bundle.name())
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(15, wallet_maker.get_pending_change_balance, 0)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, funds - 1)
await time_out_assert(10, wallet_maker.get_confirmed_balance, funds - 1)
hex_did_id = did_wallet_maker.get_my_DID()
did_id = bytes32.fromhex(hex_did_id)
target_puzhash = ph_maker
royalty_puzhash = ph_maker
royalty_basis_pts = uint16(200)
nft_wallet_maker = await NFTWallet.create_new_nft_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, name="NFT WALLET DID 1", did_id=did_id
)
metadata = Program.to(
[
("u", ["https://www.chia.net/img/branding/chia-logo.svg"]),
("h", "0xD4584AD463139FA8C0D9F68F4B59F185"),
]
)
sb = await nft_wallet_maker.generate_new_nft(
metadata,
target_puzhash,
royalty_puzhash,
royalty_basis_pts,
did_id,
)
assert sb
# ensure hints are generated
assert compute_memos(sb)
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, sb.name())
for i in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(10, len, 1, nft_wallet_maker.my_nft_coins)
# TAKER SETUP - NO DID
nft_wallet_taker = await NFTWallet.create_new_nft_wallet(
wallet_node_taker.wallet_state_manager, wallet_taker, name="NFT WALLET TAKER"
)
# maker create offer: NFT for xch
trade_manager_maker = wallet_maker.wallet_state_manager.trade_manager
trade_manager_taker = wallet_taker.wallet_state_manager.trade_manager
coins_maker = nft_wallet_maker.my_nft_coins
assert len(coins_maker) == 1
coins_taker = nft_wallet_taker.my_nft_coins
assert len(coins_taker) == 0
# Create new CAT and wallets for maker and taker
# Trade them between maker and taker to ensure multiple coins for each cat
cats_to_mint = 100000
cats_to_trade = uint64(10000)
async with wallet_node_maker.wallet_state_manager.lock:
cat_wallet_maker: CATWallet = await CATWallet.create_new_cat_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, {"identifier": "genesis_by_id"}, uint64(cats_to_mint)
)
await asyncio.sleep(1)
for i in range(0, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await asyncio.sleep(5)
await time_out_assert(15, cat_wallet_maker.get_confirmed_balance, cats_to_mint)
await time_out_assert(15, cat_wallet_maker.get_unconfirmed_balance, cats_to_mint)
cat_wallet_taker: CATWallet = await CATWallet.create_wallet_for_cat(
wallet_node_taker.wallet_state_manager, wallet_taker, cat_wallet_maker.get_asset_id()
)
ph_taker_cat_1 = await wallet_taker.get_new_puzzlehash()
ph_taker_cat_2 = await wallet_taker.get_new_puzzlehash()
cat_tx_records = await cat_wallet_maker.generate_signed_transaction(
[cats_to_trade, cats_to_trade], [ph_taker_cat_1, ph_taker_cat_2], memos=[[ph_taker_cat_1], [ph_taker_cat_2]]
)
for tx_record in cat_tx_records:
await wallet_maker.wallet_state_manager.add_pending_transaction(tx_record)
await time_out_assert(
15, tx_in_pool, True, full_node_api.full_node.mempool_manager, tx_record.spend_bundle.name() # type: ignore
)
for i in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
maker_cat_balance = cats_to_mint - (2 * cats_to_trade)
taker_cat_balance = 2 * cats_to_trade
await time_out_assert(15, cat_wallet_maker.get_confirmed_balance, maker_cat_balance)
await time_out_assert(15, cat_wallet_taker.get_confirmed_balance, taker_cat_balance)
nft_to_offer = coins_maker[0]
nft_to_offer_info: Optional[PuzzleInfo] = match_puzzle(nft_to_offer.full_puzzle)
nft_to_offer_asset_id: bytes32 = create_asset_id(nft_to_offer_info) # type: ignore
cats_requested = 1000
maker_fee = uint64(433)
offer_did_nft_for_xch = {nft_to_offer_asset_id: -1, cat_wallet_maker.id(): cats_requested}
success, trade_make, error = await trade_manager_maker.create_offer_for_ids(
offer_did_nft_for_xch, {}, fee=maker_fee
)
await asyncio.sleep(1)
assert success is True
assert error is None
assert trade_make is not None
taker_fee = 1
success, trade_take, error = await trade_manager_taker.respond_to_offer(
Offer.from_bytes(trade_make.offer), fee=uint64(taker_fee)
)
await asyncio.sleep(1)
assert error is None
assert success is True
assert trade_take is not None
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(5, len, 0, nft_wallet_maker.my_nft_coins)
await time_out_assert(5, len, 1, nft_wallet_taker.my_nft_coins)
# assert payments and royalties
expected_royalty = uint64(cats_requested * royalty_basis_pts / 10000)
expected_maker_balance = funds - 2 - cats_to_mint - maker_fee
expected_taker_balance = funds - taker_fee
expected_maker_cat_balance = maker_cat_balance + cats_requested + expected_royalty
expected_taker_cat_balance = taker_cat_balance - cats_requested - expected_royalty
await time_out_assert(10, wallet_maker.get_confirmed_balance, expected_maker_balance)
await time_out_assert(10, wallet_taker.get_confirmed_balance, expected_taker_balance)
await time_out_assert(10, cat_wallet_maker.get_confirmed_balance, expected_maker_cat_balance)
await time_out_assert(10, cat_wallet_taker.get_confirmed_balance, expected_taker_cat_balance)
@pytest.mark.parametrize(
"trusted",
[True],
)
@pytest.mark.asyncio
# @pytest.mark.skip
async def test_nft_offer_request_nft_for_cat(two_wallet_nodes: Any, trusted: Any) -> None:
num_blocks = 5
full_nodes, wallets = two_wallet_nodes
full_node_api: FullNodeSimulator = full_nodes[0]
full_node_server = full_node_api.server
wallet_node_maker, server_0 = wallets[0]
wallet_node_taker, server_1 = wallets[1]
wallet_maker = wallet_node_maker.wallet_state_manager.main_wallet
wallet_taker = wallet_node_taker.wallet_state_manager.main_wallet
ph_maker = await wallet_maker.get_new_puzzlehash()
ph_taker = await wallet_taker.get_new_puzzlehash()
ph_token = bytes32(token_bytes())
if trusted:
wallet_node_maker.config["trusted_peers"] = {
full_node_api.full_node.server.node_id.hex(): full_node_api.full_node.server.node_id.hex()
}
wallet_node_taker.config["trusted_peers"] = {
full_node_api.full_node.server.node_id.hex(): full_node_api.full_node.server.node_id.hex()
}
else:
wallet_node_maker.config["trusted_peers"] = {}
wallet_node_taker.config["trusted_peers"] = {}
await server_0.start_client(PeerInfo("localhost", uint16(full_node_server._port)), None)
await server_1.start_client(PeerInfo("localhost", uint16(full_node_server._port)), None)
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_maker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_taker))
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
funds = sum(
[calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks)]
)
await time_out_assert(10, wallet_maker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_maker.get_confirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_unconfirmed_balance, funds)
await time_out_assert(10, wallet_taker.get_confirmed_balance, funds)
did_wallet_taker: DIDWallet = await DIDWallet.create_new_did_wallet(
wallet_node_taker.wallet_state_manager, wallet_taker, uint64(1)
)
spend_bundle_list = await wallet_node_taker.wallet_state_manager.tx_store.get_unconfirmed_for_wallet(
wallet_taker.id()
)
spend_bundle = spend_bundle_list[0].spend_bundle
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, spend_bundle.name())
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(15, wallet_taker.get_pending_change_balance, 0)
await time_out_assert(10, wallet_taker.get_unconfirmed_balance, funds - 1)
await time_out_assert(10, wallet_taker.get_confirmed_balance, funds - 1)
hex_did_id = did_wallet_taker.get_my_DID()
did_id = bytes32.fromhex(hex_did_id)
target_puzhash = ph_taker
royalty_puzhash = ph_taker
royalty_basis_pts = uint16(200)
nft_wallet_taker = await NFTWallet.create_new_nft_wallet(
wallet_node_taker.wallet_state_manager, wallet_taker, name="NFT WALLET DID TAKER", did_id=did_id
)
metadata = Program.to(
[
("u", ["https://www.chia.net/img/branding/chia-logo.svg"]),
("h", "0xD4584AD463139FA8C0D9F68F4B59F185"),
]
)
sb = await nft_wallet_taker.generate_new_nft(
metadata,
target_puzhash,
royalty_puzhash,
royalty_basis_pts,
did_id,
)
assert sb
# ensure hints are generated
assert compute_memos(sb)
await time_out_assert_not_none(5, full_node_api.full_node.mempool_manager.get_spendbundle, sb.name())
for i in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(10, len, 1, nft_wallet_taker.my_nft_coins)
# MAKER SETUP - NO DID
nft_wallet_maker = await NFTWallet.create_new_nft_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, name="NFT WALLET MAKER"
)
# maker create offer: NFT for xch
trade_manager_maker = wallet_maker.wallet_state_manager.trade_manager
trade_manager_taker = wallet_taker.wallet_state_manager.trade_manager
coins_maker = nft_wallet_maker.my_nft_coins
assert len(coins_maker) == 0
coins_taker = nft_wallet_taker.my_nft_coins
assert len(coins_taker) == 1
# Create new CAT and wallets for maker and taker
# Trade them between maker and taker to ensure multiple coins for each cat
cats_to_mint = 100000
cats_to_trade = uint64(10000)
async with wallet_node_maker.wallet_state_manager.lock:
cat_wallet_maker: CATWallet = await CATWallet.create_new_cat_wallet(
wallet_node_maker.wallet_state_manager, wallet_maker, {"identifier": "genesis_by_id"}, uint64(cats_to_mint)
)
await asyncio.sleep(1)
for i in range(0, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await asyncio.sleep(5)
await time_out_assert(15, cat_wallet_maker.get_confirmed_balance, cats_to_mint)
await time_out_assert(15, cat_wallet_maker.get_unconfirmed_balance, cats_to_mint)
cat_wallet_taker: CATWallet = await CATWallet.create_wallet_for_cat(
wallet_node_taker.wallet_state_manager, wallet_taker, cat_wallet_maker.get_asset_id()
)
ph_taker_cat_1 = await wallet_taker.get_new_puzzlehash()
ph_taker_cat_2 = await wallet_taker.get_new_puzzlehash()
cat_tx_records = await cat_wallet_maker.generate_signed_transaction(
[cats_to_trade, cats_to_trade], [ph_taker_cat_1, ph_taker_cat_2], memos=[[ph_taker_cat_1], [ph_taker_cat_2]]
)
for tx_record in cat_tx_records:
await wallet_maker.wallet_state_manager.add_pending_transaction(tx_record)
await time_out_assert(
15, tx_in_pool, True, full_node_api.full_node.mempool_manager, tx_record.spend_bundle.name() # type: ignore
)
for i in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
maker_cat_balance = cats_to_mint - (2 * cats_to_trade)
taker_cat_balance = 2 * cats_to_trade
await time_out_assert(15, cat_wallet_maker.get_confirmed_balance, maker_cat_balance)
await time_out_assert(15, cat_wallet_taker.get_confirmed_balance, taker_cat_balance)
nft_to_request = coins_taker[0]
nft_to_request_info: Optional[PuzzleInfo] = match_puzzle(nft_to_request.full_puzzle)
nft_to_request_asset_id: bytes32 = create_asset_id(nft_to_request_info) # type: ignore
cats_requested = 1000
maker_fee = uint64(433)
driver_dict = {nft_to_request_asset_id: nft_to_request_info}
offer_dict = {nft_to_request_asset_id: 1, cat_wallet_maker.id(): -cats_requested}
success, trade_make, error = await trade_manager_maker.create_offer_for_ids(offer_dict, driver_dict, fee=maker_fee)
await asyncio.sleep(1)
assert success is True
assert error is None
assert trade_make is not None
taker_fee = 1
success, trade_take, error = await trade_manager_taker.respond_to_offer(
Offer.from_bytes(trade_make.offer), fee=uint64(taker_fee)
)
await asyncio.sleep(1)
assert error is None
assert success is True
assert trade_take is not None
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph_token))
await time_out_assert(5, len, 1, nft_wallet_maker.my_nft_coins)
await time_out_assert(5, len, 0, nft_wallet_taker.my_nft_coins)
# assert payments and royalties
expected_royalty = uint64(cats_requested * royalty_basis_pts / 10000)
expected_maker_balance = funds - cats_to_mint - maker_fee
expected_taker_balance = funds - 2 - taker_fee
expected_maker_cat_balance = maker_cat_balance - cats_requested - expected_royalty
expected_taker_cat_balance = taker_cat_balance + cats_requested + expected_royalty
await time_out_assert(10, wallet_maker.get_confirmed_balance, expected_maker_balance)
await time_out_assert(10, wallet_taker.get_confirmed_balance, expected_taker_balance)
await time_out_assert(10, cat_wallet_maker.get_confirmed_balance, expected_maker_cat_balance)
await time_out_assert(10, cat_wallet_taker.get_confirmed_balance, expected_taker_cat_balance)