Feature/nat 329 interface names should validated 3

test/conftest.py

@@ -17,12 +17,14 @@ from oauth2_lib.settings import oauth2lib_settings
 from orchestrator import app_settings
 from orchestrator.db import Database, db
 from orchestrator.db.database import ENGINE_ARGUMENTS, SESSION_ARGUMENTS, BaseModel
+from orchestrator.types import strEnum
 from sqlalchemy import create_engine, text
 from sqlalchemy.engine import make_url
 from sqlalchemy.orm import scoped_session, sessionmaker
 from starlette.testclient import TestClient
 
 from gso.main import init_gso_app
+from gso.utils.helpers import LAGMember
 
 logging.getLogger("faker.factory").setLevel(logging.WARNING)
 
@@ -33,6 +35,15 @@ def pytest_collection_modifyitems(config, items):
             item.add_marker(pytest.mark.skip(reason="Skipped due to SKIP_ALL_TESTS env variable"))
 
 
+class UseJuniperSide(strEnum):
+    """Define on tests on which side to use Juniper router"""
+
+    NONE = "none"
+    SIDE_A = "side_a"
+    SIDE_B = "side_b"
+    SIDE_BOTH = "side_both"
+
+
 class FakerProvider(BaseProvider):
     def ipv4_network(self):
         ipv4 = self.generator.ipv4()
@@ -72,6 +83,21 @@ class FakerProvider(BaseProvider):
     def network_interface(self) -> str:
         return self.generator.numerify("ge-@#/@#/@#")
 
+    def generate_juniper_members_list(self) -> list[LAGMember]:
+        iface_amount = self.generator.random_int(min=2, max=5)
+        interface_names = [f"{prefix}{i}" for prefix in ["xe-1/0/", "ge-3/0/", "xe-2/1/"] for i in range(iface_amount)]
+        return [
+            LAGMember(interface_name=interface_name, interface_description=self.generator.sentence())
+            for interface_name in interface_names
+        ]
+
+    def generate_nokia_members_list(self) -> list[LAGMember]:
+        iface_amount = self.generator.random_int(min=2, max=5)
+        return [
+            LAGMember(interface_name=f"Interface{i}", interface_description=self.generator.sentence())
+            for i in range(iface_amount)
+        ]
+
 
 @pytest.fixture(scope="session")
 def faker() -> Faker: