def SetUp(self):
"""
Запускается в начале каждого теста,
создает пользователя для тестирования
"""
self.name, self.total_money = 'test Wallet', 1000000
self.wallet = Wallet(self.name, self.total_money)
def test_move_money_from_wal_to_wal(self):
"""
Проверяет функцию пополнения баланса
"""
self.SetUp()
test_wallet = Wallet("wallet 2", 400)
self.wallet.move_money(300000, test_wallet)
assert self.wallet.balance == 700000
assert test_wallet.balance == 300400
class TestWalletCase(object):
def SetUp(self):
"""
Запускается в начале каждого теста,
создает пользователя для тестирования
"""
self.name, self.total_money = 'test Wallet', 1000000
self.wallet = Wallet(self.name, self.total_money)
def test_add_money(self):
"""
Проверяет функцию перемещения денег
"""
self.SetUp()
self.wallet.add_money(666)
assert self.wallet.balance == 1000666
def test_move_money_from_wal_to_wal(self):
"""
Проверяет функцию пополнения баланса
"""
self.SetUp()
test_wallet = Wallet("wallet 2", 400)
self.wallet.move_money(300000, test_wallet)
assert self.wallet.balance == 700000
assert test_wallet.balance == 300400
def test_move_money_from_wal_to_cat(self):
"""
Проверяет функцию пополнения баланса
"""
self.SetUp()
test_cat = Category("category", 400)
self.wallet.move_money(300000, test_cat)
assert self.wallet.balance == 700000
assert test_cat.balance == 300400
def run_all_test(self):
"""
Запускает все тесты
"""
self.test_add_money()
self.test_move_money_from_wal_to_wal()
self.test_move_money_from_wal_to_cat()
def test_move_money_from_cat_to_wallet(self):
self.SetUp()
test_wallet = Wallet("test Wallet", 500)
self.category.move_money(300000, test_wallet)
assert self.category.balance == 700000
assert test_wallet.balance == 300500
class Node:
"""
every node is a computer having its local blockchain instance,
where you can mine block and send transactions
therefore, it make sense that we have a node class
"""
def __init__(self):
# self.id = str(uuid4())
self.wallet = Wallet() # right now, wallet object with None keys
self.wallet.create_keys() # right now, wallet object has key pair
# pass the key to blockchain to create blockchain object
self.blockchain = Blockchain(self.wallet.public_key)
def get_transaction_value(self):
tx_recipient = input('Enter the recipient of the transaction: ')
tx_amount = float(input("Your transaction amount please: "))
return tx_recipient, tx_amount
def get_user_choice(self):
user_input = input('Your choice: ')
return user_input
def print_blockchain_elements(self):
# output the blockchain list to the console
for block in self.blockchain.chain: # this execute getter
print('Outputting Block')
print(block)
else:
print('-' * 20)
def listen_for_input(self):
waiting_for_input = True
while waiting_for_input:
print("Please choose")
print("1: Add a new transaction value")
print("2: Mine a new block")
print("3: Output the blockchain blocks")
print("4: Check transaction validity")
print("5: Create wallet")
print("6: Load wallet")
print("7: Save keys")
print("q: Quit")
user_choice = self.get_user_choice()
if user_choice == '1':
tx_data = self.get_transaction_value() # from user input
recipient, amount = tx_data
signature = self.wallet.sign_transaction(
sender=self.wallet.public_key,
recipient=recipient,
amount=amount)
# signature should be pass if you want to add transaction
if self.blockchain.add_transaction(
recipient=recipient,
sender=self.wallet.public_key,
signature=signature,
amount=amount):
print('Added transaction!')
else:
print('Transaction failed!')
print(self.blockchain.get_open_transactions())
elif user_choice == '2':
if not self.blockchain.mine_block():
print("Mining failed. Got no wallet?")
elif user_choice == '3':
self.print_blockchain_elements()
elif user_choice == '4':
if Verification.verify_transactions(
open_transactions=self.blockchain.
get_open_transactions(),
get_balance=self.blockchain.get_balance):
print('All transactions are valid')
else:
print('There are invalid transactions')
elif user_choice == '5':
self.wallet.create_keys(
) # right now, wallet object has key pair
# pass the key to blockchain to create blockchain object
self.blockchain = Blockchain(self.wallet.public_key)
elif user_choice == '6':
self.wallet.load_keys()
# after loading key, create blockchain object
self.blockchain = Blockchain(self.wallet.public_key)
elif user_choice == '7':
self.wallet.save_keys()
elif user_choice == 'q':
waiting_for_input = False
else:
print('Input was invalid, please pick a value from the list!')
if not Verification.verify_chain(blockchain=self.blockchain.chain):
self.print_blockchain_elements()
print("Invalid Blockchain!")
break
# :6 = 6 digits for it, .2f = only show two decimal places
print('Balance of {}: {:6.2f}'.format(
self.wallet.public_key, self.blockchain.get_balance()))
else:
print('User left!')
print('Done!')
def __init__(self):
# self.id = str(uuid4())
self.wallet = Wallet() # right now, wallet object with None keys
self.wallet.create_keys() # right now, wallet object has key pair
# pass the key to blockchain to create blockchain object
self.blockchain = Blockchain(self.wallet.public_key)