def show(self):
"""Start the wizard."""
clear_screen()
# convert dict to an ordered list
# this needs to be done to get an ordered list that does not change
# at runtime of the function
wallet_list = self.__wallet_to_list()
# check if wallet contains any keys
# case: wallet not empty
if not len(self.wallet) == 0:
chosen_key = self.__ask_for_key_from_wallet(wallet_list)
if chosen_key == '':
return
# ask for valid sender input in a loop
while not self.__validate_sender_input(chosen_key):
chosen_key = self.__ask_for_key_from_wallet(wallet_list)
if chosen_key == '':
return
clear_screen()
print('Invalid input! Please choose a correct index!')
print()
clear_screen()
print(u'Sender: ' + str(chosen_key))
chosen_receiver = self.__ask_for_receiver()
while not self.__validate_receiver_input(chosen_receiver):
# clear_screen()
print('Invalid input! Please choose a correct receiver!')
print(u'Sender: ' + str(chosen_key))
chosen_receiver = self.__ask_for_receiver()
print()
clear_screen()
print(u'Sender: ' + str(chosen_key))
print(u'Receiver: ' + str(chosen_receiver))
chosen_payload = self.__ask_for_payload()
while not self.__validate_payload_input(chosen_payload):
# clear_screen()
print('Invalid input! Please choose a correct payload!')
print(u'Sender: ' + str(chosen_key))
print(u'Receiver: ' + str(chosen_receiver))
chosen_payload = self.__ask_for_payload()
print()
clear_screen()
# Create transaction Object and send to network
private_key = wallet_list[int(chosen_key) - 1][2]
public_key = wallet_list[int(chosen_key) - 1][1]
new_transaction = Transaction(str(public_key), str(chosen_receiver), str(chosen_payload))
new_transaction.sign_transaction(self.crypto_helper, private_key)
transaction_hash = self.crypto_helper.hash(new_transaction.get_json())
self.network_interface.sendTransaction(new_transaction)
print('Transaction successfully created!')
print()
print(u'Sender: ' + public_key)
print(u'Receiver: ' + str(chosen_receiver))
print(u'Payload: ' + str(chosen_payload))
print(u'Hash: ' + str(transaction_hash))
print()
# case: wallet is empty
else:
print(u'Wallet does not contain any keys! Please create one first!')
input('Press any key to go back to the main menu!')
class DbTestCase(unittest.TestCase):
def setUp(self):
self.database = Db(block_chain_db_file=os.path.abspath(
os.path.join(os.path.dirname(__file__),
'../labchain/resources/labchaindb.sqlite')))
self.init_components()
self.create_transactions()
self.create_blocks()
def test_create_tables(self):
self.assertTrue(self.database.create_tables())
def test_save_block(self):
self.assertTrue(self.database.save_block(self.block1))
self.database.get_blockchain_from_db()
def init_components(self):
node_config = './labchain/resources/node_configuration.ini'
config_reader = ConfigReader(node_config)
tolerance = config_reader.get_config(section='BLOCK_CHAIN',
option='TOLERANCE_LEVEL')
pruning = config_reader.get_config(section='BLOCK_CHAIN',
option='TIME_TO_PRUNE')
min_blocks = config_reader.get_config(
section='MINING', option='NUM_OF_BLOCKS_FOR_DIFFICULTY')
self.consensus = Consensus()
self.crypto_helper_obj = crypto.instance()
self.txpool = TxPool(self.crypto_helper_obj)
self.block_list = []
self.blockchain = BlockChain(node_id="nodeId1",
tolerance_value=tolerance,
pruning_interval=pruning,
consensus_obj=self.consensus,
txpool_obj=self.txpool,
crypto_helper_obj=self.crypto_helper_obj,
min_blocks_for_difficulty=min_blocks,
db=self.database,
q=None)
def create_transactions(self):
pr_key1, pub_key1 = self.crypto_helper_obj.generate_key_pair()
pr_key2, pub_key2 = self.crypto_helper_obj.generate_key_pair()
pr_key3, pub_key3 = self.crypto_helper_obj.generate_key_pair()
pr_key4, pub_key4 = self.crypto_helper_obj.generate_key_pair()
self.txn1 = Transaction(pub_key1, pub_key2, "Payload1")
self.txn2 = Transaction(pub_key2, pub_key4, "Payload2")
self.txn3 = Transaction(pub_key3, pub_key1, "Payload3")
self.txn4 = Transaction(pub_key4, pub_key3, "Payload3")
self.txn1.sign_transaction(self.crypto_helper_obj, pr_key1)
self.txn2.sign_transaction(self.crypto_helper_obj, pr_key2)
self.txn3.sign_transaction(self.crypto_helper_obj, pr_key3)
self.txn4.sign_transaction(self.crypto_helper_obj, pr_key4)
self.txn1.transaction_hash = self.crypto_helper_obj.hash(
self.txn1.get_json())
self.txn2.transaction_hash = self.crypto_helper_obj.hash(
self.txn2.get_json())
self.txn3.transaction_hash = self.crypto_helper_obj.hash(
self.txn3.get_json())
self.txn4.transaction_hash = self.crypto_helper_obj.hash(
self.txn4.get_json())
def create_blocks(self):
self.block1 = self.blockchain.create_block([self.txn1, self.txn2])
self.block2 = self.blockchain.create_block([self.txn3, self.txn4])
self.block3 = self.blockchain.create_block([self.txn1, self.txn4])
self.block4 = self.blockchain.create_block([self.txn1, self.txn3])
self.block5 = self.blockchain.create_block([self.txn1, self.txn2])
self.block6 = self.blockchain.create_block([self.txn3, self.txn4])
self.block7 = self.blockchain.create_block([self.txn2, self.txn4])
class BlockChainComponent(unittest.TestCase):
"""Class of testcases for the Blockchain module"""
def setUp(self):
"""Setup phase for each testcase"""
self.init_components()
self.create_transactions()
self.create_blocks()
def test_get_block_range(self):
# get_block_range doesn't consider genesis block so expected length = 0
blocks = self.blockchain.get_block_range(0)
self.assertEqual(len(blocks), 0)
def test_get_block_by_id(self):
# fetching first block whose id = 0
blocks = self.blockchain.get_block_by_id(0)
for block in blocks:
self.assertEqual(block._block_id, 0)
def test_get_block_by_hash(self):
block_info = json.loads(
self.blockchain.get_block_by_hash(
self.blockchain._first_block_hash))
self.assertEqual(block_info['nr'], 0)
def test_add_block(self):
self.blockchain.add_block(self.block1)
#blocks = self.blockchain.get_block_range(0)
#self.assertEqual(len(blocks), 1)
"""
def test_add_block1(self):
# now block8 has a branch with block 6
self.block8 = self.block1 = self.blockchain.create_block([self.txn2, self.txn4])
self.assertFalse(self.blockchain.add_block(self.block8), "Block is deleted")
# block 9 has a normal predecessor block 7
self.block9 = self.block1 = self.blockchain.create_block([self.txn2, self.txn4])
self.assertTrue(self.blockchain.add_block(self.block9), "Block is saved")
"""
def test_switch_to_longest_branch(self):
# now block8 has a branch with block 6
self.block8 = self.block1 = self.blockchain.create_block(
[self.txn2, self.txn4])
# we are trying to add a new block in the block chain
self.blockchain._blockchain[self.crypto_helper_obj.hash(
self.block8.get_json())] = self.block8
# calculating the length of the blockchain before adding new block
prev_block_length = len(self.blockchain._blockchain.items())
# calling the switching branch method
self.blockchain.switch_to_longest_branch()
# calculating the length of the blockchain after adding new block
after_block_length = len(self.blockchain._blockchain.items())
# after branch switching the length of the block should be same
self.assertEqual(prev_block_length, after_block_length,
"Block is deleted")
def test_calculate_diff(self):
# blocks added in setup
blocks, t1, t2, diff = self.blockchain.calculate_diff()
self.assertIsNotNone(blocks)
self.assertIsNotNone(t1)
self.assertIsNotNone(t2)
self.assertIsNotNone(diff)
def test_create_block(self):
# creating new block based on given transaction list
new_block = self.blockchain.create_block([self.txn2, self.txn4])
self.assertIsNotNone(new_block, "New block Created")
"""
def test_send_block_to_neighbour(self):
block_as_json = self.blockchain.send_block_to_neighbour(self.block1)
block_as_object = Block.from_json(block_as_json)
self.assertIsInstance(block_as_object, Block, "Sent the Block information requested by any neighbour")
"""
def request_block_from_neighbour(self):
self.assertEqual(1, 2 - 1, "They are equal")
def init_components(self):
node_config = './labchain/resources/node_configuration.ini'
config_reader = ConfigReader(node_config)
tolerance = config_reader.get_config(section='BLOCK_CHAIN',
option='TOLERANCE_LEVEL')
pruning = config_reader.get_config(section='BLOCK_CHAIN',
option='TIME_TO_PRUNE')
min_blocks = config_reader.get_config(
section='MINING', option='NUM_OF_BLOCKS_FOR_DIFFICULTY')
self.consensus = Consensus()
self.crypto_helper_obj = crypto.instance()
self.txpool = TxPool(self.crypto_helper_obj)
self.block_list = []
self.blockchain = BlockChain(node_id="nodeId1",
tolerance_value=tolerance,
pruning_interval=pruning,
consensus_obj=self.consensus,
txpool_obj=self.txpool,
crypto_helper_obj=self.crypto_helper_obj,
min_blocks_for_difficulty=min_blocks,
db=None,
q=None)
def create_transactions(self):
pr_key1, pub_key1 = self.crypto_helper_obj.generate_key_pair()
pr_key2, pub_key2 = self.crypto_helper_obj.generate_key_pair()
pr_key3, pub_key3 = self.crypto_helper_obj.generate_key_pair()
pr_key4, pub_key4 = self.crypto_helper_obj.generate_key_pair()
self.txn1 = Transaction(pub_key1, pub_key2, "Payload1")
self.txn2 = Transaction(pub_key2, pub_key4, "Payload2")
self.txn3 = Transaction(pub_key3, pub_key1, "Payload3")
self.txn4 = Transaction(pub_key4, pub_key3, "Payload3")
self.txn1.sign_transaction(self.crypto_helper_obj, pr_key1)
self.txn2.sign_transaction(self.crypto_helper_obj, pr_key2)
self.txn3.sign_transaction(self.crypto_helper_obj, pr_key3)
self.txn4.sign_transaction(self.crypto_helper_obj, pr_key4)
self.txn1.transaction_hash = self.crypto_helper_obj.hash(
self.txn1.get_json())
self.txn2.transaction_hash = self.crypto_helper_obj.hash(
self.txn2.get_json())
self.txn3.transaction_hash = self.crypto_helper_obj.hash(
self.txn3.get_json())
self.txn4.transaction_hash = self.crypto_helper_obj.hash(
self.txn4.get_json())
def create_blocks(self):
self.block1 = self.blockchain.create_block([self.txn1, self.txn2])
self.block2 = self.blockchain.create_block([self.txn3, self.txn4])
self.block3 = self.blockchain.create_block([self.txn1, self.txn4])
self.block4 = self.blockchain.create_block([self.txn1, self.txn3])
self.block5 = self.blockchain.create_block([self.txn1, self.txn2])
self.block6 = self.blockchain.create_block([self.txn3, self.txn4])
self.block7 = self.blockchain.create_block([self.txn2, self.txn4])