utils/test/result_collection_api/tests/unit/test_pod.py

   1 import unittest
   2
   3 from test_base import TestBase
   4 from resources.pod_models import PodCreateRequest, Pod, Pods
   5 from common.constants import HTTP_OK, HTTP_BAD_REQUEST, \
   6     HTTP_FORBIDDEN, HTTP_NOT_FOUND
   7
   8
   9 class TestPodBase(TestBase):
  10     def setUp(self):
  11         super(TestPodBase, self).setUp()
  12         self.req_d = PodCreateRequest('zte-1', 'virtual', 'zte pod 1','ci-pod')
  13         self.req_e = PodCreateRequest('zte-2', 'metal', 'zte pod 2')
  14         self.get_res = Pod
  15         self.list_res = Pods
  16         self.basePath = '/pods'
  17
  18     def assert_get_body(self, pod, req=None):
  19         if not req:
  20             req = self.req_d
  21         self.assertEqual(pod.name, req.name)
  22         self.assertEqual(pod.mode, req.mode)
  23         self.assertEqual(pod.details, req.details)
  24         self.assertEqual(pod.role, req.role)
  25         self.assertIsNotNone(pod.creation_date)
  26         self.assertIsNotNone(pod._id)
  27
  28
  29 class TestPodCreate(TestPodBase):
  30     def test_withoutBody(self):
  31         (code, body) = self.create()
  32         self.assertEqual(code, HTTP_BAD_REQUEST)
  33
  34     def test_success(self):
  35         code, body = self.create_d()
  36         self.assertEqual(code, HTTP_OK)
  37         self.assert_create_body(body)
  38
  39     def test_alreadyExist(self):
  40         self.create_d()
  41         code, body = self.create_d()
  42         self.assertEqual(code, HTTP_FORBIDDEN)
  43         self.assertIn('already exists', body)
  44
  45     def _assertMeta(self, meta, success):
  46         self.assertEqual(meta.success, success)
  47         if success:
  48             self.assertEqual(meta.uri, '/pods/{}'.format(self.req_d.name))
  49
  50
  51 class TestPodGet(TestPodBase):
  52     def test_notExist(self):
  53         code, body = self.get('notExist')
  54         self.assertEqual(code, HTTP_NOT_FOUND)
  55
  56     def test_getOne(self):
  57         self.create_d()
  58         code, body = self.get(self.req_d.name)
  59         self.assert_get_body(body)
  60
  61     def test_list(self):
  62         self.create_d()
  63         self.create_e()
  64         code, body = self.get()
  65         self.assertEqual(len(body.pods), 2)
  66         for pod in body.pods:
  67             if self.req_d.name == pod.name:
  68                 self.assert_get_body(pod)
  69             else:
  70                 self.assert_get_body(pod, self.req_e)
  71
  72 if __name__ == '__main__':
  73     unittest.main()