def test_get_transaction(self):
tx = Transaction()
input = TransactionInput()
input.address = ""
tx.add_input(input)
tx.generate_hash()
mp = Mempool()
mp.flush_mempool()
mp.add_transaction(tx)
self.assertEqual(mp.get_transaction().to_json(), tx.to_json())
mp.close()
def test_remove_transaction(self):
tx = Transaction()
input = TransactionInput()
input.address = ""
tx.add_input(input)
tx.generate_hash()
mp = Mempool()
mp.flush_mempool()
mp.add_transaction(tx)
self.assertEqual(mp.get_size(), 1)
mp.remove_transaction(tx.hash)
self.assertEqual(mp.get_size(), 0)
mp.close()