Fix FileVDI.getAllocatedSize

Signed-off-by: Damien Thenot <damien.thenot@vates.tech>
xcp-ng · Nov 27, 2024 · 7c0a23a · 7c0a23a
1 parent e207c0a
commit 7c0a23a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/drivers/cleanup.py b/drivers/cleanup.py
@@ -1193,6 +1193,11 @@ def delete(self):
             raise util.SMException("os.unlink(%s) failed" % self.path)
         VDI.delete(self)
 
+    def getAllocatedSize(self):
+        if self._sizeAllocated == -1:
+            self._sizeAllocated = vhdutil.getAllocatedSize(self.path)
+        return self._sizeAllocated
+
 
 class LVHDVDI(VDI):
     """Object representing a VDI in an LVHD SR"""

diff --git a/tests/test_cleanup.py b/tests/test_cleanup.py
@@ -1365,6 +1365,7 @@ def add_vdis_for_coalesce(self, sr):
         vdi = cleanup.FileVDI(sr, vdi_uuid, False)
         vdi.path = '%s.vhd' % (vdi_uuid)
         vdi.parent = parent
+        vdi._sizeAllocated = 20971520 #10 blocks of 2MB changed in the child
         parent.children.append(vdi)
 
         sr.vdis[vdi_uuid] = vdi