test_api.py 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
  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 tenancy.models import Tenant, TenantGroup
  7. from users.models import Token
  8. from utilities.tests import HttpStatusMixin
  9. class TenantGroupTest(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.tenantgroup1 = TenantGroup.objects.create(name='Test Tenant Group 1', slug='test-tenant-group-1')
  15. self.tenantgroup2 = TenantGroup.objects.create(name='Test Tenant Group 2', slug='test-tenant-group-2')
  16. self.tenantgroup3 = TenantGroup.objects.create(name='Test Tenant Group 3', slug='test-tenant-group-3')
  17. def test_get_tenantgroup(self):
  18. url = reverse('tenancy-api:tenantgroup-detail', kwargs={'pk': self.tenantgroup1.pk})
  19. response = self.client.get(url, **self.header)
  20. self.assertEqual(response.data['name'], self.tenantgroup1.name)
  21. def test_list_tenantgroups(self):
  22. url = reverse('tenancy-api:tenantgroup-list')
  23. response = self.client.get(url, **self.header)
  24. self.assertEqual(response.data['count'], 3)
  25. def test_create_tenantgroup(self):
  26. data = {
  27. 'name': 'Test Tenant Group 4',
  28. 'slug': 'test-tenant-group-4',
  29. }
  30. url = reverse('tenancy-api:tenantgroup-list')
  31. response = self.client.post(url, data, format='json', **self.header)
  32. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  33. self.assertEqual(TenantGroup.objects.count(), 4)
  34. tenantgroup4 = TenantGroup.objects.get(pk=response.data['id'])
  35. self.assertEqual(tenantgroup4.name, data['name'])
  36. self.assertEqual(tenantgroup4.slug, data['slug'])
  37. def test_create_tenantgroup_bulk(self):
  38. data = [
  39. {
  40. 'name': 'Test Tenant Group 4',
  41. 'slug': 'test-tenant-group-4',
  42. },
  43. {
  44. 'name': 'Test Tenant Group 5',
  45. 'slug': 'test-tenant-group-5',
  46. },
  47. {
  48. 'name': 'Test Tenant Group 6',
  49. 'slug': 'test-tenant-group-6',
  50. },
  51. ]
  52. url = reverse('tenancy-api:tenantgroup-list')
  53. response = self.client.post(url, data, format='json', **self.header)
  54. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  55. self.assertEqual(TenantGroup.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_tenantgroup(self):
  60. data = {
  61. 'name': 'Test Tenant Group X',
  62. 'slug': 'test-tenant-group-x',
  63. }
  64. url = reverse('tenancy-api:tenantgroup-detail', kwargs={'pk': self.tenantgroup1.pk})
  65. response = self.client.put(url, data, format='json', **self.header)
  66. self.assertHttpStatus(response, status.HTTP_200_OK)
  67. self.assertEqual(TenantGroup.objects.count(), 3)
  68. tenantgroup1 = TenantGroup.objects.get(pk=response.data['id'])
  69. self.assertEqual(tenantgroup1.name, data['name'])
  70. self.assertEqual(tenantgroup1.slug, data['slug'])
  71. def test_delete_tenantgroup(self):
  72. url = reverse('tenancy-api:tenantgroup-detail', kwargs={'pk': self.tenantgroup1.pk})
  73. response = self.client.delete(url, **self.header)
  74. self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
  75. self.assertEqual(TenantGroup.objects.count(), 2)
  76. class TenantTest(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.tenantgroup1 = TenantGroup.objects.create(name='Test Tenant Group 1', slug='test-tenant-group-1')
  82. self.tenantgroup2 = TenantGroup.objects.create(name='Test Tenant Group 2', slug='test-tenant-group-2')
  83. self.tenant1 = Tenant.objects.create(name='Test Tenant 1', slug='test-tenant-1', group=self.tenantgroup1)
  84. self.tenant2 = Tenant.objects.create(name='Test Tenant 2', slug='test-tenant-2', group=self.tenantgroup1)
  85. self.tenant3 = Tenant.objects.create(name='Test Tenant 3', slug='test-tenant-3', group=self.tenantgroup1)
  86. def test_get_tenant(self):
  87. url = reverse('tenancy-api:tenant-detail', kwargs={'pk': self.tenant1.pk})
  88. response = self.client.get(url, **self.header)
  89. self.assertEqual(response.data['name'], self.tenant1.name)
  90. def test_list_tenants(self):
  91. url = reverse('tenancy-api:tenant-list')
  92. response = self.client.get(url, **self.header)
  93. self.assertEqual(response.data['count'], 3)
  94. def test_create_tenant(self):
  95. data = {
  96. 'name': 'Test Tenant 4',
  97. 'slug': 'test-tenant-4',
  98. 'group': self.tenantgroup1.pk,
  99. }
  100. url = reverse('tenancy-api:tenant-list')
  101. response = self.client.post(url, data, format='json', **self.header)
  102. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  103. self.assertEqual(Tenant.objects.count(), 4)
  104. tenant4 = Tenant.objects.get(pk=response.data['id'])
  105. self.assertEqual(tenant4.name, data['name'])
  106. self.assertEqual(tenant4.slug, data['slug'])
  107. self.assertEqual(tenant4.group_id, data['group'])
  108. def test_create_tenant_bulk(self):
  109. data = [
  110. {
  111. 'name': 'Test Tenant 4',
  112. 'slug': 'test-tenant-4',
  113. },
  114. {
  115. 'name': 'Test Tenant 5',
  116. 'slug': 'test-tenant-5',
  117. },
  118. {
  119. 'name': 'Test Tenant 6',
  120. 'slug': 'test-tenant-6',
  121. },
  122. ]
  123. url = reverse('tenancy-api:tenant-list')
  124. response = self.client.post(url, data, format='json', **self.header)
  125. self.assertHttpStatus(response, status.HTTP_201_CREATED)
  126. self.assertEqual(Tenant.objects.count(), 6)
  127. self.assertEqual(response.data[0]['name'], data[0]['name'])
  128. self.assertEqual(response.data[1]['name'], data[1]['name'])
  129. self.assertEqual(response.data[2]['name'], data[2]['name'])
  130. def test_update_tenant(self):
  131. data = {
  132. 'name': 'Test Tenant X',
  133. 'slug': 'test-tenant-x',
  134. 'group': self.tenantgroup2.pk,
  135. }
  136. url = reverse('tenancy-api:tenant-detail', kwargs={'pk': self.tenant1.pk})
  137. response = self.client.put(url, data, format='json', **self.header)
  138. self.assertHttpStatus(response, status.HTTP_200_OK)
  139. self.assertEqual(Tenant.objects.count(), 3)
  140. tenant1 = Tenant.objects.get(pk=response.data['id'])
  141. self.assertEqual(tenant1.name, data['name'])
  142. self.assertEqual(tenant1.slug, data['slug'])
  143. self.assertEqual(tenant1.group_id, data['group'])
  144. def test_delete_tenant(self):
  145. url = reverse('tenancy-api:tenant-detail', kwargs={'pk': self.tenant1.pk})
  146. response = self.client.delete(url, **self.header)
  147. self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
  148. self.assertEqual(Tenant.objects.count(), 2)