def test_filter(self): Room.create(325, 100, False) Room.create(314, 20, True) response = self.client.get(self.FILTER_PATH, data={'is_computer_room': 'true'}) self.assertEqual(response.status_code, status.HTTP_200_OK) response_data = response.json() for dict_ in response_data: self.assertTrue(dict_['is_computer_room'])
def test_filter_min_capacity(self): Room.create(326, 100, False) Room.create(315, 20, True) response = self.client.get(self.FILTER_PATH, data={'min_capacity': 50}) self.assertEqual(response.status_code, status.HTTP_200_OK) response_data = response.json() for dict_ in response_data: self.assertGreaterEqual(dict_['capacity'], 50)
def setUp(self): self.user = User.objects.get(username='******') self.client.force_authenticate(user=self.user) Room.create(self.ROOM_NUMBER, self.ROOM_CAPACITY, True)