|
47 | 47 | UnspecifiedEvCharger, |
48 | 48 | UnspecifiedInverter, |
49 | 49 | VoltageTransformer, |
| 50 | + WindTurbine, |
50 | 51 | ) |
51 | 52 |
|
52 | 53 | # No client_args or client_kwargs needed for this call |
@@ -228,17 +229,21 @@ def assert_stub_method_call(stub_method: Any) -> None: |
228 | 229 | id=25, |
229 | 230 | category=ec_pb2.ELECTRICAL_COMPONENT_CATEGORY_POWER_TRANSFORMER, |
230 | 231 | ), |
231 | | - # Problematic components |
232 | 232 | ec_pb2.ElectricalComponent( |
233 | 233 | id=26, |
234 | | - category=ec_pb2.ELECTRICAL_COMPONENT_CATEGORY_UNSPECIFIED, |
| 234 | + category=ec_pb2.ELECTRICAL_COMPONENT_CATEGORY_WIND_TURBINE, |
235 | 235 | ), |
| 236 | + # Problematic components |
236 | 237 | ec_pb2.ElectricalComponent( |
237 | 238 | id=27, |
238 | | - category=999, # type: ignore[arg-type] |
| 239 | + category=ec_pb2.ELECTRICAL_COMPONENT_CATEGORY_UNSPECIFIED, |
239 | 240 | ), |
240 | 241 | ec_pb2.ElectricalComponent( |
241 | 242 | id=28, |
| 243 | + category=999, # type: ignore[arg-type] |
| 244 | + ), |
| 245 | + ec_pb2.ElectricalComponent( |
| 246 | + id=29, |
242 | 247 | category=ec_pb2.ELECTRICAL_COMPONENT_CATEGORY_BATTERY, |
243 | 248 | category_specific_info=ec_pb2.ElectricalComponentCategorySpecificInfo( |
244 | 249 | # Mismatched: battery category with inverter metadata |
@@ -292,13 +297,14 @@ def assert_client_result(result: Any) -> None: |
292 | 297 | primary_voltage=0.0, |
293 | 298 | secondary_voltage=0.0, |
294 | 299 | ), |
| 300 | + WindTurbine(id=ComponentId(26), microgrid_id=MicrogridId(0)), |
295 | 301 | # Problematic components |
296 | | - UnspecifiedComponent(id=ComponentId(26), microgrid_id=MicrogridId(0)), |
| 302 | + UnspecifiedComponent(id=ComponentId(27), microgrid_id=MicrogridId(0)), |
297 | 303 | UnrecognizedComponent( |
298 | | - id=ComponentId(27), microgrid_id=MicrogridId(0), category=999 |
| 304 | + id=ComponentId(28), microgrid_id=MicrogridId(0), category=999 |
299 | 305 | ), |
300 | 306 | MismatchedCategoryComponent( |
301 | | - id=ComponentId(28), |
| 307 | + id=ComponentId(29), |
302 | 308 | microgrid_id=MicrogridId(0), |
303 | 309 | category=ComponentCategory.BATTERY, |
304 | 310 | category_specific_metadata={ |
|
0 commit comments