Refactor progress

Author: brunorijsman

common/allocation.py

@@ -22,6 +22,10 @@ class Allocation:
     _fragments: list[Fragment]
 
     def __init__(self, fragments: list[Fragment]):
+        # Don't call this directly. Instead use one of the following:
+        #   Pool.allocate
+        #   Allocation.from_api
+        #   Allocation.from_enc_str
         self._fragments = fragments
 
     @property

common/fragment.py

@@ -30,7 +30,7 @@ class Fragment:
 
     def __init__(self, block, start, size, data):
         # Don't call this directly. Instead use one of the following:
-        #   Block.allocate
+        #   Block.allocate_fragment
         #   Fragment.from_api
         #   Fragment.from_enc_str
         self._block = block

common/tests/test_allocation.py

@@ -0,0 +1,33 @@
+"""
+Unit tests for the Allocation class.
+"""
+
+from uuid import uuid4
+from common.allocation import Allocation
+from common.block import Block
+from common.fragment import Fragment
+
+
+def _bytes_test_pattern(size):
+    return bytes([i % 255 for i in range(size)])
+
+
+def _create_test_block(size):
+    uuid = uuid4()
+    data = _bytes_test_pattern(size)
+    block = Block(uuid, data)
+    return block
+
+
+def test_init():
+    """
+    Initialize an allocation.
+    """
+    block = _create_test_block(100)
+    fragment = Fragment(
+        block=block,
+        start=0,
+        size=10,
+        data=bytes.fromhex("00010203040506070809"),
+    )
+    _allocation = Allocation(fragments=[fragment])