class ActiveModelTests(TestCase):
cleans_up_after_itself = True
def setUp(self):
self.duck = ActiveDuck()
self.duck.save()
def test_ActiveModel_delete(self):
self.assertNotEqual(self.duck.effective_from, None)
self.assertEqual(self.duck.effective_to, None)
self.assertQuerysetEqual(ActiveDuck.objects.all(), [self.duck.id],
transform=operator.attrgetter('id'))
self.assertTrue(self.duck.is_active())
self.assertFalse(self.duck.is_deleted())
self.assertQuerysetEqual(ActiveDuck.objects_all.all(), [self.duck.id],
transform=operator.attrgetter('id'))
self.duck.delete()
self.assertNotEqual(self.duck.effective_from, None)
self.assertNotEqual(self.duck.effective_to, None)
self.assertQuerysetEqual(ActiveDuck.objects.all(), [],
transform=operator.attrgetter('id'))
self.assertFalse(self.duck.is_active())
self.assertTrue(self.duck.is_deleted())
self.assertQuerysetEqual(ActiveDuck.objects_all.all(), [self.duck.id],
transform=operator.attrgetter('id'))
def test_ActiveModel_bulk_delete(self):
self.assertEqual(self.duck.effective_to, None)
ActiveDuck.objects.filter(id__in=[self.duck.id]).delete()
self.duck = ActiveDuck.objects_all.get(id=self.duck.id)
self.assertNotEqual(self.duck.effective_from, None)
self.assertNotEqual(self.duck.effective_to, None)
self.assertFalse(self.duck.is_active())
self.assertTrue(self.duck.is_deleted())
self.assertQuerysetEqual(ActiveDuck.objects.all(), [],
transform=operator.attrgetter('id'))
self.assertQuerysetEqual(ActiveDuck.objects_all.all(), [self.duck.id],
transform=operator.attrgetter('id'))
def test_ActiveModel_wrong_objects_manager(self):
class WrongActiveModel(models.ActiveModel):
objects = Manager()
self.assertRaises(ImproperlyConfigured, WrongActiveModel)
def setUp(self):
self.duck = ActiveDuck()
self.duck.save()
