added site test & param

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"])