testfile.write("test")
testfile.close()
- with NamedTemporaryFile("w", suffix=".wks") as wks:
- wks.writelines(['part / --source extra_partition --ondisk sda --fstype=ext4 --label foo --align 4 --size 5M\n',
- 'part / --source extra_partition --ondisk sda --fstype=ext4 --uuid e7d0824e-cda3-4bed-9f54-9ef5312d105d --align 4 --size 5M\n',
- 'part / --source extra_partition --ondisk sda --fstype=ext4 --label bar --align 4 --size 5M\n'])
- wks.flush()
- _, wicimg = self._get_wic(wks.name)
+ oldpath = os.environ['PATH']
+ os.environ['PATH'] = get_bb_var("PATH", "wic-tools")
+
+ try:
+ with NamedTemporaryFile("w", suffix=".wks") as wks:
+ wks.writelines(['part / --source extra_partition --ondisk sda --fstype=ext4 --label foo --align 4 --size 5M\n',
+ 'part / --source extra_partition --ondisk sda --fstype=ext4 --uuid e7d0824e-cda3-4bed-9f54-9ef5312d105d --align 4 --size 5M\n',
+ 'part / --source extra_partition --ondisk sda --fstype=ext4 --label bar --align 4 --size 5M\n'])
+ wks.flush()
+ _, wicimg = self._get_wic(wks.name)
- result = runCmd("wic ls %s | wc -l" % wicimg)
- self.assertEqual('4', result.output, msg="Expect 3 partitions, not %s" % result.output)
+ result = runCmd("wic ls %s | wc -l" % wicimg)
+ self.assertEqual('4', result.output, msg="Expect 3 partitions, not %s" % result.output)
- for part, file in enumerate(["foo.conf", "foobar.conf", "bar.conf"]):
- result = runCmd("wic ls %s:%d | grep -q \"%s\"" % (wicimg, part + 1, file))
- self.assertEqual(0, result.status, msg="File '%s' not found in the partition #%d" % (file, part))
+ for part, file in enumerate(["foo.conf", "foobar.conf", "bar.conf"]):
+ result = runCmd("wic ls %s:%d | grep -q \"%s\"" % (wicimg, part + 1, file))
+ self.assertEqual(0, result.status, msg="File '%s' not found in the partition #%d" % (file, part))
- self.remove_config(config)
+ self.remove_config(config)
+
+ finally:
+ os.environ['PATH'] = oldpath
def test_fs_types(self):
"""Test filesystem types for empty and not empty partitions"""