def test_close_budgets(wallet_3hf: Wallet, opened_budgets_same_acc):
budgets = opened_budgets_same_acc # just renaming
validate_response(
wallet_3hf.close_budget(budgets[0]["uuid"], budgets[0]["owner"], budgets[0]["type"]),
wallet_3hf.close_budget.__name__
)
rest_budgets = wallet_3hf.get_budgets([budgets[-1]["owner"]], budgets[-1]["type"])
assert len(rest_budgets) == len(budgets) - 1
assert all(rb["id"] != budgets[0]["id"] for rb in rest_budgets)
# delete already deleted
validate_error_response(
wallet_3hf.close_budget(budgets[0]['uuid'], budgets[0]["owner"], budgets[0]["type"]),
wallet_3hf.close_budget.__name__,
RE_BUDGET_NOT_EXIST
)
def test_create_max_budgets(wallet_3hf: Wallet, budget):
re_budgets_limit = r"Can't create more then .* budgets per owner."
limit = wallet_3hf.get_config()["SCORUM_BUDGETS_LIMIT_PER_OWNER"]
balance = "0.000000001 SCR"
update_budget_time(wallet_3hf, budget, start=5, deadline=300)
budgets = []
for i in range(1, limit + 1):
budget_cp = copy(budget)
budget_cp.update({'uuid': gen_uid(), 'balance': balance})
budgets.append(budget_cp)
validate_response(
wallet_3hf.broadcast_multiple_ops('create_budget_operation', budgets, {budget['owner']}),
'create_budget_operation'
)
update_budget_time(wallet_3hf, budget, start=5, deadline=300)
validate_error_response(
wallet_3hf.create_budget(**budget),
wallet_3hf.create_budget.__name__,
re_budgets_limit
)
wallet_3hf.close_budget(str(budgets[0]['uuid']), budget['owner'], budget['type'])
update_budget_time(wallet_3hf, budget, start=5, deadline=300)
validate_response(wallet_3hf.create_budget(**budget), wallet_3hf.create_budget.__name__)
def test_close_before_starttime(wallet_3hf: Wallet, budget):
update_budget_time(wallet_3hf, budget, start=30, deadline=60) # to delay opening time for budget
balance_before = wallet_3hf.get_account_scr_balance(budget["owner"])
wallet_3hf.create_budget(**budget)
update_budget_balance(wallet_3hf, budget) # update budget params / set budget id
response = wallet_3hf.close_budget(budget['uuid'], budget["owner"], budget["type"])
validate_response(response, wallet_3hf.close_budget.__name__)
balance_after = wallet_3hf.get_account_scr_balance(budget["owner"])
assert balance_after == balance_before
check_virt_ops(
wallet_3hf, response['block_num'], response['block_num'],
{'close_budget', 'budget_closing', 'budget_owner_income'}
)
assert len(wallet_3hf.get_budgets([budget['owner']], budget['type'])) == 0
assert len(wallet_3hf.list_buddget_owners(budget_type=budget['type'])) == 0
def test_close_after_starttime(wallet_3hf: Wallet, budget):
update_budget_time(wallet_3hf, budget)
balance_before = wallet_3hf.get_account_scr_balance(budget["owner"])
response = wallet_3hf.create_budget(**budget)
create_block = response["block_num"]
update_budget_balance(wallet_3hf, budget) # update budget params / set budget id
per_block = Amount(budget["per_block"])
response = wallet_3hf.close_budget(budget['uuid'], budget["owner"], budget["type"])
validate_response(response, wallet_3hf.close_budget.__name__)
close_block = response["block_num"]
balance_after = wallet_3hf.get_account_scr_balance(budget["owner"])
assert balance_before == balance_after + per_block * (close_block - create_block)
check_virt_ops(
wallet_3hf, close_block, close_block,
{'close_budget', 'budget_closing', 'budget_outgo', 'budget_owner_income'}
)
assert len(wallet_3hf.get_budgets([budget['owner']], budget['type'])) == 0
assert len(wallet_3hf.list_buddget_owners(budget_type=budget['type'])) == 0
def test_close_post_vs_banner(wallet_3hf: Wallet, post_budget, banner_budget):
new_budget = copy(post_budget)
update_budget_time(wallet_3hf, post_budget)
wallet_3hf.create_budget(**post_budget)
update_budget_balance(wallet_3hf, post_budget) # update budget params / set budget id
update_budget_time(wallet_3hf, banner_budget)
wallet_3hf.create_budget(**banner_budget)
update_budget_balance(wallet_3hf, banner_budget) # update budget params / set budget id
assert post_budget["id"] == banner_budget["id"] # both = 0
response = wallet_3hf.close_budget(post_budget['uuid'], post_budget["owner"], post_budget["type"])
validate_response(response, wallet_3hf.close_budget.__name__)
post_budgets = wallet_3hf.get_budgets([post_budget['owner']], post_budget['type'])
assert len(post_budgets) == 0
banner_budgets = wallet_3hf.get_budgets([banner_budget['owner']], banner_budget['type'])
assert len(banner_budgets) == 1
update_budget_time(wallet_3hf, new_budget)
validate_response(wallet_3hf.create_budget(**new_budget), wallet_3hf.create_budget.__name__)
update_budget_balance(wallet_3hf, new_budget) # update budget params / set budget id
assert new_budget["id"] > banner_budget["id"], "Newly created budget should have incremented index"
def test_close_budget_locked(wallet: Wallet, budget):
response = wallet.close_budget(gen_uid(), DEFAULT_WITNESS, budget["type"])
validate_error_response(response, wallet.close_budget.__name__,
RE_OP_IS_LOCKED)
def test_unknown_uuid(wallet_3hf: Wallet, opened_budgets, uuid):
validate_error_response(
wallet_3hf.close_budget(uuid(), opened_budgets[0]["owner"], opened_budgets[0]["type"]),
wallet_3hf.close_budget.__name__,
RE_BUDGET_NOT_EXIST
)
