From b7c99c138e4409f24be4ee74e914db0c58dfd519 Mon Sep 17 00:00:00 2001
From: Shinji Suzuki <shinji.suzuki@gmail.com>
Date: Wed, 4 Apr 2018 06:02:55 +0900
Subject: [PATCH] Add unit test.

---
 tests/test-grub.py | 88 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 tests/test-grub.py

diff --git a/tests/test-grub.py b/tests/test-grub.py
new file mode 100644
index 0000000..2f1f457
--- /dev/null
+++ b/tests/test-grub.py
@@ -0,0 +1,88 @@
+from functools import reduce
+import os
+import sys
+from unittest.mock import MagicMock, patch, sentinel
+
+sys.path = ['..'] + sys.path
+from scripts import config
+from scripts import gen
+from scripts import grub
+from scripts import iso
+
+class OpenMock:
+    def __init__(self, *args, **kw):
+        self.writes = []
+
+    def read(self):
+        return """LABEL core
+	MENU LABEL SliTaz core Live
+	COM32 c32box.c32
+	append linux /boot/bzImage initrd=/boot/rootfs4.gz,/boot/rootfs3.gz,/boot/rootfs2.gz,/boot/rootfs1.gz rw root=/dev/null video=-32 autologin
+"""
+    def write(self, data):
+        self.writes.append(data)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *args):
+        pass
+
+OPEN_MOCK = OpenMock()
+
+def gen_log(obj):
+    print ('[*] %s' % obj)
+
+def my_open(fname, mode, **kw):
+    return OPEN_MOCK
+
+def iso_bin_dir(iso_image):
+    return 'isolinux'
+
+def os_path_exists(f):
+    chunks = reduce(lambda accum, x : accum + x.split('/'), f.split('\\'), [])
+    if chunks[-1] in ['multibootusb.log', 'loopback.cfg']:
+        return True
+    if 'arch' in chunks:
+        return True
+    if chunks[1] in ['rootfs1.gz', 'rootfs2.gz']:
+        return True
+    if chunks == ['multibootusb', 'debian-sid', 'boot', 'rootfs4.gz']:
+        return True
+    return False
+
+def os_walk(dirpath):
+    return [('isolinux', [], ['isolinux.cfg'])]
+
+def test_iso2grub2():
+
+    gen_log_mock = MagicMock()
+    iso_bin_dir_mock = MagicMock()
+    os_walk_mock = MagicMock()
+    os_path_exists_mock = MagicMock()
+    open_mock = MagicMock()
+    @patch('scripts.gen.log', gen_log_mock)
+    @patch('scripts.iso.isolinux_bin_dir', iso_bin_dir_mock)
+    @patch('os.walk', os_walk)
+    @patch('os.path.exists', os_path_exists)
+    @patch('builtins.open', open_mock)
+    def _():
+        gen_log_mock.side_effect = gen_log
+        open_mock.side_effect = my_open
+        os_path_exists_mock.side_effect = os_path_exists
+        os_walk_mock.side_effect = os_walk
+        iso_bin_dir_mock.side_effect = iso_bin_dir
+
+        config.image_path = '/home/suzuki/Downloads/debian-sid.iso'
+        grub.iso2grub2('/tmp/mbusb/debian', 'loopback.cfg')
+    _()
+    assert ''.join(OPEN_MOCK.writes)=="""# Extracted from isolinux/isolinux.cfg
+menuentry "SliTaz core Live" {
+    linux /multibootusb/debian-sid/arch/boot/bzImage rw root=/dev/null video=-32 autologin
+    initrd /multibootusb/debian-sid/boot/rootfs4.gz /multibootusb/debian-sid/arch/boot/rootfs3.gz /boot/rootfs2.gz /boot/rootfs1.gz
+}
+
+"""
+    print ("Test Passed.")
+
+test_iso2grub2()