Skip to content
Snippets Groups Projects
Commit 339face7 authored by Erik Reid's avatar Erik Reid
Browse files

Revert "pep8 & styling"

This reverts commit fadb41db.
parent 71f8ea58
No related branches found
No related tags found
No related merge requests found
# mypy: ignore-errors
from enum import Enum, auto
from typing import List
from typing import List, Set
from gso import settings
......@@ -23,8 +23,11 @@ def _dummy_router_interfaces(site_name) -> dict:
"physical": [{"name": f"ifc-{x}", "state": InterfaceAllocationState.AVAILABLE} for x in range(250)],
}
_DUMMY_SITES = {"site-a", "site-b", "site-c"}
_DUMMY_SITES = {
"site-a",
"site-b",
"site-c"
}
_DUMMY_INVENTORY = {
"fqdn-a": _dummy_router_interfaces("site-a"),
......@@ -32,7 +35,7 @@ _DUMMY_INVENTORY = {
"fqdn-c": _dummy_router_interfaces("site-b"),
"fqdn-d": _dummy_router_interfaces("site-b"),
"fqdn-e": _dummy_router_interfaces("site-c"),
"fqdn-f": _dummy_router_interfaces("site-c"),
"fqdn-f": _dummy_router_interfaces("site-c")
}
......@@ -42,7 +45,10 @@ def import_new_site(new_site_name: str, oss_params=settings.OSSParams):
_DUMMY_SITES.add(new_site_name)
def import_new_router(new_router_fqdn: str, site_name: str, oss_params=settings.OSSParams):
def import_new_router(
new_router_fqdn: str,
site_name: str,
oss_params=settings.OSSParams):
# TODO: this is a dummy implementation
# TODO: specify if this should be an error (and if not, what it means)
......
......@@ -19,18 +19,24 @@ def test_new_site():
def test_new_router():
site_name = _random_string(10)
resource_manager.import_new_site(new_site_name=site_name)
router_name = _random_string(10)
assert router_name not in resource_manager._DUMMY_INVENTORY
resource_manager.import_new_router(new_router_fqdn=router_name, site_name=site_name)
resource_manager.import_new_router(
new_router_fqdn=router_name,
site_name=site_name)
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) for _ in range(10)}
new_lags = {
resource_manager.next_lag(router_fqdn=router_name)
for _ in range(10)
}
assert len(new_lags) == 10
assert new_lags <= set(resource_manager._DUMMY_INVENTORY[router_name]["lags"])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment