Merge "Add pod role in pod description in the API"

2 files changed, 15 insertions, 5 deletions

diff --git a/utils/test/result_collection_api/resources/pod_models.py b/utils/test/result_collection_api/resources/pod_models.py
index 65b82f770..b02e3c222 100644
--- a/utils/test/result_collection_api/resources/pod_models.py
+++ b/utils/test/result_collection_api/resources/pod_models.py
@@ -1,24 +1,32 @@
 __author__ = '__serena__'
 
+# name: name of the POD e.g. zte-1
+# mode: metal or virtual
+# details: any detail
+# role: ci-pod or community-pod or single-node
+
 
 class PodCreateRequest(object):
-    def __init__(self, name='', mode='', details=''):
+    def __init__(self, name='', mode='', details='', role=""):
         self.name = name
         self.mode = mode
         self.details = details
+        self.role = role
 
     def format(self):
         return {
             "name": self.name,
             "mode": self.mode,
             "details": self.details,
+            "role": self.role,
         }
 
 
 class Pod(PodCreateRequest):
     """ describes a POD platform """
-    def __init__(self, name='', mode='', details='', _id='', create_date=''):
-        super(Pod, self).__init__(name, mode, details)
+    def __init__(self, name='', mode='', details='', role="",
+                 _id='', create_date=''):
+        super(Pod, self).__init__(name, mode, details, role)
         self._id = _id
         self.creation_date = create_date
 
@@ -33,6 +41,7 @@ class Pod(PodCreateRequest):
         p.name = pod_dict.get('name')
         p.mode = pod_dict.get('mode')
         p.details = pod_dict.get('details')
+        p.role = pod_dict.get('role')
         return p
 
     def format(self):
diff --git a/utils/test/result_collection_api/tests/unit/test_pod.py b/utils/test/result_collection_api/tests/unit/test_pod.py
index 07f55db8d..b6adc25ec 100644
--- a/utils/test/result_collection_api/tests/unit/test_pod.py
+++ b/utils/test/result_collection_api/tests/unit/test_pod.py
@@ -9,8 +9,8 @@ from common.constants import HTTP_OK, HTTP_BAD_REQUEST, \
 class TestPodBase(TestBase):
     def setUp(self):
         super(TestPodBase, self).setUp()
-        self.req_d = PodCreateRequest('zte-1', 'fuel', 'zte pod 1')
-        self.req_e = PodCreateRequest('zte-2', 'apex', 'zte pod 2')
+        self.req_d = PodCreateRequest('zte-1', 'virtual', 'zte pod 1','ci-pod')
+        self.req_e = PodCreateRequest('zte-2', 'metal', 'zte pod 2')
         self.get_res = Pod
         self.list_res = Pods
         self.basePath = '/pods'
@@ -21,6 +21,7 @@ class TestPodBase(TestBase):
         self.assertEqual(pod.name, req.name)
         self.assertEqual(pod.mode, req.mode)
         self.assertEqual(pod.details, req.details)
+        self.assertEqual(pod.role, req.role)        
         self.assertIsNotNone(pod.creation_date)
         self.assertIsNotNone(pod._id)