def test_organization_can_have_hierarchy(self):
parent_org = OrganizationFactory()
child_org = OrganizationFactory(parent=parent_org)
grandchild_org = OrganizationFactory(parent=child_org)
self.assertEqual(child_org.parent, parent_org)
self.assertEqual(grandchild_org.parent, child_org)
self.assertIn(grandchild_org, parent_org.get_descendants().all())
def test_organization_can_have_hierarchy(self):
parent_org = OrganizationFactory()
child_org = OrganizationFactory(parent=parent_org)
grandchild_org = OrganizationFactory(parent=child_org)
self.assertEqual(child_org.parent, parent_org)
self.assertEqual(grandchild_org.parent, child_org)
self.assertIn(grandchild_org, parent_org.get_descendants().all())
def test_organization_can_have_hierarchy(self):
parent_org = OrganizationFactory(name='Parent')
child_org = OrganizationFactory(parent=parent_org, name='Child')
grandchild_org = OrganizationFactory(parent=child_org, name='Grandchild')
# Refresh the tree state of the newly created orgs
parent_org.refresh_from_db()
child_org.refresh_from_db()
grandchild_org.refresh_from_db()
self.assertEqual(child_org.parent, parent_org)
self.assertEqual(grandchild_org.parent, child_org)
self.assertIn(grandchild_org, parent_org.get_descendants().all())
