disk: Create a fake disk in test mode

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/python/disk.py b/src/python/disk.py
index 5e9293ff3d0739d5b0fcd56be735dba8b30852e1..8a64ea81e0caf8957d4573393c4bbbb53a8e0c38 100644 (file)
--- a/src/python/disk.py
+++ b/src/python/disk.py
@@ -22,12 +22,15 @@ import logging
 import parted
 
 from . import step
+from . import util
 from .errors import *
 from .i18n import _
 
 # Setup logging
 log = logging.getLogger("bricklayer.disk")
 
+TEST_DISK_SIZE = 4294967296 # 4GiB
+
 class Disks(object):
        """
                Disks abstraction class
@@ -46,18 +49,37 @@ class Disks(object):
 
                log.debug("Scanning for disks...")
 
-               for device in parted.getAllDevices():
-                       disk = Disk(self.bricklayer, device)
+               # Perform a fake scan in test mode
+               if self.bricklayer.test:
+                       self._scan_test()
 
-                       # Skip whatever isn't suitable
-                       if not disk.supported:
-                               continue
+               # Otherwise scan for all devices
+               else:
+                       for device in parted.getAllDevices():
+                               disk = Disk(self.bricklayer, device)
 
-                       self.disks.append(disk)
+                               # Skip whatever isn't suitable
+                               if not disk.supported:
+                                       continue
+
+                               self.disks.append(disk)
 
                # Sort them alphabetically
                self.disks.sort()
 
+       def _scan_test(self):
+               """
+                       Fake scan in test mode and create a temporary image
+               """
+               path = util.create_sparse_file("disk", TEST_DISK_SIZE)
+
+               # Open the file with parted
+               device = parted.Device(path)
+
+               # Create a Disk object
+               disk = Disk(self.bricklayer, device)
+               self.disks.append(disk)
+
        @property
        def supported(self):
                """

diff --git a/src/python/util.py b/src/python/util.py
index f5d26fb317aa8474d9fd6c816f7d658adcbf8835..9f1068578ecc8435ed210af8756603c2de86fc9c 100644 (file)
--- a/src/python/util.py
+++ b/src/python/util.py
@@ -18,6 +18,9 @@
 #                                                                             #
 ###############################################################################
 
+import os
+import tempfile
+
 def config_read(f):
        """
                Read a configuration file in key/value format
@@ -40,3 +43,14 @@ def config_read(f):
                res[key] = value
 
        return res
+
+def create_sparse_file(name, size):
+       fd, path = tempfile.mkstemp(prefix="bricklayer-", suffix="-%s" % name)
+
+       # Truncate the file at size (to make it sparse)
+       os.ftruncate(fd, size)
+
+       # Close the file immediately
+       os.close(fd)
+
+       return path