class OrganizationTest(basetest.TestCase): def setUp(self): self.org1 = Organization('1234567890', display_name='My org', lifecycle_state=OrgLifecycleState.ACTIVE) def test_create_org_getters_are_correct(self): """Test the Organization getters.""" my_org_id = '1234567890' my_org_name = 'My org name' org = Organization(my_org_id, display_name=my_org_name, lifecycle_state=OrgLifecycleState.ACTIVE) self.assertEqual(my_org_id, org.id) self.assertEqual(my_org_name, org.display_name) self.assertEqual(Organization.RESOURCE_NAME_FMT % my_org_id, org.name) self.assertEqual(ResourceType.ORGANIZATION, org.type) self.assertEqual(None, org.parent) self.assertEqual(OrgLifecycleState.ACTIVE, org.lifecycle_state) def test_org_type_is_organization(self): """Test that a created Organization is a ResourceType.ORGANIZATION.""" self.assertEqual(ResourceType.ORGANIZATION, self.org1.type) def test_equality(self): """Test equality of an Organization to another is true.""" id_1 = '1234567890' name_1 = 'My org 1' org1 = Organization(id_1, display_name=name_1) id_2 = '1234567890' name_2 = 'My org 1' org2 = Organization(id_2, display_name=name_2) self.assertTrue(org1 == org2) def test_not_equals(self): """Test inequality of an Organization to another.""" id_1 = '1234567890' name_1 = 'My org 1' org1 = Organization(id_1, display_name=name_1) id_2 = '1234567891' name_2 = 'My org 2' org2 = Organization(id_2, display_name=name_2) self.assertTrue(org1 != org2) def test_org_notequals_project(self): """Test that an Organization != Project.""" proj_id = 'my-project-1' proj_num = 1234567890 proj_name = 'My project 1' project1 = Project(proj_id, proj_num, display_name=proj_name) folder_id = '88888' folder_name = 'My folder' folder1 = folder.Folder(folder_id, display_name=folder_name) org_id = '1234567890' org_name = 'My org 1' org1 = Organization(org_id, display_name=org_name) self.assertTrue(org1 != project1) self.assertTrue(org1 != folder1) @mock.patch.object(CloudResourceManagerClient, 'get_organization', autospec=True) def test_org_exists(self, mock_crm): """Tests that the organization exists.""" mock_crm.return_value = True self.assertTrue(self.org1.exists())
class OrganizationTest(basetest.TestCase): def setUp(self): self.org1 = Organization('1234567890', org_name='My org', lifecycle_state=OrgLifecycleState.ACTIVE) def test_create_org_getters_are_correct(self): """Test the Organization getters.""" my_org_id = '1234567890' my_org_name = 'My org name' org = Organization(my_org_id, org_name=my_org_name, lifecycle_state=OrgLifecycleState.ACTIVE) self.assertEqual(my_org_id, org.get_id()) self.assertEqual(my_org_name, org.get_name()) self.assertEqual(ResourceType.ORGANIZATION, org.get_type()) self.assertEqual(None, org.get_parent()) self.assertEqual(OrgLifecycleState.ACTIVE, org.get_lifecycle_state()) def test_org_type_is_organization(self): """Test that a created Organization is a ResourceType.ORGANIZATION.""" self.assertEqual(ResourceType.ORGANIZATION, self.org1.get_type()) def test_org_equals_other_org_is_true(self): """Test equality of an Organization to another is true.""" id_1 = '1234567890' name_1 = 'My org 1' org1 = Organization(id_1, org_name=name_1) id_2 = '1234567890' name_2 = 'My org 1' org2 = Organization(id_2, org_name=name_2) self.assertTrue(org1 == org2) def test_org_notequals_other_org_is_true(self): """Test inequality of an Organization to another.""" id_1 = '1234567890' name_1 = 'My org 1' org1 = Organization(id_1, org_name=name_1) id_2 = '1234567891' name_2 = 'My org 2' org2 = Organization(id_2, org_name=name_2) self.assertTrue(org1 != org2) def test_org_notequals_project_is_true(self): """Test inequality of an Organization to a Project.""" id_1 = 'my-project-1' number_1 = 1234567890 name_1 = 'My project 1' project = Project(id_1, number_1, project_name=name_1) id_2 = '1234567890' name_2 = 'My org 1' org = Organization(id_2, org_name=name_2) self.assertTrue(project != org) def test_org_empty_ancestors(self): """Test that an Organization has no ancestors.""" expected = [] actual = [a for a in self.org1.get_ancestors(include_self=False)] self.assertEqual(expected, actual) def test_org_ancestors_include_self(self): """Test getting ancestry when including self.""" expected = [self.org1] actual = [a for a in self.org1.get_ancestors()] self.assertEqual(expected, actual) @mock.patch.object(CloudResourceManagerClient, 'get_organization', autospec=True) def test_org_exists(self, mock_crm): """Tests that the organization exists.""" mock_crm.return_value = True self.assertTrue(self.org1.exists())