def test_prefetch_tags_doesnt_break(self):
mission_burrito = Place(name='Mission Burrito')
mission_burrito.tags.add('mexican', 'burrito')
mission_burrito.save()
atomic_burger = Place(name='Atomic Burger')
atomic_burger.tags.add('burger')
atomic_burger.save()
places = list(Place.objects.order_by('name').prefetch_related('tags'))
self.assertEqual(places[0].name, 'Atomic Burger')
self.assertEqual(places[0].tags.first().name, 'burger')
def test_prefetch_tags_doesnt_break(self):
mission_burrito = Place(name='Mission Burrito')
mission_burrito.tags.add('mexican', 'burrito')
mission_burrito.save()
atomic_burger = Place(name='Atomic Burger')
atomic_burger.tags.add('burger')
atomic_burger.save()
places = list(Place.objects.order_by('name').prefetch_related('tags'))
self.assertEqual(places[0].name, 'Atomic Burger')
self.assertEqual(places[0].tags.first().name, 'burger')
def test_can_access_tags_on_unsaved_instance(self):
mission_burrito = Place(name='Mission Burrito')
self.assertEqual(0, mission_burrito.tags.count())
mission_burrito.tags.add('mexican', 'burrito')
self.assertEqual(2, mission_burrito.tags.count())
self.assertEqual(Tag, mission_burrito.tags.all()[0].__class__)
self.assertTrue([tag for tag in mission_burrito.tags.all() if tag.name == 'mexican'])
mission_burrito.save()
self.assertEqual(2, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.tags.remove('burrito')
self.assertEqual(1, mission_burrito.tags.count())
# should not affect database until we save
self.assertEqual(2, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.save()
self.assertEqual(1, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.tags.clear()
self.assertEqual(0, mission_burrito.tags.count())
# should not affect database until we save
self.assertEqual(1, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.save()
self.assertEqual(0, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.tags.set('mexican', 'burrito')
self.assertEqual(2, mission_burrito.tags.count())
self.assertEqual(0, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.save()
self.assertEqual(2, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
def test_can_access_tags_on_unsaved_instance(self):
mission_burrito = Place(name='Mission Burrito')
self.assertEqual(0, mission_burrito.tags.count())
mission_burrito.tags.add('mexican', 'burrito')
self.assertEqual(2, mission_burrito.tags.count())
self.assertEqual(Tag, mission_burrito.tags.all()[0].__class__)
self.assertTrue([tag for tag in mission_burrito.tags.all() if tag.name == 'mexican'])
mission_burrito.save()
self.assertEqual(2, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.tags.remove('burrito')
self.assertEqual(1, mission_burrito.tags.count())
# should not affect database until we save
self.assertEqual(2, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.save()
self.assertEqual(1, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.tags.clear()
self.assertEqual(0, mission_burrito.tags.count())
# should not affect database until we save
self.assertEqual(1, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.save()
self.assertEqual(0, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.tags.set('mexican', 'burrito')
self.assertEqual(2, mission_burrito.tags.count())
self.assertEqual(0, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())
mission_burrito.save()
self.assertEqual(2, TaggedPlace.objects.filter(content_object_id=mission_burrito.id).count())