chia-blockchain/tests/wallet/test_singleton_lifecycle_fa...

188 lines
8.0 KiB
Python

from typing import List, Tuple
from blspy import G2Element
from clvm_tools import binutils
from chia.types.blockchain_format.program import Program, INFINITE_COST
from chia.types.announcement import Announcement
from chia.types.blockchain_format.coin import Coin
from chia.types.blockchain_format.sized_bytes import bytes32
from chia.types.coin_solution import CoinSolution
from chia.types.spend_bundle import SpendBundle
from chia.util.condition_tools import ConditionOpcode
from chia.wallet.cc_wallet.debug_spend_bundle import debug_spend_bundle
from chia.wallet.puzzles.load_clvm import load_clvm
from tests.clvm.coin_store import CoinStore, CoinTimestamp
SINGLETON_MOD = load_clvm("singleton_top_layer.clvm")
LAUNCHER_PUZZLE = load_clvm("singleton_launcher.clvm")
P2_SINGLETON_MOD = load_clvm("p2_singleton.clvm")
POOL_MEMBER_MOD = load_clvm("pool_member_innerpuz.clvm")
POOL_ESCAPING_MOD = load_clvm("pool_escaping_innerpuz.clvm")
LAUNCHER_PUZZLE_HASH = LAUNCHER_PUZZLE.get_tree_hash()
SINGLETON_MOD_HASH = SINGLETON_MOD.get_tree_hash()
POOL_REWARD_PREFIX_MAINNET = bytes32.fromhex("ccd5bb71183532bff220ba46c268991a00000000000000000000000000000000")
def check_coin_solution(coin_solution: CoinSolution):
breakpoint()
try:
cost, result = coin_solution.puzzle_reveal.run_with_cost(INFINITE_COST, coin_solution.solution)
except Exception as ex:
print(ex)
breakpoint()
print(ex)
def adaptor_for_singleton_inner_puzzle(puzzle: Program) -> Program:
# this is prety slow
return Program.to(binutils.assemble("(a (q . %s) 3)" % binutils.disassemble(puzzle)))
def launcher_conditions_and_spend_bundle(
parent_coin_id: bytes32,
launcher_amount: int,
initial_singleton_inner_puzzle: Program,
metadata: List[Tuple[str, str]],
launcher_puzzle: Program = LAUNCHER_PUZZLE,
) -> Tuple[bytes32, List[Program], SpendBundle]:
launcher_puzzle_hash = launcher_puzzle.get_tree_hash()
launcher_coin = Coin(parent_coin_id, launcher_puzzle_hash, launcher_amount)
singleton_full_puzzle = SINGLETON_MOD.curry(
SINGLETON_MOD_HASH, launcher_coin.name(), launcher_puzzle_hash, initial_singleton_inner_puzzle
)
singleton_full_puzzle_hash = singleton_full_puzzle.get_tree_hash()
message_program = Program.to([singleton_full_puzzle_hash, launcher_amount, metadata])
expected_announcement = Announcement(launcher_coin.name(), message_program.get_tree_hash())
expected_conditions = []
expected_conditions.append(
Program.to(
binutils.assemble(f"(0x{ConditionOpcode.ASSERT_COIN_ANNOUNCEMENT.hex()} 0x{expected_announcement.name()})")
)
)
expected_conditions.append(
Program.to(
binutils.assemble(f"(0x{ConditionOpcode.CREATE_COIN.hex()} 0x{launcher_puzzle_hash} {launcher_amount})")
)
)
launcher_solution = Program.to([singleton_full_puzzle_hash, launcher_amount, metadata])
coin_solution = CoinSolution(launcher_coin, launcher_puzzle, launcher_solution)
spend_bundle = SpendBundle([coin_solution], G2Element())
lineage_proof = Program.to([parent_coin_id, launcher_amount])
return lineage_proof, launcher_coin.name(), expected_conditions, spend_bundle
def singleton_puzzle(launcher_id: Program, launcher_puzzle_hash: bytes32, inner_puzzle: Program) -> Program:
return SINGLETON_MOD.curry(SINGLETON_MOD_HASH, launcher_id, launcher_puzzle_hash, inner_puzzle)
def singleton_puzzle_hash(launcher_id: Program, launcher_puzzle_hash: bytes32, inner_puzzle: Program) -> bytes32:
return singleton_puzzle(launcher_id, launcher_puzzle_hash, inner_puzzle).get_tree_hash()
def solution_for_singleton_puzzle(lineage_proof: Program, my_amount: int, inner_solution: Program) -> Program:
return Program.to([lineage_proof, my_amount, inner_solution])
def p2_singleton_puzzle(launcher_id: Program, launcher_puzzle_hash: bytes32) -> Program:
return P2_SINGLETON_MOD.curry(SINGLETON_MOD_HASH, launcher_id, launcher_puzzle_hash)
def p2_singleton_puzzle_hash(launcher_id: Program, launcher_puzzle_hash: bytes32) -> bytes32:
return p2_singleton_puzzle(launcher_id, launcher_puzzle_hash).get_tree_hash()
def test_lifecycle_with_coinstore():
metadata = [("foo", "bar")]
ANYONE_CAN_SPEND_PUZZLE = Program.to(1)
coin_store = CoinStore(int.from_bytes(POOL_REWARD_PREFIX_MAINNET, "big"))
now = CoinTimestamp(10012300, 1)
parent_coin_amount = 100000
farmed_coin = coin_store.farm_coin(ANYONE_CAN_SPEND_PUZZLE.get_tree_hash(), now, amount=parent_coin_amount)
now.seconds += 500
now.height += 1
launcher_amount = 1
launcher_puzzle = LAUNCHER_PUZZLE
launcher_puzzle_hash = launcher_puzzle.get_tree_hash()
initial_singleton_puzzle = adaptor_for_singleton_inner_puzzle(ANYONE_CAN_SPEND_PUZZLE)
lineage_proof, launcher_id, condition_list, launcher_spend_bundle = launcher_conditions_and_spend_bundle(
farmed_coin.name(), launcher_amount, initial_singleton_puzzle, metadata, launcher_puzzle
)
conditions = Program.to(condition_list)
coin_solution = CoinSolution(farmed_coin, ANYONE_CAN_SPEND_PUZZLE, conditions)
spend_bundle = SpendBundle.aggregate([launcher_spend_bundle, SpendBundle([coin_solution], G2Element())])
debug_spend_bundle(spend_bundle)
coin_store.update_coin_store_for_spend_bundle(spend_bundle, now)
launcher_coin = launcher_spend_bundle.coin_solutions[0].coin
assert coin_store.coin_record(launcher_coin.name()).spent
assert coin_store.coin_record(farmed_coin.name()).spent
singleton_expected_puzzle = singleton_puzzle(launcher_id, launcher_puzzle_hash, initial_singleton_puzzle)
singleton_expected_puzzle_hash = singleton_expected_puzzle.get_tree_hash()
expected_singleton_coin = Coin(launcher_coin.name(), singleton_expected_puzzle_hash, launcher_amount)
assert coin_store.coin_record(expected_singleton_coin.name()).spent is False
# farm a `p2_singleton`
pool_reward_puzzle_hash = p2_singleton_puzzle_hash(launcher_id, launcher_puzzle_hash)
p2_singleton_coin = coin_store.farm_coin(pool_reward_puzzle_hash, now)
assert p2_singleton_coin.puzzle_hash == pool_reward_puzzle_hash
# now collect the `p2_singleton`
# build `CoinSolution` for the `p2_singleton`
singleton_inner_puzzle_hash = initial_singleton_puzzle.get_tree_hash()
p2_singleton_solution = Program.to([singleton_inner_puzzle_hash, p2_singleton_coin.name()])
p2_singleton_coin_solution = CoinSolution(
p2_singleton_coin, p2_singleton_puzzle(launcher_id, launcher_puzzle_hash), p2_singleton_solution
)
# build `CoinSolution` for the singleton
expected_p2_singleton_announcement = Announcement(p2_singleton_coin.name(), bytes([0x80])).name()
conditions = [
Program.to([ConditionOpcode.CREATE_PUZZLE_ANNOUNCEMENT, p2_singleton_coin.name()]),
Program.to([ConditionOpcode.CREATE_COIN, singleton_inner_puzzle_hash, 1]),
Program.to([ConditionOpcode.ASSERT_COIN_ANNOUNCEMENT, expected_p2_singleton_announcement]),
]
singleton_inner_solution = conditions
singleton_solution = Program.to([lineage_proof, expected_singleton_coin.amount, singleton_inner_solution])
singleton_coin_solution = CoinSolution(expected_singleton_coin, singleton_expected_puzzle, singleton_solution)
spend_bundle = SpendBundle([p2_singleton_coin_solution, singleton_coin_solution], G2Element())
debug_spend_bundle(spend_bundle)
coin_store.update_coin_store_for_spend_bundle(spend_bundle, now)
# spend_bundle = claim_p2_singleton(p2_singleton_coin, singleton_inner_puzzle_hash, my_id)
# next up: spend the expected_singleton_coin
# it's an adapted `ANYONE_CAN_SPEND_PUZZLE`
# then try a bad lineage proof
# then try writing two odd coins
# then try writing zero odd coins
# then do a `p2_singleton` and collect it
# then commit to a pool
# then do a `p2_singleton` and collect it
# then escape the pool
# then do a `p2_singleton` and collect it
# then finish leaving the pool
# then, destroy the singleton with the -113 hack
return 0