diff --git a/test/conftest.py b/test/conftest.py index 001484ac71332882f7e118bc4c29dde118194fe1..ea1823988c8cba72ec7adbd565df289c99dbb73e 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -23,6 +23,7 @@ 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) @@ -72,6 +73,14 @@ class FakerProvider(BaseProvider): def network_interface(self) -> str: return self.generator.numerify("ge-@#/@#/@#") + def generate_junniper_members_list(self) -> list[LAGMember]: + iface_amount = self.generator.random_int(min=1, 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.sentence()) + for interface_name in interface_names] + + @pytest.fixture(scope="session") def faker() -> Faker: