def test_project_in_folder_returns_folder_ancestor(self): """Test whether the ancestry includes the folder, for a project.""" folder = Folder('folder-1', folder_name='My folder name') project = Project('my-project-id', 333, project_name='My project', parent=folder) expected = [folder] actual = [a for a in project.get_ancestors(include_self=False)] self.assertEqual(expected, actual)
def test_project_in_org_returns_org_ancestor(self): """Test that a Project with Org ancestor returns Org ancestor.""" org = Organization('1234567890', org_name='My org name') project = Project('my-project-id', 333, project_name='My project', parent=org) expected = [org] actual = [a for a in project.get_ancestors(include_self=False)] self.assertEqual(expected, actual)
def test_project_ancestors_include_self(self): """Test Project ancestors when including self.""" org = Organization('1234567890', org_name='My org name') project = Project('my-project-id', 333, project_name='My project', parent=org) expected = [project, org] actual = [a for a in project.get_ancestors()] self.assertEqual(expected, actual)
def test_project_no_org_returns_empty_ancestors(self): """Test that a Project with no parent has no ancestors.""" project = Project('my-project-id', 333, project_name='My project') expected = [] actual = [a for a in project.get_ancestors(include_self=False)] self.assertEqual(expected, actual)