@@ -120,7 +120,7 @@ async def test_get_limit_value(self, mock_get_client):
120120 "compartment_id" : "ocid1.compartment.oc1..xxxx" ,
121121 "service_name" : "service1" ,
122122 "name" : "limit_value1" ,
123- "scope_type" : "GLOBAL"
123+ "scope_type" : "GLOBAL" ,
124124 },
125125 )
126126 ).structured_content ["result" ]
@@ -131,13 +131,17 @@ async def test_get_limit_value(self, mock_get_client):
131131 @pytest .mark .asyncio
132132 @patch ("oracle.oci_limits_mcp_server.server.get_limits_client" )
133133 @patch ("oracle.oci_limits_mcp_server.server.list_availability_domains" )
134- async def test_get_resource_availability_ad_scope (self , mock_list_ad , mock_get_client ):
134+ async def test_get_resource_availability_ad_scope (
135+ self , mock_list_ad , mock_get_client
136+ ):
135137 mock_client = MagicMock ()
136138 mock_get_client .return_value = mock_client
137139
138- mock_limit_definition = create_autospec (oci .limits .models .LimitDefinitionSummary )
140+ mock_limit_definition = create_autospec (
141+ oci .limits .models .LimitDefinitionSummary
142+ )
139143 mock_limit_definition .is_resource_availability_supported = True
140- mock_limit_definition .scope_type = 'AD'
144+ mock_limit_definition .scope_type = "AD"
141145
142146 mock_response = create_autospec (oci .response .Response )
143147 mock_response .data = [mock_limit_definition ]
@@ -148,11 +152,18 @@ async def test_get_resource_availability_ad_scope(self, mock_list_ad, mock_get_c
148152 mock_list_ad .return_value = [{"name" : "AD1" }, {"name" : "AD2" }]
149153
150154 mock_response_ad1 = create_autospec (oci .response .Response )
151- mock_response_ad1 .data = oci .limits .models .ResourceAvailability (used = 10 , available = 100 )
155+ mock_response_ad1 .data = oci .limits .models .ResourceAvailability (
156+ used = 10 , available = 100
157+ )
152158 mock_response_ad2 = create_autospec (oci .response .Response )
153- mock_response_ad2 .data = oci .limits .models .ResourceAvailability (used = 20 , available = 200 )
159+ mock_response_ad2 .data = oci .limits .models .ResourceAvailability (
160+ used = 20 , available = 200
161+ )
154162
155- mock_client .get_resource_availability .side_effect = [mock_response_ad1 , mock_response_ad2 ]
163+ mock_client .get_resource_availability .side_effect = [
164+ mock_response_ad1 ,
165+ mock_response_ad2 ,
166+ ]
156167
157168 async with Client (mcp ) as client :
158169 result = await client .call_tool (
@@ -164,21 +175,29 @@ async def test_get_resource_availability_ad_scope(self, mock_list_ad, mock_get_c
164175 },
165176 )
166177
167- assert len (result .structured_content ['result' ]) == 2
168- assert result .structured_content ['result' ][0 ]["availabilityDomain" ] == "AD1"
169- assert result .structured_content ['result' ][0 ]["resourceAvailability" ]["used" ] == 10
170- assert result .structured_content ['result' ][1 ]["availabilityDomain" ] == "AD2"
171- assert result .structured_content ['result' ][1 ]["resourceAvailability" ]["used" ] == 20
178+ assert len (result .structured_content ["result" ]) == 2
179+ assert result .structured_content ["result" ][0 ]["availabilityDomain" ] == "AD1"
180+ assert (
181+ result .structured_content ["result" ][0 ]["resourceAvailability" ]["used" ]
182+ == 10
183+ )
184+ assert result .structured_content ["result" ][1 ]["availabilityDomain" ] == "AD2"
185+ assert (
186+ result .structured_content ["result" ][1 ]["resourceAvailability" ]["used" ]
187+ == 20
188+ )
172189
173190 @pytest .mark .asyncio
174191 @patch ("oracle.oci_limits_mcp_server.server.get_limits_client" )
175192 async def test_get_resource_availability_non_ad_scope (self , mock_get_client ):
176193 mock_client = MagicMock ()
177194 mock_get_client .return_value = mock_client
178195
179- mock_limit_definition = create_autospec (oci .limits .models .LimitDefinitionSummary )
196+ mock_limit_definition = create_autospec (
197+ oci .limits .models .LimitDefinitionSummary
198+ )
180199 mock_limit_definition .is_resource_availability_supported = True
181- mock_limit_definition .scope_type = ' REGION'
200+ mock_limit_definition .scope_type = " REGION"
182201
183202 mock_response = create_autospec (oci .response .Response )
184203 mock_response .data = [mock_limit_definition ]
@@ -187,7 +206,9 @@ async def test_get_resource_availability_non_ad_scope(self, mock_get_client):
187206 mock_client .list_limit_definitions .return_value = mock_response
188207
189208 mock_response = create_autospec (oci .response .Response )
190- mock_response .data = oci .limits .models .ResourceAvailability (used = 10 , available = 100 )
209+ mock_response .data = oci .limits .models .ResourceAvailability (
210+ used = 10 , available = 100
211+ )
191212 mock_client .get_resource_availability .return_value = mock_response
192213
193214 async with Client (mcp ) as client :
@@ -200,6 +221,6 @@ async def test_get_resource_availability_non_ad_scope(self, mock_get_client):
200221 },
201222 )
202223
203- assert len (result .structured_content [' result' ]) == 1
204- assert result .structured_content [' result' ][0 ]["used" ] == 10
205- assert result .structured_content [' result' ][0 ]["available" ] == 100
224+ assert len (result .structured_content [" result" ]) == 1
225+ assert result .structured_content [" result" ][0 ]["used" ] == 10
226+ assert result .structured_content [" result" ][0 ]["available" ] == 100
0 commit comments