Add type to bisect documents.

Change-Id: I2f8daeff3b7a9af6a324c23546253bc153cda166

2 files changed, 13 insertions, 5 deletions

diff --git a/app/models/bisect.py b/app/models/bisect.py
index b75f026..7e2af85 100644
--- a/app/models/bisect.py
+++ b/app/models/bisect.py
@@ -38,6 +38,7 @@ class BisectDocument(modb.BaseDocument):
         self.good_commit_url = None
         self.job = None
         self.job_id = None
+        self.type = None
 
     @property
     def collection(self):
@@ -103,11 +104,12 @@ class BisectDocument(modb.BaseDocument):
             models.BISECT_GOOD_COMMIT_DATE: self.good_commit_date,
             models.BISECT_GOOD_COMMIT_KEY: self.good_commit,
             models.BISECT_GOOD_COMMIT_URL: self.good_commit_url,
-            models.CREATED_KEY: self.created_on,
             models.COMPARE_TO_KEY: self.compare_to,
+            models.CREATED_KEY: self.created_on,
             models.JOB_ID_KEY: self.job_id,
             models.JOB_KEY: self.job,
             models.NAME_KEY: self.name,
+            models.TYPE_KEY: self.type,
             models.VERSION_KEY: self.version
         }
 
@@ -130,6 +132,7 @@ class BootBisectDocument(BisectDocument):
         self.board = None
         self.defconfig_id = None
         self.boot_id = None
+        self.type = "boot"
 
     def to_dict(self):
         boot_b_dict = super(BootBisectDocument, self).to_dict()
@@ -149,6 +152,7 @@ class DefconfigBisectDocument(BisectDocument):
         self.defconfig_id = None
         self.defconfig_full = None
         self.arch = None
+        self.type = "build"
 
     def to_dict(self):
         def_b_dict = super(DefconfigBisectDocument, self).to_dict()
diff --git a/app/models/tests/test_bisect_model.py b/app/models/tests/test_bisect_model.py
index 697282d..d968140 100644
--- a/app/models/tests/test_bisect_model.py
+++ b/app/models/tests/test_bisect_model.py
@@ -52,7 +52,8 @@ class TestBisectModel(unittest.TestCase):
             "bad_commit_date": None,
             "bad_commit_url": None,
             "version": None,
-            "job_id": None
+            "job_id": None,
+            "type": None
         }
         self.assertDictEqual(expected, bisect_doc.to_dict())
 
@@ -74,7 +75,8 @@ class TestBisectModel(unittest.TestCase):
             "bad_commit_date": None,
             "bad_commit_url": None,
             "version": None,
-            "job_id": None
+            "job_id": None,
+            "type": None
         }
         self.assertDictEqual(expected, bisect_doc.to_dict())
 
@@ -104,7 +106,8 @@ class TestBisectModel(unittest.TestCase):
             "version": "1.0",
             "boot_id": "boot-id",
             "defconfig_id": "defconfig-id",
-            "job_id": "job-id"
+            "job_id": "job-id",
+            "type": "boot"
         }
         self.assertDictEqual(expected, bisect_doc.to_dict())
 
@@ -168,7 +171,8 @@ class TestBisectModel(unittest.TestCase):
             "defconfig": "defconfig-name",
             "job_id": "job-id",
             "defconfig_full": "defconfig-full",
-            "arch": "arm"
+            "arch": "arm",
+            "type": "build"
         }
 
         self.assertDictEqual(expected, bisect_doc.to_dict())