def test_admin_delete_himself(self):
admin = UserFactory.create_admin(verified=True)
self.client.force_authenticate(user=admin)
url = '{}{}/'.format(self.user_url, admin.id)
response = self.client.delete(url, format='json')
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
def test_admin_create_apartment(self):
admin = UserFactory.create_admin(verified=True)
self.client.force_authenticate(user=admin)
response = self.client.post(self.url,
self.apartment_data,
format='json')
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
def test_admin_get_users(self):
admin = UserFactory.create_admin(verified=True)
self.client.force_authenticate(user=admin)
response = self.client.get(self.user_url, format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(len(response.json().get('results')), 1)
def test_admin_invite_new_user_with_invalid_email(self):
admin = UserFactory.create_admin(verified=True)
self.client.force_authenticate(user=admin)
url = '{}invite/'.format(self.user_url)
body = {'email': 'new'}
response = self.client.post(url, body, format='json')
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
def test_admin_delete_user(self):
admin = UserFactory.create_admin(verified=True)
client = UserFactory.create_client()
self.client.force_authenticate(user=admin)
url = '{}{}/'.format(self.user_url, client.id)
self.client.delete(url, format='json')
with self.assertRaises(User.DoesNotExist):
User.objects.get(pk=client.id)
def test_admin_invite_existing_user(self):
admin = UserFactory.create_admin(verified=True)
client = UserFactory.create_client()
self.client.force_authenticate(user=admin)
url = '{}invite/'.format(self.user_url)
body = {'email': client.email}
response = self.client.post(url, body, format='json')
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
def test_admin_update_user(self):
admin = UserFactory.create_admin(verified=True)
client = UserFactory.create_client()
self.client.force_authenticate(user=admin)
url = '{}{}/'.format(self.user_url, client.id)
body = {'last_name': 'User Editted'}
response = self.client.patch(url, body, format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.json().get('last_name'), 'User Editted')
def test_realtor_admin_get_apartments(self):
owner = UserFactory.create_realtor(verified=True)
another_realtor = UserFactory.create_realtor('*****@*****.**',
verified=True)
admin = UserFactory.create_admin(verified=True)
ApartmentFactory.create_apartment(owner)
ApartmentFactory.create_apartment(owner, status='Rented')
self.client.force_authenticate(user=another_realtor)
response = self.client.get(self.url, format='json')
self.assertEqual(len(response.json().get('results')), 2)
self.client.force_authenticate(user=admin)
response = self.client.get(self.url, format='json')
self.assertEqual(len(response.json().get('results')), 2)
def test_admin_update_delete_apartment(self):
realtor = UserFactory.create_realtor(verified=True)
admin = UserFactory.create_admin(verified=True)
apartment = ApartmentFactory.create_apartment(realtor)
self.client.force_authenticate(user=admin)
url = '{}{}/'.format(self.url, apartment.id)
body = {'name': 'Apartment New Name'}
response = self.client.patch(url, body, format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
apartment.refresh_from_db()
self.assertEqual(apartment.name, body['name'])
response = self.client.delete(url, format='json')
with self.assertRaises(Apartment.DoesNotExist):
Apartment.objects.get(pk=apartment.id)