def save(self, *args, **kwargs): """Perform checks before saving.""" # There can only be one. if self.parent_org is not None and self.parent_org.parent_org is not None: raise TooManyNestedOrgs super(Organization, self).save(*args, **kwargs) # Create a default cycle for the organization if there isn't one already from seed.models import Cycle Cycle.get_or_create_default(self) from seed.models import Measure Measure.populate_measures(self.id)
def test_default_cycle(self): year = date.today().year - 1 cycle_name = str(year) + ' Calendar Year' self.org = Organization.objects.create() self.assertEqual(self.org.cycles.count(), 1) cycle = Cycle.objects.filter(name=cycle_name, organization=self.org) self.assertEqual(self.org.cycles.count(), 1) cycle.delete() cycle = Cycle.objects.filter(name=cycle_name, organization=self.org) self.assertEqual(self.org.cycles.count(), 0) cycle = Cycle.get_or_create_default(self.org) self.assertEqual(self.org.cycles.count(), 1)