diff --git a/blockchain_integration/pi_network/tests/test_pi_network_db.py b/blockchain_integration/pi_network/tests/test_pi_network_db.py index 758ac7237..7f985a065 100644 --- a/blockchain_integration/pi_network/tests/test_pi_network_db.py +++ b/blockchain_integration/pi_network/tests/test_pi_network_db.py @@ -1,13 +1,15 @@ import unittest + from pi_network.pi_network_db import PiNetworkDB -from pi_network.utils.cryptographic_helpers import encrypt_data, decrypt_data +from pi_network.utils.cryptographic_helpers import decrypt_data, encrypt_data + class TestPiNetworkDB(unittest.TestCase): def setUp(self): self.pi_network_db = PiNetworkDB() def test_create_transaction(self): - transaction_data = {'amount': 10.99, 'ender': 'Alice', 'eceiver': 'Bob'} + transaction_data = {"amount": 10.99, "ender": "Alice", "eceiver": "Bob"} encrypted_data = encrypt_data(transaction_data) transaction_id = self.pi_network_db.create_transaction(encrypted_data) self.assertIsNotNone(transaction_id) @@ -16,10 +18,12 @@ def test_get_transaction(self): transaction_id = 1 encrypted_data = self.pi_network_db.get_transaction(transaction_id) decrypted_data = decrypt_data(encrypted_data) - self.assertEqual(decrypted_data, {'amount': 10.99, 'ender': 'Alice', 'eceiver': 'Bob'}) + self.assertEqual( + decrypted_data, {"amount": 10.99, "ender": "Alice", "eceiver": "Bob"} + ) def test_create_user(self): - user_data = {'username': 'Alice', 'password': 'password123'} + user_data = {"username": "Alice", "password": "password123"} encrypted_data = encrypt_data(user_data) user_id = self.pi_network_db.create_user(encrypted_data) self.assertIsNotNone(user_id) @@ -28,11 +32,13 @@ def test_get_user(self): user_id = 1 encrypted_data = self.pi_network_db.get_user(user_id) decrypted_data = decrypt_data(encrypted_data) - self.assertEqual(decrypted_data, {'username': 'Alice', 'password': 'password123'}) + self.assertEqual( + decrypted_data, {"username": "Alice", "password": "password123"} + ) def test_update_transaction(self): transaction_id = 1 - updated_data = {'amount': 20.99, 'ender': 'Alice', 'eceiver': 'Bob'} + updated_data = {"amount": 20.99, "ender": "Alice", "eceiver": "Bob"} encrypted_data = encrypt_data(updated_data) self.pi_network_db.update_transaction(transaction_id, encrypted_data) @@ -40,5 +46,6 @@ def test_delete_transaction(self): transaction_id = 1 self.pi_network_db.delete_transaction(transaction_id) -if __name__ == '__main__': + +if __name__ == "__main__": unittest.main()