diff --git a/test/test_resource_manager.py b/test/test_resource_manager.py index c809617d5f44d72465c0090c2ab4af298842d0e0..5258f4b80e7f12f85d0d194a7c2df3cbf40e2cc5 100644 --- a/test/test_resource_manager.py +++ b/test/test_resource_manager.py @@ -11,32 +11,16 @@ 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, - site_name=site_name) + resource_manager.import_new_router(new_router_fqdn=router_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"])