def test_integers(self):
"""Adding an integer as a tag should raise a ValueError"""
mission_burrito = Place(name='Mission Burrito')
with self.assertRaisesRegexp(ValueError, (
r"Cannot add 1 \(<(type|class) 'int'>\). "
r"Expected <class 'django.db.models.base.ModelBase'> or str.")):
mission_burrito.tags.add(1)
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_create_with_tags(self):
class PlaceForm(ClusterForm):
class Meta:
model = Place
exclude_formsets = ['tagged_items', 'reviews']
fields = ['name', 'tags']
form = PlaceForm({
'name': "Mission Burrito",
'tags': "burrito, fajita"
}, instance=Place())
self.assertTrue(form.is_valid())
mission_burrito = form.save()
reloaded_mission_burrito = Place.objects.get(pk=mission_burrito.pk)
self.assertEqual(
set(reloaded_mission_burrito.tags.all()),
set([Tag.objects.get(name='burrito'), Tag.objects.get(name='fajita')])
)
