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)
