diff --git a/test/workflows/iptrunk/test_create_iptrunk.py b/test/workflows/iptrunk/test_create_iptrunk.py index c2428cbff52bd3e2161b9902ffe73e5d83ecd4e4..e8aded0a1456dcaeb8d8088c6de8b65af046a913 100644 --- a/test/workflows/iptrunk/test_create_iptrunk.py +++ b/test/workflows/iptrunk/test_create_iptrunk.py @@ -44,18 +44,19 @@ def netbox_client_mock(): @pytest.fixture def input_form_wizard_data(request, router_subscription_factory, faker): + # only side b is set to JUNIPER vendor = getattr(request, 'param', RouterVendor.NOKIA) - router_side_a = router_subscription_factory(router_vendor=vendor) + router_side_a = router_subscription_factory() router_side_b = router_subscription_factory(router_vendor=vendor) - interface_members = None + side_a_members = [ + LAGMember(interface_name=f"Interface{interface}", interface_description=faker.sentence()) + for interface in range(5)] + side_b_members = None if vendor == RouterVendor.JUNIPER: - interface_members = ["et-1/0/0", "xe-1/0/0"] + side_b_members = [{"interface_name": "et-1/0/0", "interface_description": faker.sentence()}, {"interface_name": "xe-1/0/0", "interface_description": faker.sentence()}] # type: ignore[assignment] else: - interface_members = [ - LAGMember(interface_name=f"Interface{interface}", interface_description=faker.sentence()) - for interface in range(5) - ] + side_b_members = side_a_members create_ip_trunk_step = { "tt_number": faker.tt_number(), @@ -70,16 +71,13 @@ def input_form_wizard_data(request, router_subscription_factory, faker): create_ip_trunk_side_a_step = { "side_a_ae_iface": "LAG1", "side_a_ae_geant_a_sid": faker.geant_sid(), - "side_a_ae_members": interface_members, + "side_a_ae_members": side_a_members, } create_ip_trunk_side_b_router_name = {"side_b_node_id": router_side_b} create_ip_trunk_side_b_step = { "side_b_ae_iface": "LAG4", "side_b_ae_geant_a_sid": faker.geant_sid(), - "side_b_ae_members": [ - LAGMember(interface_name=f"Interface{interface}", interface_description=faker.sentence()) - for interface in range(5) - ], + "side_b_ae_members": side_b_members, } return [