def test_filter_apartments(self):
realtor = UserFactory.create_realtor(verified=True)
client = UserFactory.create_client(verified=True)
ApartmentFactory.create_apartment(realtor)
ApartmentFactory.create_apartment(realtor)
self.client.force_authenticate(user=client)
url = '{}?min_size=70'.format(self.url)
response = self.client.get(url, format='json')
self.assertEqual(len(response.json().get('results')), 2)
url = '{}?min_size=90'.format(self.url)
response = self.client.get(url, format='json')
self.assertEqual(len(response.json().get('results')), 0)
url = '{}?max_price=300'.format(self.url)
response = self.client.get(url, format='json')
self.assertEqual(len(response.json().get('results')), 0)
url = '{}?number_of_rooms=4'.format(self.url)
response = self.client.get(url, format='json')
self.assertEqual(len(response.json().get('results')), 2)
url = '{}?number_of_rooms=2'.format(self.url)
response = self.client.get(url, format='json')
self.assertEqual(len(response.json().get('results')), 0)
def test_client_get_apartments(self):
client = UserFactory.create_client(verified=True)
owner = UserFactory.create_realtor(verified=True)
ApartmentFactory.create_apartment(owner)
ApartmentFactory.create_apartment(owner, status='Rented')
self.client.force_authenticate(user=client)
response = self.client.get(self.url, format='json')
self.assertEqual(len(response.json().get('results')), 1)
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_client_update_delete_apartment(self):
client = UserFactory.create_client(verified=True)
owner = UserFactory.create_realtor(verified=True)
apartment = ApartmentFactory.create_apartment(owner)
body = {'name': 'Apartment new name'}
url = '{}{}/'.format(self.url, apartment.id)
self.client.force_authenticate(user=client)
response = self.client.patch(url, body, format='json')
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
response = self.client.delete(url, format='json')
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
def test_realtor_update_delete_apartment_of_another(self):
realtor = UserFactory.create_realtor(verified=True)
realtor1 = UserFactory.create_realtor(email='*****@*****.**',
verified=True)
apartment = ApartmentFactory.create_apartment(realtor1)
self.client.force_authenticate(user=realtor)
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_403_FORBIDDEN)
response = self.client.delete(url, format='json')
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
def test_realtor_update_delete_apartment(self):
realtor = UserFactory.create_realtor(verified=True)
apartment = ApartmentFactory.create_apartment(realtor)
self.client.force_authenticate(user=realtor)
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)