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