Coverage: add a test for Bootloader.newDefault

Coverage change:
xcp.bootloader: 72.8% -> 81.9%
total: 65.4% -> 66.6%

Signed-off-by: Yann Dirson <[email protected]>

Author: ydirson

tests/data/grub-linux.cfg

@@ -0,0 +1,5 @@
+set default=0
+menuentry 'linux' {
+	linux /boot/vmlinuz-1 ro
+	initrd /boot/initrd.img-1
+}

tests/test_bootloader.py

@@ -1,7 +1,8 @@
 import unittest
 import os
+import shutil
 import subprocess
-from tempfile import NamedTemporaryFile
+from tempfile import NamedTemporaryFile, mkdtemp
 
 from xcp.bootloader import Bootloader
 
@@ -21,6 +22,25 @@ def test_grub2(self):
         proc.stdout.close()
         proc.wait()
 
+class TestLinuxBootloader(unittest.TestCase):
+    def setUp(self):
+        self.tmpdir = mkdtemp(prefix="testbl")
+        bootdir = os.path.join(self.tmpdir, "boot")
+        grubdir = os.path.join(bootdir, "grub")
+        os.makedirs(grubdir)
+        shutil.copyfile("tests/data/grub-linux.cfg", os.path.join(grubdir, "grub.cfg"))
+        with open(os.path.join(bootdir, "vmlinuz-1"), "w"):
+            pass
+        with open(os.path.join(bootdir, "vmlinuz-2"), "w"):
+            pass
+        with open(os.path.join(bootdir, "initrd.img-1"), "w"):
+            pass
+        with open(os.path.join(bootdir, "initrd.img-2"), "w"):
+            pass
+    def tearDown(self):
+        shutil.rmtree(self.tmpdir)
+    def test_grub2_newdefault(self):
+        Bootloader.newDefault("/boot/vmlinuz-2", "/boot/initrd.img-2", root=self.tmpdir)
 
 class TestBootloaderAdHoc(unittest.TestCase):
     def setUp(self):