66from openstack import exceptions
77
88from openstack_mcp_server .tools .identity_tools import IdentityTools
9- from openstack_mcp_server .tools .response .identity import Domain , Region , Project
9+ from openstack_mcp_server .tools .response .identity import (
10+ Domain ,
11+ Region ,
12+ Project ,
13+ )
1014
1115
1216class TestIdentityTools :
@@ -736,14 +740,17 @@ def test_get_projects_success(self, mock_get_openstack_conn_identity):
736740 mock_project2 .is_enabled = False
737741 mock_project2 .domain_id = "domain22222222222222222222222222"
738742 mock_project2 .parent_id = "default"
739-
743+
740744 # Configure mock project.projects()
741- mock_conn .identity .projects .return_value = [mock_project1 , mock_project2 ]
742-
745+ mock_conn .identity .projects .return_value = [
746+ mock_project1 ,
747+ mock_project2 ,
748+ ]
749+
743750 # Test get_projects()
744751 identity_tools = self .get_identity_tools ()
745752 result = identity_tools .get_projects ()
746-
753+
747754 # Verify results
748755 assert result == [
749756 Project (
@@ -763,29 +770,27 @@ def test_get_projects_success(self, mock_get_openstack_conn_identity):
763770 parent_id = "default" ,
764771 ),
765772 ]
766-
773+
767774 # Verify mock calls
768775 mock_conn .identity .projects .assert_called_once ()
769-
770-
771-
776+
772777 def test_get_projects_empty_list (self , mock_get_openstack_conn_identity ):
773778 """Test getting identity projects when there are no projects."""
774779 mock_conn = mock_get_openstack_conn_identity
775780
776781 # Empty project list
777782 mock_conn .identity .projects .return_value = []
778-
783+
779784 # Test get_projects()
780785 identity_tools = self .get_identity_tools ()
781786 result = identity_tools .get_projects ()
782-
787+
783788 # Verify results
784789 assert result == []
785-
790+
786791 # Verify mock calls
787792 mock_conn .identity .projects .assert_called_once ()
788-
793+
789794 def test_get_project_success (self , mock_get_openstack_conn_identity ):
790795 """Test getting a identity project successfully."""
791796 mock_conn = mock_get_openstack_conn_identity
@@ -805,7 +810,7 @@ def test_get_project_success(self, mock_get_openstack_conn_identity):
805810 # Test get_project()
806811 identity_tools = self .get_identity_tools ()
807812 result = identity_tools .get_project (name = "ProjectOne" )
808-
813+
809814 # Verify results
810815 assert result == Project (
811816 id = "project1111111111111111111111111" ,
@@ -815,13 +820,13 @@ def test_get_project_success(self, mock_get_openstack_conn_identity):
815820 domain_id = "domain1111111111111111111111111" ,
816821 parent_id = "parentproject1111111111111111111" ,
817822 )
818-
823+
819824 # Verify mock calls
820825 mock_conn .identity .find_project .assert_called_once_with (
821826 name_or_id = "ProjectOne" ,
822827 ignore_missing = False ,
823828 )
824-
829+
825830 def test_get_project_not_found (self , mock_get_openstack_conn_identity ):
826831 """Test getting a identity project that does not exist."""
827832 mock_conn = mock_get_openstack_conn_identity
@@ -835,22 +840,16 @@ def test_get_project_not_found(self, mock_get_openstack_conn_identity):
835840
836841 # Test get_project()
837842 identity_tools = self .get_identity_tools ()
838-
843+
839844 # Verify exception is raised
840845 with pytest .raises (
841846 exceptions .NotFoundException ,
842847 match = "Project 'ProjectOne' not found" ,
843848 ):
844849 identity_tools .get_project (name = "ProjectOne" )
845-
850+
846851 # Verify mock calls
847852 mock_conn .identity .find_project .assert_called_once_with (
848853 name_or_id = "ProjectOne" ,
849854 ignore_missing = False ,
850855 )
851-
852-
853-
854-
855-
856-
0 commit comments