def sequence_of_transfers(context):
expected_transfers = {(wallet_from_name(context, row['source']),
wallet_from_name(context, row['destination']),
Decimal(row['amount']))
for row in context.table}
actual_transfers = {(transfer.source, transfer.destination,
transfer.amount)
for transfer in context.leveller.proposed_transfers()}
assert_that(actual_transfers, equal_to(expected_transfers))
def proposed_transfers(context, transfer_amount, source, destination):
source_wallet = wallet_from_name(context, source)
destination_wallet = wallet_from_name(context, destination)
transfers = list(context.leveller.proposed_transfers())
assert_that(len(transfers), equal_to(1), 'One transfer expected')
transfer = transfers[0]
assert_that(transfer_amount, equal_to(transfer.amount))
assert_that(source_wallet, equal_to(transfer.source))
assert_that(destination_wallet, equal_to(transfer.destination))
def connection_list(context):
context.connections_correct = True
for row in context.table:
source_wallet = wallet_from_name(context, row['source'])
destination_wallet = wallet_from_name(context, row['destination'])
weight = int(row['weight']) if row['weight'] else 1
try:
context.leveller.connect(source_wallet, destination_wallet, weight)
except DuplicateConnection:
context.connections_correct = False
context.connections_correct = context.connections_correct and context.leveller.is_valid
def connection_list(context):
context.connections_correct = True
for row in context.table:
source_wallet = wallet_from_name(context, row['source'])
destination_wallet = wallet_from_name(context, row['destination'])
weight = int(row['weight']) if row['weight'] else 1
try:
context.leveller.connect(source_wallet, destination_wallet, weight)
except DuplicateConnection:
context.connections_correct = False
context.connections_correct = context.connections_correct and context.leveller.is_valid
def introduce_funds(context, modified_wallet, payment_amount):
wallet = wallet_from_name(context, modified_wallet)
wallet.move_to_account(
DEFAULT_ACCOUNT, context.account_names[modified_wallet],
payment_amount,
'Payment to/from test wallet'
)
def add_wallet(context, wallet_name, desired_level, low_water_mark, high_water_mark):
wallet = wallet_from_name(context, wallet_name)
try:
context.leveller.add_wallet(wallet, desired_level, low_water_mark, high_water_mark)
except ConflictingWaterMarks:
context.watermark_conflict = True
else:
context.watermark_conflict = False
def sequence_of_transfers(context):
expected_transfers = {
(
wallet_from_name(context, row['source']),
wallet_from_name(context, row['destination']),
Decimal(row['amount'])
)
for row in context.table
}
actual_transfers = {
(
transfer.source,
transfer.destination,
transfer.amount
)
for transfer in context.leveller.proposed_transfers()
}
assert_that(actual_transfers, equal_to(expected_transfers))
def check_balance_with_fee(context, wallet_name, expected_balance):
wallet = wallet_from_name(context, wallet_name)
assert_that(wallet.balance, equal_to(expected_balance + context.fee))
def move_funds(context, transfer_amount, source, destination):
source_wallet = wallet_from_name(context, source)
destination_wallet = wallet_from_name(context, destination)
context.transaction_id, context.fee = source_wallet.transfer_to(
destination_wallet, transfer_amount)
def check_balance(context, wallet_name, adjusted_balance):
wallet = wallet_from_name(context, wallet_name)
assert_that(context.leveller.adjusted_balance(wallet), equal_to(adjusted_balance))
def add_wallet_no_hwm(context, desired_level, wallet_name, low_water_mark):
wallet = wallet_from_name(context, wallet_name)
context.leveller.add_wallet(wallet, desired_level, low_water_mark, None)
def check_balance_with_fee(context, wallet_name, expected_balance):
wallet = wallet_from_name(context, wallet_name)
assert_that(wallet.balance, equal_to(expected_balance + context.fee))
def connect_wallets(context, source, destination, link_weight):
source_wallet = wallet_from_name(context, source)
destination_wallet = wallet_from_name(context, destination)
context.leveller.connect(source_wallet, destination_wallet, link_weight)
def check_balance(context, wallet_name, adjusted_balance):
wallet = wallet_from_name(context, wallet_name)
assert_that(context.leveller.adjusted_balance(wallet),
equal_to(adjusted_balance))
def move_funds(context, transfer_amount, source, destination):
source_wallet = wallet_from_name(context, source)
destination_wallet = wallet_from_name(context, destination)
context.transaction_id, context.fee = source_wallet.transfer_to(destination_wallet, transfer_amount)
def add_wallet_no_lwm(context, desired_level, wallet_name, high_water_mark):
wallet = wallet_from_name(context, wallet_name)
context.leveller.add_wallet(wallet, desired_level, None, high_water_mark)
def introduce_funds(context, modified_wallet, payment_amount):
wallet = wallet_from_name(context, modified_wallet)
wallet.move_to_account(DEFAULT_ACCOUNT,
context.account_names[modified_wallet],
payment_amount, 'Payment to/from test wallet')
def leveller_moves_funds(context, transfer_amount, source, destination):
# TODO which agent should be responsible for executing the transfers?
source_wallet = wallet_from_name(context, source)
destination_wallet = wallet_from_name(context, destination)
context.transaction_id, context.fee = source_wallet.transfer_to(destination_wallet, transfer_amount)
def required_funds(context, wallet_name, additional_amount):
wallet = wallet_from_name(context, wallet_name)
required = context.leveller.required_funds(wallet)
assert_that(required, equal_to(additional_amount))
def surplus_funds(context, wallet_name, surplus_amount):
wallet = wallet_from_name(context, wallet_name)
surplus = context.leveller.surplus_funds(wallet)
assert_that(surplus, equal_to(surplus_amount))