diff options
-rw-r--r-- | app/models/boot.py | 2 | ||||
-rw-r--r-- | app/models/tests/test_boot_model.py | 35 | ||||
-rw-r--r-- | app/utils/bootimport.py | 1 | ||||
-rw-r--r-- | app/utils/tests/test_bootimport.py | 7 |
4 files changed, 27 insertions, 18 deletions
diff --git a/app/models/boot.py b/app/models/boot.py index f5e75cd..1d02b4b 100644 --- a/app/models/boot.py +++ b/app/models/boot.py @@ -89,6 +89,7 @@ class BootDocument(modb.BaseDocument): self.job_id = None self.kernel_image = None self.load_addr = None + self.mach = None self.metadata = {} self.qemu = None self.qemu_command = None @@ -223,6 +224,7 @@ class BootDocument(modb.BaseDocument): models.KERNEL_KEY: self.kernel, models.LAB_NAME_KEY: self.lab_name, models.LOAD_ADDR_KEY: self.load_addr, + models.MACH_KEY: self.mach, models.METADATA_KEY: self.metadata, models.NAME_KEY: self.name, models.QEMU_COMMAND_KEY: self.qemu_command, diff --git a/app/models/tests/test_boot_model.py b/app/models/tests/test_boot_model.py index 125a469..c34cecb 100644 --- a/app/models/tests/test_boot_model.py +++ b/app/models/tests/test_boot_model.py @@ -55,21 +55,33 @@ class TestBootModel(unittest.TestCase): boot_doc.qemu = "qemu_binary" boot_doc.qemu_command = "qemu_command" boot_doc.metadata = {"foo": "bar"} + boot_doc.mach = "soc" expected = { "_id": "id", + "arch": "arm", "board": "board", + "board_instance": "instance", "boot_log": "boot-log", "boot_log_html": "boot-log-html", "boot_result_description": None, "created_on": "now", "defconfig": "defconfig", + "defconfig_full": "defconfig", "defconfig_id": "defconfig_id", "dtb": None, "dtb_addr": None, "dtb_append": False, "endian": None, "fastboot": False, + "fastboot_cmd": "fastboot", + "file_server_resource": "file-resource", + "file_server_url": "file-server", + "git_branch": "git-branch", + "git_commit": "git-commit", + "git_describe": "git-describe", + "git_url": "git-url", + "initrd": "initrd", "initrd_addr": None, "job": "job", "job_id": "job-id", @@ -77,28 +89,18 @@ class TestBootModel(unittest.TestCase): "kernel_image": None, "lab_name": "lab", "load_addr": None, + "mach": "soc", "metadata": {"foo": "bar"}, "name": "board-job-kernel-defconfig-arm", + "qemu": "qemu_binary", + "qemu_command": "qemu_command", "retries": 10, "status": None, "time": 0, - "version": "1.0", - "warnings": 2, - "git_commit": "git-commit", - "git_branch": "git-branch", - "git_describe": "git-describe", - "git_url": "git-url", - "arch": "arm", - "fastboot_cmd": "fastboot", - "defconfig_full": "defconfig", - "file_server_url": "file-server", - "file_server_resource": "file-resource", - "initrd": "initrd", - "board_instance": "instance", "uimage": "path/to/uImage", "uimage_addr": "uimage_addr", - "qemu": "qemu_binary", - "qemu_command": "qemu_command" + "version": "1.0", + "warnings": 2 } self.assertDictEqual(expected, boot_doc.to_dict()) @@ -152,6 +154,7 @@ class TestBootModel(unittest.TestCase): "kernel_image": "kernel_image", "lab_name": "lab", "load_addr": "12345", + "mach": "soc", "metadata": {"foo": "bar"}, "name": "board-job-kernel-defconfig_full-arm", "qemu": "qemu_binary", @@ -162,7 +165,7 @@ class TestBootModel(unittest.TestCase): "uimage": "path/to/uImage", "uimage_addr": "uimage_addr", "version": "1.0", - "warnings": 2, + "warnings": 2 } boot_doc = mboot.BootDocument.from_json(boot_json) diff --git a/app/utils/bootimport.py b/app/utils/bootimport.py index 9b53073..d7cdd27 100644 --- a/app/utils/bootimport.py +++ b/app/utils/bootimport.py @@ -382,6 +382,7 @@ def _update_boot_doc_from_json(boot_doc, boot_json, json_pop_f): boot_doc.initrd_addr = json_pop_f(models.INITRD_ADDR_KEY, None) boot_doc.kernel_image = json_pop_f(models.KERNEL_IMAGE_KEY, None) boot_doc.load_addr = json_pop_f(models.BOOT_LOAD_ADDR_KEY, None) + boot_doc.mach = json_pop_f(models.MACH_KEY, None) boot_doc.metadata = json_pop_f(models.METADATA_KEY, {}) boot_doc.qemu = json_pop_f(models.QEMU_KEY, None) boot_doc.qemu_command = json_pop_f(models.QEMU_COMMAND_KEY, None) diff --git a/app/utils/tests/test_bootimport.py b/app/utils/tests/test_bootimport.py index 70b7df2..e0fe4ca 100644 --- a/app/utils/tests/test_bootimport.py +++ b/app/utils/tests/test_bootimport.py @@ -51,8 +51,9 @@ class TestParseBoot(unittest.TestCase): kernel_image="zImage", loadaddr="0x80200000", endian="little", - uImage=True, - uimage_addr="xip" + uimage="uimage", + uimage_addr="xip", + mach="soc" ) def tearDown(self): @@ -66,6 +67,8 @@ class TestParseBoot(unittest.TestCase): self.assertEqual(doc.load_addr, "0x80200000") self.assertEqual(doc.endian, "little") self.assertEqual(doc.version, "1.0") + self.assertEqual(doc.mach, "soc") + self.assertEqual(doc.uimage, "uimage") self.assertIsInstance(doc.metadata, types.DictionaryType) def test_check_for_null_with_none(self): |