From 95432b78c622fc1f0f360ee6240fe1f3941958e7 Mon Sep 17 00:00:00 2001 From: Erik Reid <erik.reid@geant.org> Date: Sun, 16 Jul 2023 10:14:28 +0200 Subject: [PATCH] added site test & param --- test/test_resource_manager.py | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/test/test_resource_manager.py b/test/test_resource_manager.py index 5258f4b8..c809617d 100644 --- a/test/test_resource_manager.py +++ b/test/test_resource_manager.py @@ -11,16 +11,32 @@ def _random_string(n=None, letters=string.ascii_letters + string.digits + string return "".join(random.choices(letters, k=n)) # noqa: S311 +def test_new_site(): + site_name = _random_string(10) + assert site_name not in resource_manager._DUMMY_SITES + resource_manager.import_new_site(new_site_name=site_name) + assert site_name in resource_manager._DUMMY_SITES + + 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) + 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"]) -- GitLab