test_api.py 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403
  1. from __future__ import unicode_literals
  2. from django.contrib.auth.models import User
  3. from django.urls import reverse
  4. from rest_framework import status
  5. from rest_framework.test import APITestCase
  6. from users.models import Token
  7. from utilities.tests import HttpStatusMixin
  8. from virtualization.models import Cluster, ClusterGroup, ClusterType, VirtualMachine
  9. class ClusterTypeTest(HttpStatusMixin, APITestCase):
  10. def setUp(self):
  11. user = User.objects.create(username='testuser', is_superuser=True)
  12. token = Token.objects.create(user=user)
  13. self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)}
  14. self.clustertype1 = ClusterType.objects.create(name='Test Cluster Type 1', slug='test-cluster-type-1')
  15. self.clustertype2 = ClusterType.objects.create(name='Test Cluster Type 2', slug='test-cluster-type-2')
  16. self.clustertype3 = ClusterType.objects.create(name='Test Cluster Type 3', slug='test-cluster-type-3')
  17. def test_get_clustertype(self):
  18. url = reverse('virtualization-api:clustertype-detail', kwargs={'pk': self.clustertype1.pk})
  19. response = self.client.get(url, **self.header)
  20. self.assertEqual(response.data['name'], self.clustertype1.name)
  21. def test_list_clustertypes(self):
  22. url = reverse('virtualization-api:clustertype-list')
  23. response = self.client.get(url, **self.header)
  24. self.assertEqual(response.data['count'], 3)
  25. def test_create_clustertype(self):
  26. data = {
  27. 'name': 'Test Cluster Type 4',
  28. 'slug': 'test-cluster-type-4',
  29. }
  30. url = reverse('virtualization-api:clustertype-list')
  31. response = self.client.post(url, data, format='json', **self.header)
  32. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  33. self.assertEqual(ClusterType.objects.count(), 4)
  34. clustertype4 = ClusterType.objects.get(pk=response.data['id'])
  35. self.assertEqual(clustertype4.name, data['name'])
  36. self.assertEqual(clustertype4.slug, data['slug'])
  37. def test_create_clustertype_bulk(self):
  38. data = [
  39. {
  40. 'name': 'Test Cluster Type 4',
  41. 'slug': 'test-cluster-type-4',
  42. },
  43. {
  44. 'name': 'Test Cluster Type 5',
  45. 'slug': 'test-cluster-type-5',
  46. },
  47. {
  48. 'name': 'Test Cluster Type 6',
  49. 'slug': 'test-cluster-type-6',
  50. },
  51. ]
  52. url = reverse('virtualization-api:clustertype-list')
  53. response = self.client.post(url, data, format='json', **self.header)
  54. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  55. self.assertEqual(ClusterType.objects.count(), 6)
  56. self.assertEqual(response.data[0]['name'], data[0]['name'])
  57. self.assertEqual(response.data[1]['name'], data[1]['name'])
  58. self.assertEqual(response.data[2]['name'], data[2]['name'])
  59. def test_update_clustertype(self):
  60. data = {
  61. 'name': 'Test Cluster Type X',
  62. 'slug': 'test-cluster-type-x',
  63. }
  64. url = reverse('virtualization-api:clustertype-detail', kwargs={'pk': self.clustertype1.pk})
  65. response = self.client.put(url, data, format='json', **self.header)
  66. self.assertHttpStatus(response, status.HTTP_200_OK)
  67. self.assertEqual(ClusterType.objects.count(), 3)
  68. clustertype1 = ClusterType.objects.get(pk=response.data['id'])
  69. self.assertEqual(clustertype1.name, data['name'])
  70. self.assertEqual(clustertype1.slug, data['slug'])
  71. def test_delete_clustertype(self):
  72. url = reverse('virtualization-api:clustertype-detail', kwargs={'pk': self.clustertype1.pk})
  73. response = self.client.delete(url, **self.header)
  74. self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
  75. self.assertEqual(ClusterType.objects.count(), 2)
  76. class ClusterGroupTest(HttpStatusMixin, APITestCase):
  77. def setUp(self):
  78. user = User.objects.create(username='testuser', is_superuser=True)
  79. token = Token.objects.create(user=user)
  80. self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)}
  81. self.clustergroup1 = ClusterGroup.objects.create(name='Test Cluster Group 1', slug='test-cluster-group-1')
  82. self.clustergroup2 = ClusterGroup.objects.create(name='Test Cluster Group 2', slug='test-cluster-group-2')
  83. self.clustergroup3 = ClusterGroup.objects.create(name='Test Cluster Group 3', slug='test-cluster-group-3')
  84. def test_get_clustergroup(self):
  85. url = reverse('virtualization-api:clustergroup-detail', kwargs={'pk': self.clustergroup1.pk})
  86. response = self.client.get(url, **self.header)
  87. self.assertEqual(response.data['name'], self.clustergroup1.name)
  88. def test_list_clustergroups(self):
  89. url = reverse('virtualization-api:clustergroup-list')
  90. response = self.client.get(url, **self.header)
  91. self.assertEqual(response.data['count'], 3)
  92. def test_create_clustergroup(self):
  93. data = {
  94. 'name': 'Test Cluster Group 4',
  95. 'slug': 'test-cluster-group-4',
  96. }
  97. url = reverse('virtualization-api:clustergroup-list')
  98. response = self.client.post(url, data, format='json', **self.header)
  99. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  100. self.assertEqual(ClusterGroup.objects.count(), 4)
  101. clustergroup4 = ClusterGroup.objects.get(pk=response.data['id'])
  102. self.assertEqual(clustergroup4.name, data['name'])
  103. self.assertEqual(clustergroup4.slug, data['slug'])
  104. def test_create_clustergroup_bulk(self):
  105. data = [
  106. {
  107. 'name': 'Test Cluster Group 4',
  108. 'slug': 'test-cluster-group-4',
  109. },
  110. {
  111. 'name': 'Test Cluster Group 5',
  112. 'slug': 'test-cluster-group-5',
  113. },
  114. {
  115. 'name': 'Test Cluster Group 6',
  116. 'slug': 'test-cluster-group-6',
  117. },
  118. ]
  119. url = reverse('virtualization-api:clustergroup-list')
  120. response = self.client.post(url, data, format='json', **self.header)
  121. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  122. self.assertEqual(ClusterGroup.objects.count(), 6)
  123. self.assertEqual(response.data[0]['name'], data[0]['name'])
  124. self.assertEqual(response.data[1]['name'], data[1]['name'])
  125. self.assertEqual(response.data[2]['name'], data[2]['name'])
  126. def test_update_clustergroup(self):
  127. data = {
  128. 'name': 'Test Cluster Group X',
  129. 'slug': 'test-cluster-group-x',
  130. }
  131. url = reverse('virtualization-api:clustergroup-detail', kwargs={'pk': self.clustergroup1.pk})
  132. response = self.client.put(url, data, format='json', **self.header)
  133. self.assertHttpStatus(response, status.HTTP_200_OK)
  134. self.assertEqual(ClusterGroup.objects.count(), 3)
  135. clustergroup1 = ClusterGroup.objects.get(pk=response.data['id'])
  136. self.assertEqual(clustergroup1.name, data['name'])
  137. self.assertEqual(clustergroup1.slug, data['slug'])
  138. def test_delete_clustergroup(self):
  139. url = reverse('virtualization-api:clustergroup-detail', kwargs={'pk': self.clustergroup1.pk})
  140. response = self.client.delete(url, **self.header)
  141. self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
  142. self.assertEqual(ClusterGroup.objects.count(), 2)
  143. class ClusterTest(HttpStatusMixin, APITestCase):
  144. def setUp(self):
  145. user = User.objects.create(username='testuser', is_superuser=True)
  146. token = Token.objects.create(user=user)
  147. self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)}
  148. cluster_type = ClusterType.objects.create(name='Test Cluster Type 1', slug='test-cluster-type-1')
  149. cluster_group = ClusterGroup.objects.create(name='Test Cluster Group 1', slug='test-cluster-group-1')
  150. self.cluster1 = Cluster.objects.create(name='Test Cluster 1', type=cluster_type, group=cluster_group)
  151. self.cluster2 = Cluster.objects.create(name='Test Cluster 2', type=cluster_type, group=cluster_group)
  152. self.cluster3 = Cluster.objects.create(name='Test Cluster 3', type=cluster_type, group=cluster_group)
  153. def test_get_cluster(self):
  154. url = reverse('virtualization-api:cluster-detail', kwargs={'pk': self.cluster1.pk})
  155. response = self.client.get(url, **self.header)
  156. self.assertEqual(response.data['name'], self.cluster1.name)
  157. def test_list_clusters(self):
  158. url = reverse('virtualization-api:cluster-list')
  159. response = self.client.get(url, **self.header)
  160. self.assertEqual(response.data['count'], 3)
  161. def test_create_cluster(self):
  162. data = {
  163. 'name': 'Test Cluster 4',
  164. 'type': ClusterType.objects.first().pk,
  165. 'group': ClusterGroup.objects.first().pk,
  166. }
  167. url = reverse('virtualization-api:cluster-list')
  168. response = self.client.post(url, data, format='json', **self.header)
  169. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  170. self.assertEqual(Cluster.objects.count(), 4)
  171. cluster4 = Cluster.objects.get(pk=response.data['id'])
  172. self.assertEqual(cluster4.name, data['name'])
  173. self.assertEqual(cluster4.type.pk, data['type'])
  174. self.assertEqual(cluster4.group.pk, data['group'])
  175. def test_create_cluster_bulk(self):
  176. data = [
  177. {
  178. 'name': 'Test Cluster 4',
  179. 'type': ClusterType.objects.first().pk,
  180. 'group': ClusterGroup.objects.first().pk,
  181. },
  182. {
  183. 'name': 'Test Cluster 5',
  184. 'type': ClusterType.objects.first().pk,
  185. 'group': ClusterGroup.objects.first().pk,
  186. },
  187. {
  188. 'name': 'Test Cluster 6',
  189. 'type': ClusterType.objects.first().pk,
  190. 'group': ClusterGroup.objects.first().pk,
  191. },
  192. ]
  193. url = reverse('virtualization-api:cluster-list')
  194. response = self.client.post(url, data, format='json', **self.header)
  195. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  196. self.assertEqual(Cluster.objects.count(), 6)
  197. self.assertEqual(response.data[0]['name'], data[0]['name'])
  198. self.assertEqual(response.data[1]['name'], data[1]['name'])
  199. self.assertEqual(response.data[2]['name'], data[2]['name'])
  200. def test_update_cluster(self):
  201. cluster_type2 = ClusterType.objects.create(name='Test Cluster Type 2', slug='test-cluster-type-2')
  202. cluster_group2 = ClusterGroup.objects.create(name='Test Cluster Group 2', slug='test-cluster-group-2')
  203. data = {
  204. 'name': 'Test Cluster X',
  205. 'type': cluster_type2.pk,
  206. 'group': cluster_group2.pk,
  207. }
  208. url = reverse('virtualization-api:cluster-detail', kwargs={'pk': self.cluster1.pk})
  209. response = self.client.put(url, data, format='json', **self.header)
  210. self.assertHttpStatus(response, status.HTTP_200_OK)
  211. self.assertEqual(Cluster.objects.count(), 3)
  212. cluster1 = Cluster.objects.get(pk=response.data['id'])
  213. self.assertEqual(cluster1.name, data['name'])
  214. self.assertEqual(cluster1.type.pk, data['type'])
  215. self.assertEqual(cluster1.group.pk, data['group'])
  216. def test_delete_cluster(self):
  217. url = reverse('virtualization-api:cluster-detail', kwargs={'pk': self.cluster1.pk})
  218. response = self.client.delete(url, **self.header)
  219. self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
  220. self.assertEqual(Cluster.objects.count(), 2)
  221. class VirtualMachineTest(HttpStatusMixin, APITestCase):
  222. def setUp(self):
  223. user = User.objects.create(username='testuser', is_superuser=True)
  224. token = Token.objects.create(user=user)
  225. self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)}
  226. cluster_type = ClusterType.objects.create(name='Test Cluster Type 1', slug='test-cluster-type-1')
  227. cluster_group = ClusterGroup.objects.create(name='Test Cluster Group 1', slug='test-cluster-group-1')
  228. self.cluster1 = Cluster.objects.create(name='Test Cluster 1', type=cluster_type, group=cluster_group)
  229. self.virtualmachine1 = VirtualMachine.objects.create(name='Test Virtual Machine 1', cluster=self.cluster1)
  230. self.virtualmachine2 = VirtualMachine.objects.create(name='Test Virtual Machine 2', cluster=self.cluster1)
  231. self.virtualmachine3 = VirtualMachine.objects.create(name='Test Virtual Machine 3', cluster=self.cluster1)
  232. def test_get_virtualmachine(self):
  233. url = reverse('virtualization-api:virtualmachine-detail', kwargs={'pk': self.virtualmachine1.pk})
  234. response = self.client.get(url, **self.header)
  235. self.assertEqual(response.data['name'], self.virtualmachine1.name)
  236. def test_list_virtualmachines(self):
  237. url = reverse('virtualization-api:virtualmachine-list')
  238. response = self.client.get(url, **self.header)
  239. self.assertEqual(response.data['count'], 3)
  240. def test_create_virtualmachine(self):
  241. data = {
  242. 'name': 'Test Virtual Machine 4',
  243. 'cluster': self.cluster1.pk,
  244. }
  245. url = reverse('virtualization-api:virtualmachine-list')
  246. response = self.client.post(url, data, format='json', **self.header)
  247. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  248. self.assertEqual(VirtualMachine.objects.count(), 4)
  249. virtualmachine4 = VirtualMachine.objects.get(pk=response.data['id'])
  250. self.assertEqual(virtualmachine4.name, data['name'])
  251. self.assertEqual(virtualmachine4.cluster.pk, data['cluster'])
  252. def test_create_virtualmachine_bulk(self):
  253. data = [
  254. {
  255. 'name': 'Test Virtual Machine 4',
  256. 'cluster': self.cluster1.pk,
  257. },
  258. {
  259. 'name': 'Test Virtual Machine 5',
  260. 'cluster': self.cluster1.pk,
  261. },
  262. {
  263. 'name': 'Test Virtual Machine 6',
  264. 'cluster': self.cluster1.pk,
  265. },
  266. ]
  267. url = reverse('virtualization-api:virtualmachine-list')
  268. response = self.client.post(url, data, format='json', **self.header)
  269. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  270. self.assertEqual(VirtualMachine.objects.count(), 6)
  271. self.assertEqual(response.data[0]['name'], data[0]['name'])
  272. self.assertEqual(response.data[1]['name'], data[1]['name'])
  273. self.assertEqual(response.data[2]['name'], data[2]['name'])
  274. def test_update_virtualmachine(self):
  275. cluster2 = Cluster.objects.create(
  276. name='Test Cluster 2',
  277. type=ClusterType.objects.first(),
  278. group=ClusterGroup.objects.first()
  279. )
  280. data = {
  281. 'name': 'Test Virtual Machine X',
  282. 'cluster': cluster2.pk,
  283. }
  284. url = reverse('virtualization-api:virtualmachine-detail', kwargs={'pk': self.virtualmachine1.pk})
  285. response = self.client.put(url, data, format='json', **self.header)
  286. self.assertHttpStatus(response, status.HTTP_200_OK)
  287. self.assertEqual(VirtualMachine.objects.count(), 3)
  288. virtualmachine1 = VirtualMachine.objects.get(pk=response.data['id'])
  289. self.assertEqual(virtualmachine1.name, data['name'])
  290. self.assertEqual(virtualmachine1.cluster.pk, data['cluster'])
  291. def test_delete_virtualmachine(self):
  292. url = reverse('virtualization-api:virtualmachine-detail', kwargs={'pk': self.virtualmachine1.pk})
  293. response = self.client.delete(url, **self.header)
  294. self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
  295. self.assertEqual(VirtualMachine.objects.count(), 2)