boot: Add missing fields to boot schema.

Change-Id: I1cb78bf97ce16bd9c6a434f80d2474f3b7834cb1

3 files changed, 28 insertions, 3 deletions

diff --git a/app/models/__init__.py b/app/models/__init__.py
index fe798a7..c111459 100644
--- a/app/models/__init__.py
+++ b/app/models/__init__.py
@@ -92,6 +92,8 @@ NAME_KEY = "name"
 NOT_FIELD_KEY = 'nfield'
 PRIVATE_KEY = 'private'
 PROPERTIES_KEY = 'properties'
+QEMU_COMMAND_KEY = 'qemu_command'
+QEMU_KEY = 'qemu'
 RESULT_KEY = "result"
 RETRIES_KEY = 'retries'
 SKIP_KEY = 'skip'
@@ -103,6 +105,8 @@ SYSTEM_MAP_KEY = 'system_map'
 TEXT_OFFSET_KEY = 'text_offset'
 TIME_KEY = 'time'
 TOKEN_KEY = 'token'
+UIMAGE_ADDR_KEY = 'uimage_addr'
+UIMAGE_KEY = 'uimage'
 UPDATED_KEY = 'updated_on'
 USERNAME_KEY = 'username'
 VERSION_FULL_KEY = 'full_version'
diff --git a/app/models/boot.py b/app/models/boot.py
index 022699a..f5e75cd 100644
--- a/app/models/boot.py
+++ b/app/models/boot.py
@@ -90,9 +90,13 @@ class BootDocument(modb.BaseDocument):
         self.kernel_image = None
         self.load_addr = None
         self.metadata = {}
+        self.qemu = None
+        self.qemu_command = None
         self.retries = 0
         self.status = None
         self.time = 0
+        self.uimage = None
+        self.uimage_addr = None
         self.warnings = 0
 
     @property
@@ -221,9 +225,13 @@ class BootDocument(modb.BaseDocument):
             models.LOAD_ADDR_KEY: self.load_addr,
             models.METADATA_KEY: self.metadata,
             models.NAME_KEY: self.name,
+            models.QEMU_COMMAND_KEY: self.qemu_command,
+            models.QEMU_KEY: self.qemu,
             models.RETRIES_KEY: self.retries,
             models.STATUS_KEY: self.status,
             models.TIME_KEY: self.time,
+            models.UIMAGE_ADDR_KEY: self.uimage_addr,
+            models.UIMAGE_KEY: self.uimage,
             models.VERSION_KEY: self.version,
             models.WARNINGS_KEY: self.warnings
         }
diff --git a/app/models/tests/test_boot_model.py b/app/models/tests/test_boot_model.py
index 7fbc9f5..125a469 100644
--- a/app/models/tests/test_boot_model.py
+++ b/app/models/tests/test_boot_model.py
@@ -50,6 +50,11 @@ class TestBootModel(unittest.TestCase):
         boot_doc.file_server_resource = "file-resource"
         boot_doc.initrd = "initrd"
         boot_doc.board_instance = "instance"
+        boot_doc.uimage = "path/to/uImage"
+        boot_doc.uimage_addr = "uimage_addr"
+        boot_doc.qemu = "qemu_binary"
+        boot_doc.qemu_command = "qemu_command"
+        boot_doc.metadata = {"foo": "bar"}
 
         expected = {
             "_id": "id",
@@ -72,7 +77,7 @@ class TestBootModel(unittest.TestCase):
             "kernel_image": None,
             "lab_name": "lab",
             "load_addr": None,
-            "metadata": {},
+            "metadata": {"foo": "bar"},
             "name": "board-job-kernel-defconfig-arm",
             "retries": 10,
             "status": None,
@@ -89,7 +94,11 @@ class TestBootModel(unittest.TestCase):
             "file_server_url": "file-server",
             "file_server_resource": "file-resource",
             "initrd": "initrd",
-            "board_instance": "instance"
+            "board_instance": "instance",
+            "uimage": "path/to/uImage",
+            "uimage_addr": "uimage_addr",
+            "qemu": "qemu_binary",
+            "qemu_command": "qemu_command"
         }
 
         self.assertDictEqual(expected, boot_doc.to_dict())
@@ -145,11 +154,15 @@ class TestBootModel(unittest.TestCase):
             "load_addr": "12345",
             "metadata": {"foo": "bar"},
             "name": "board-job-kernel-defconfig_full-arm",
+            "qemu": "qemu_binary",
+            "qemu_command": "qemu_command",
             "retries": 10,
             "status": "PASS",
             "time": 0,
+            "uimage": "path/to/uImage",
+            "uimage_addr": "uimage_addr",
             "version": "1.0",
-            "warnings": 2
+            "warnings": 2,
         }
 
         boot_doc = mboot.BootDocument.from_json(boot_json)