chia-blockchain/tests/test_spendbundle.py

39 lines
1.4 KiB
Python

from src.util.Hash import std_hash
from src.util.condition_tools import conditions_by_opcode, aggsig_in_conditions_dict, created_outputs_for_conditions_dict, \
conditions_for_solution
from src.wallet.puzzles import p2_delegated_puzzle
from src.wallet.puzzles.puzzle_utils import make_create_coin_condition
from tests.keys import puzzle_program_for_index
def test_1():
puzzle_program_0 = puzzle_program_for_index(0)
puzzle_program_1 = puzzle_program_for_index(1)
puzzle_program_2 = puzzle_program_for_index(2)
conditions = [
make_create_coin_condition(std_hash(bytes(pp)), amount)
for pp, amount in [(puzzle_program_1, 1000), (puzzle_program_2, 2000),]
]
assert conditions is not None
puzzle_hash_solution = p2_delegated_puzzle.solution_for_conditions(
puzzle_program_0, conditions
)
error, output_conditions = conditions_for_solution(puzzle_hash_solution)
assert error is None
from pprint import pprint
assert output_conditions is not None
output_conditions_dict = conditions_by_opcode(output_conditions)
pprint(output_conditions_dict)
input_coin_info_hash = bytes([0] * 32)
additions = created_outputs_for_conditions_dict(
output_conditions_dict, input_coin_info_hash
)
aggsigs = aggsig_in_conditions_dict(output_conditions_dict)
pprint(aggsigs)
assert len(additions) == 2
assert len(aggsigs) == 1