def setUp(self):
self.num_of_mocks_to_create = 5
self.mock_user = UserFactory()
self.mock_homes = SpaceFactory.create_batch(self.num_of_mocks_to_create,
user=self.mock_user)
self.mock_holiday_homes = SpaceFactory.create_batch(self.num_of_mocks_to_create,
user=self.mock_user,
type='holidayHome')
self.url = reverse('spaces-list')
def test_should_only_return_my_spaces(self):
mock_other_user = UserFactory()
mock_other_spaces = SpaceFactory.create_batch(self.num_of_mocks_to_create,
user= mock_other_user)
response = self.client.get(self.url)
for space in response.data:
self.assertEqual(space['user'], self.mock_user.id)