Update token model with new fields.

Change-Id: I70455477319aee246f11bb5dec6fc8a911f05c4d

2 files changed, 16 insertions, 0 deletions

diff --git a/app/models/tests/test_token_model.py b/app/models/tests/test_token_model.py
index f153e0c..7d5e4d9 100644
--- a/app/models/tests/test_token_model.py
+++ b/app/models/tests/test_token_model.py
@@ -115,6 +115,12 @@ class TestTokenModel(unittest.TestCase):
         self.assertEqual(token_obj.is_delete_token, 1)
         self.assertEqual(token_obj.is_post_token, 1)
 
+    def test_token_is_lab_token(self):
+        token_obj = modt.Token()
+        token_obj.is_lab_token = 1
+
+        self.assertEqual(token_obj.is_lab_token, 1)
+
     def test_token_wrong_numeric_value(self):
         token_obj = modt.Token()
         self.assertRaises(ValueError, setattr, token_obj, "is_admin", 2)
diff --git a/app/models/token.py b/app/models/token.py
index 3a643e0..017c4cc 100644
--- a/app/models/token.py
+++ b/app/models/token.py
@@ -55,6 +55,7 @@ class Token(modb.BaseDocument):
     - 4: if the token can perform DELETE
     - 5: if the token is IP restricted
     - 6: if the token can create new tokens
+    - 7: if the token is a boot lab token
     """
 
     def __init__(self):
@@ -286,6 +287,15 @@ class Token(modb.BaseDocument):
         value = check_attribute_value(value)
         self._properties[6] = value
 
+    @property
+    def is_lab_token(self):
+        return self._properties[7]
+
+    @is_lab_token.setter
+    def is_lab_token(self, value):
+        value = check_attribute_value(value)
+        self._properties[7] = value
+
     def is_valid_ip(self, address):
         """Check if an IP address is valid for a token.