Refactor progress

Author: brunorijsman

common/block.py

@@ -49,6 +49,20 @@ def uuid(self):
         """
         return self._block_uuid
 
+    @property
+    def size(self):
+        """
+        The size of the block in bytes.
+        """
+        return self._size
+
+    @property
+    def data(self):
+        """
+        The data of the block.
+        """
+        return self._data
+
     @property
     def nr_used_bytes(self):
         """

common/tests/test_block.py

@@ -44,6 +44,8 @@ def test_properties():
     data = _bytes_test_pattern(size)
     block = Block(uuid, data)
     assert block.uuid == uuid
+    assert block.size == size
+    assert block.data == data
     assert block.nr_unused_bytes == size
 
 

common/tests/test_pool.py

@@ -6,6 +6,7 @@
 from typing import List
 from common.pool import Pool
 from common.block import Block
+from common.utils import bytes_to_str
 
 
 def _bytes_test_pattern(size):
@@ -52,3 +53,24 @@ def test_nr_unused_bytes():
     assert pool.nr_unused_bytes == 60
     _allocation = pool.allocate(10, purpose="test")
     assert pool.nr_unused_bytes == 50
+
+
+def test_to_mgmt():
+    """
+    Get the management status.
+    """
+    pool, blocks = _create_test_pool_and_block([10, 20])
+    pool_mgmt = pool.to_mgmt()
+    assert pool_mgmt == {
+        "blocks": [
+            {
+                "uuid": str(block.uuid),
+                "size": block.size,
+                "data": bytes_to_str(block.data, truncate=True),
+                "nr_used_bytes": block.nr_used_bytes,
+                "nr_unused_bytes": block.nr_unused_bytes,
+            }
+            for block in blocks
+        ],
+        "owner": str(pool.owner),
+    }