diff --git a/gso/services/resource_manager.py b/gso/services/resource_manager.py index 61e359e300c3c6dbd168973fb2449571d84073fd..b9169a8b2959d42d632951c291978408ed197ae5 100644 --- a/gso/services/resource_manager.py +++ b/gso/services/resource_manager.py @@ -73,7 +73,9 @@ def _find_physical(router_fqdn: str, interface_name: str) -> dict: raise AssertionError(f"interface {interface_name} not found on {router_fqdn}") -def reserve_physical_interface(router_fqdn: str, interface_name: str, subscription_id: str, oss_params=settings.OSSParams): +def reserve_physical_interface( + router_fqdn: str, interface_name: str, subscription_id: str, oss_params=settings.OSSParams +): # TODO: this is a dummy implementation ifc = _find_physical(router_fqdn, interface_name) diff --git a/test/test_resource_manager.py b/test/test_resource_manager.py index d81b5cacd8f0fac341899568840034defbf3dcb2..34cf53506c94d37bc36cfbae31d6998cf88a7a30 100644 --- a/test/test_resource_manager.py +++ b/test/test_resource_manager.py @@ -14,46 +14,38 @@ def _random_string(n=None, letters=string.ascii_letters + string.digits + string def test_new_router(): router_name = _random_string(10) assert router_name not in resource_manager._DUMMY_INVENTORY - resource_manager.import_new_router( - new_router_fqdn=router_name, - subscription_id=_random_string(10)) + resource_manager.import_new_router(new_router_fqdn=router_name, subscription_id=_random_string(10)) assert router_name in resource_manager._DUMMY_INVENTORY def test_new_lag(): router_name = list(resource_manager._DUMMY_INVENTORY.keys())[0] new_lags = { - resource_manager.next_lag( - router_fqdn=router_name, subscription_id=_random_string(10)) - for _ in range(10)} + resource_manager.next_lag(router_fqdn=router_name, subscription_id=_random_string(10)) for _ in range(10) + } assert len(new_lags) == 10 - assert new_lags <= set( - resource_manager._DUMMY_INVENTORY[router_name]["lags"]) + assert new_lags <= set(resource_manager._DUMMY_INVENTORY[router_name]["lags"]) def test_physical_allocation_lifecycle_happy(): router_name = list(resource_manager._DUMMY_INVENTORY.keys())[0] def _interfaces(): - return resource_manager.available_physical_interfaces( - router_fqdn=router_name) + return resource_manager.available_physical_interfaces(router_fqdn=router_name) initial_available = _interfaces() interface_name = initial_available[0] resource_manager.reserve_physical_interface( - router_fqdn=router_name, - subscription_id=_random_string(10), - interface_name=interface_name) + router_fqdn=router_name, subscription_id=_random_string(10), interface_name=interface_name + ) current_available = _interfaces() assert interface_name not in current_available - resource_manager.allocate_physical_interface( - router_fqdn=router_name, interface_name=interface_name) - resource_manager.free_physical_interface( - router_fqdn=router_name, interface_name=interface_name) + resource_manager.allocate_physical_interface(router_fqdn=router_name, interface_name=interface_name) + resource_manager.free_physical_interface(router_fqdn=router_name, interface_name=interface_name) current_available = _interfaces() assert interface_name in current_available