Complete section: Test the application

Author: 15Dkatz

README.md

@@ -7,3 +7,13 @@ source blockchain-env/bin/activate
 ```
 pip install -r requirements.txt
 ```
+
+**Run the tests**
+
+
+Make sure to activate the virtual environment.
+
+
+```
+python3 -m pytest backend/tests
+```

backend/blockchain/block.py

@@ -2,6 +2,13 @@
 
 from backend.util.crypto_hash import crypto_hash
 
+GENESIS_DATA = {
+    'timestamp': 1,
+    'last_hash': 'genesis_last_hash',
+    'hash': 'genesis_hash',
+    'data': []
+}
+
 class Block:
     """
     Block: a unit of storage.
@@ -41,8 +48,13 @@ def genesis():
         """
         Generate the genesis block.
         """
-        return Block(1, 'genesis_last_hash', 'genesis_hash', [])
-
+        # return Block(
+        #     timestamp=GENESIS_DATA['timestamp'],
+        #     last_hash=GENESIS_DATA['last_hash'],
+        #     hash=GENESIS_DATA['hash'],
+        #     data=GENESIS_DATA['data']
+        # )
+        return Block(**GENESIS_DATA)
 
 def main():
     genesis_block = Block.genesis()

backend/tests/blockchain/test_block.py

@@ -0,0 +1,16 @@
+from backend.blockchain.block import Block, GENESIS_DATA
+
+def test_mine_block():
+    last_block = Block.genesis()
+    data = 'test-data'
+    block = Block.mine_block(last_block, data)
+
+    assert isinstance(block, Block)
+    assert block.data == data
+    assert block.last_hash == last_block.hash
+
+def test_genesis():
+    genesis = Block.genesis()
+    assert isinstance(genesis, Block)
+    for key, value in GENESIS_DATA.items():
+        assert getattr(genesis, key) == value

backend/tests/blockchain/test_blockchain.py

@@ -0,0 +1,13 @@
+from backend.blockchain.blockchain import Blockchain
+from backend.blockchain.block import GENESIS_DATA
+
+def test_blockchain_instance():
+    blockchain = Blockchain()
+    assert blockchain.chain[0].hash == GENESIS_DATA['hash']
+
+def test_add_block():
+    blockchain = Blockchain()
+    data = 'test-data'
+    blockchain.add_block(data)
+
+    assert blockchain.chain[-1].data == data

backend/tests/util/test_crypto_hash.py

@@ -0,0 +1,7 @@
+from backend.util.crypto_hash import crypto_hash
+
+def test_crypto_hash():
+    # It should create the same hash with arguments of different data types in
+    # any order
+    assert crypto_hash(1, [2], 'three') == crypto_hash('three', 1, [2])
+    assert crypto_hash('foo') == 'b2213295d564916f89a6a42455567c87c3f480fcd7a1c15e220f17d7169a790b'