Update fixtures to allow for un-imported varieties

test/fixtures.py

@@ -23,11 +23,11 @@ from gso.products.product_blocks.iptrunk import (
 )
 from gso.products.product_blocks.router import RouterRole
 from gso.products.product_blocks.site import SiteTier
-from gso.products.product_types.iptrunk import IptrunkInactive
-from gso.products.product_types.office_router import OfficeRouterInactive
-from gso.products.product_types.router import Router, RouterInactive
-from gso.products.product_types.site import Site, SiteInactive
-from gso.products.product_types.super_pop_switch import SuperPopSwitchInactive
+from gso.products.product_types.iptrunk import IptrunkInactive, ImportedIptrunkInactive
+from gso.products.product_types.office_router import OfficeRouterInactive, ImportedOfficeRouterInactive
+from gso.products.product_types.router import Router, RouterInactive, ImportedRouterInactive
+from gso.products.product_types.site import ImportedSiteInactive, Site, SiteInactive
+from gso.products.product_types.super_pop_switch import SuperPopSwitchInactive, ImportedSuperPopSwitchInactive
 from gso.services import subscriptions
 from gso.utils.shared_enums import Vendor
 from test.workflows import WorkflowInstanceForTests
@@ -50,6 +50,7 @@ def site_subscription_factory(faker, geant_partner):
         site_ts_address=None,
         status: SubscriptionLifecycle | None = None,
         partner: dict | None = None,
+        is_imported: bool | None = True,
     ) -> UUIDstr:
         if partner is None:
             partner = geant_partner
@@ -65,8 +66,13 @@ def site_subscription_factory(faker, geant_partner):
         site_internal_id = site_internal_id or faker.pyint()
         site_ts_address = site_ts_address or faker.ipv4()
 
-        product_id = subscriptions.get_product_id_by_name(ProductName.SITE)
-        site_subscription = SiteInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+        if is_imported:
+            product_id = subscriptions.get_product_id_by_name(ProductName.SITE)
+            site_subscription = SiteInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+        else:
+            product_id = subscriptions.get_product_id_by_name(ProductName.IMPORTED_SITE)
+            site_subscription = ImportedSiteInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+
         site_subscription.site.site_city = site_city
         site_subscription.site.site_name = site_name
         site_subscription.site.site_country = site_country
@@ -107,6 +113,7 @@ def nokia_router_subscription_factory(site_subscription_factory, faker, geant_pa
         router_site=None,
         status: SubscriptionLifecycle | None = None,
         partner: dict | None = None,
+        is_imported: bool | None = True,
     ) -> UUIDstr:
         if partner is None:
             partner = geant_partner
@@ -120,8 +127,13 @@ def nokia_router_subscription_factory(site_subscription_factory, faker, geant_pa
         router_lo_iso_address = router_lo_iso_address or faker.word()
         router_site = router_site or site_subscription_factory()
 
-        product_id = subscriptions.get_product_id_by_name(ProductName.ROUTER)
-        router_subscription = RouterInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+        if is_imported:
+            product_id = subscriptions.get_product_id_by_name(ProductName.ROUTER)
+            router_subscription = RouterInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+        else:
+            product_id = subscriptions.get_product_id_by_name(ProductName.IMPORTED_ROUTER)
+            router_subscription = ImportedRouterInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+
         router_subscription.router.router_fqdn = router_fqdn
         router_subscription.router.router_ts_port = router_ts_port
         router_subscription.router.router_access_via_ts = router_access_via_ts
@@ -162,6 +174,7 @@ def juniper_router_subscription_factory(site_subscription_factory, faker, geant_
         router_site=None,
         status: SubscriptionLifecycle | None = None,
         partner: dict | None = None,
+        is_imported: bool | None = True,
     ) -> UUIDstr:
         if partner is None:
             partner = geant_partner
@@ -175,9 +188,13 @@ def juniper_router_subscription_factory(site_subscription_factory, faker, geant_
         router_lo_iso_address = router_lo_iso_address or faker.word()
         router_site = router_site or site_subscription_factory()
 
-        product_id = subscriptions.get_product_id_by_name(ProductName.ROUTER)
+        if is_imported:
+            product_id = subscriptions.get_product_id_by_name(ProductName.ROUTER)
+            router_subscription = RouterInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+        else:
+            product_id = subscriptions.get_product_id_by_name(ProductName.IMPORTED_ROUTER)
+            router_subscription = ImportedRouterInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
 
-        router_subscription = RouterInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
         router_subscription.router.router_fqdn = router_fqdn
         router_subscription.router.router_ts_port = router_ts_port
         router_subscription.router.router_access_via_ts = router_access_via_ts
@@ -256,13 +273,19 @@ def iptrunk_subscription_factory(iptrunk_side_subscription_factory, faker, geant
         iptrunk_sides=None,
         status: SubscriptionLifecycle | None = None,
         partner: dict | None = None,
+        is_imported: bool | None = True,
     ) -> UUIDstr:
         if partner is None:
             partner = geant_partner
 
-        product_id = subscriptions.get_product_id_by_name(ProductName.IP_TRUNK)
-        description = description or faker.sentence()
+        if is_imported:
+            product_id = subscriptions.get_product_id_by_name(ProductName.IP_TRUNK)
+            iptrunk_subscription = IptrunkInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+        else:
+            product_id = subscriptions.get_product_id_by_name(ProductName.IMPORTED_IP_TRUNK)
+            iptrunk_subscription = ImportedIptrunkInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
 
+        description = description or faker.sentence()
         geant_s_sid = geant_s_sid or faker.geant_sid()
         iptrunk_description = iptrunk_description or faker.sentence()
         iptrunk_isis_metric = iptrunk_isis_metric or faker.pyint()
@@ -273,9 +296,6 @@ def iptrunk_subscription_factory(iptrunk_side_subscription_factory, faker, geant
         iptrunk_side_b = iptrunk_side_subscription_factory()
         iptrunk_sides = iptrunk_sides or [iptrunk_side_a, iptrunk_side_b]
 
-        iptrunk_subscription = IptrunkInactive.from_product_id(
-            product_id, customer_id=partner["partner_id"], insync=True
-        )
         iptrunk_subscription.iptrunk.geant_s_sid = geant_s_sid
         iptrunk_subscription.iptrunk.iptrunk_description = iptrunk_description
         iptrunk_subscription.iptrunk.iptrunk_type = iptrunk_type
@@ -316,6 +336,7 @@ def office_router_subscription_factory(site_subscription_factory, faker, geant_p
         office_router_site=None,
         status: SubscriptionLifecycle | None = None,
         partner: dict | None = None,
+        is_imported: bool | None = True,
     ) -> UUIDstr:
         if partner is None:
             partner = geant_partner
@@ -327,10 +348,13 @@ def office_router_subscription_factory(site_subscription_factory, faker, geant_p
         office_router_lo_ipv6_address = office_router_lo_ipv6_address or ipaddress.IPv6Address(faker.ipv6())
         office_router_site = office_router_site or site_subscription_factory()
 
-        product_id = subscriptions.get_product_id_by_name(ProductName.OFFICE_ROUTER)
-        office_router_subscription = OfficeRouterInactive.from_product_id(
-            product_id, customer_id=partner["partner_id"], insync=True
-        )
+        if is_imported:
+            product_id = subscriptions.get_product_id_by_name(ProductName.OFFICE_ROUTER)
+            office_router_subscription = OfficeRouterInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+        else:
+            product_id = subscriptions.get_product_id_by_name(ProductName.IMPORTED_OFFICE_ROUTER)
+            office_router_subscription = ImportedOfficeRouterInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+
         office_router_subscription.office_router.office_router_fqdn = office_router_fqdn
         office_router_subscription.office_router.office_router_ts_port = office_router_ts_port
         office_router_subscription.office_router.office_router_lo_ipv4_address = office_router_lo_ipv4_address
@@ -366,6 +390,7 @@ def super_pop_switch_subscription_factory(site_subscription_factory, faker, gean
         super_pop_switch_site=None,
         status: SubscriptionLifecycle | None = None,
         partner: dict | None = None,
+        is_imported: bool | None = True,
     ) -> UUIDstr:
         if partner is None:
             partner = geant_partner
@@ -376,10 +401,13 @@ def super_pop_switch_subscription_factory(site_subscription_factory, faker, gean
         super_pop_switch_mgmt_ipv4_address = super_pop_switch_mgmt_ipv4_address or ipaddress.IPv4Address(faker.ipv4())
         super_pop_switch_site = super_pop_switch_site or site_subscription_factory()
 
-        product_id = subscriptions.get_product_id_by_name(ProductName.SUPER_POP_SWITCH)
-        super_pop_switch_subscription = SuperPopSwitchInactive.from_product_id(
-            product_id, customer_id=partner["partner_id"], insync=True
-        )
+        if is_imported:
+            product_id = subscriptions.get_product_id_by_name(ProductName.SUPER_POP_SWITCH)
+            super_pop_switch_subscription = SuperPopSwitchInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+        else:
+            product_id = subscriptions.get_product_id_by_name(ProductName.IMPORTED_SUPER_POP_SWITCH)
+            super_pop_switch_subscription = ImportedSuperPopSwitchInactive.from_product_id(product_id, customer_id=partner["partner_id"], insync=True)
+
         super_pop_switch_subscription.super_pop_switch.super_pop_switch_fqdn = super_pop_switch_fqdn
         super_pop_switch_subscription.super_pop_switch.super_pop_switch_ts_port = super_pop_switch_ts_port
         super_pop_switch_subscription.super_pop_switch.super_pop_switch_mgmt_ipv4_address = (