Only delete the translation if it exists (bug 624852)

This commit is contained in:
Kumar McMillan 2011-01-11 17:24:30 -06:00
Родитель e7781c5170
Коммит 9834990363
2 изменённых файлов: 11 добавлений и 1 удалений

Просмотреть файл

@ -1826,6 +1826,15 @@ class TestProfileStatusBar(TestProfileBase):
eq_(addon.takes_contributions, False)
eq_(addon.wants_contributions, False)
def test_remove_profile_without_content(self):
# See bug 624852
self.addon.the_reason = self.addon.the_future = None
self.addon.save()
self.client.post(self.remove_url)
addon = self.get_addon()
eq_(addon.the_reason, None)
eq_(addon.the_future, None)
def test_remove_both(self):
self.addon.the_reason = self.addon.the_future = '...'
self.addon.wants_contributions = True

Просмотреть файл

@ -157,4 +157,5 @@ def delete_translation(obj, fieldname):
field = obj._meta.get_field(fieldname)
trans = getattr(obj, field.name)
obj.update(**{field.name: None})
Translation.objects.filter(id=trans.id).delete()
if trans:
Translation.objects.filter(id=trans.id).delete()