Add pod role in pod description in the API

[releng.git] / utils / test / result_collection_api / tests / unit / test_pod.py

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 07f55db..b6adc25 100644 (file)
--- 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)