def test_restaurant_list_visitors(self): # Create a visit to a restaurant restaurant = self.restaurants[0] visit = VisitFactory.create(restaurant=restaurant) visit.save() # Ask for the restaurant details response = self.client.get(reverse('restaurant-detail', args=(visit.restaurant.id,))) # Check the if the visit is displayed self.assertTrue(response.data['visitors'])
def setUp(self): super(VisitsTest, self).setUp() # Create a restaurants self.restaurant = RestaurantFactory.create() self.restaurant.save() # Create a batch of visits self.visits = VisitFactory.create_batch(VISIT_BATCH, restaurant=self.restaurant) [v.save() for v in self.visits] # Get views urls self.visit_url = reverse('visit-list')
def test_restaurant_list_visitors(self): # Create a visit to a restaurant restaurant = self.restaurants[0] visit = VisitFactory.create(restaurant=restaurant) visit.save() # Ask for the restaurant details response = self.client.get( reverse('restaurant-detail', args=(visit.restaurant.id, ))) # Check the if the visit is displayed self.assertTrue(response.data['visitors'])