diff --git a/test/test_resource_manager.py b/test/test_resource_manager.py index 5258f4b80e7f12f85d0d194a7c2df3cbf40e2cc5..c809617d5f44d72465c0090c2ab4af298842d0e0 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"])