chore: depend on flag

Author: pbastia

bc_obps/registration/tests/endpoints/test_regulated_products.py

@@ -1,11 +1,9 @@
-import pytest
 from registration.models import RegulatedProduct
 from registration.tests.utils.helpers import CommonTestSetup, TestUtils
 from registration.utils import custom_reverse_lazy
 
 
 class TestRegulatedProductsEndpoint(CommonTestSetup):
-    @pytest.mark.skip("Until the new pulp and paper product takes effect")
     def test_users_can_get_regulated_products(self):
         response = TestUtils.mock_get_with_auth_role(
             self, "industry_user", custom_reverse_lazy("list_regulated_products")

bc_obps/service/data_access_service/regulated_product_service.py

@@ -1,4 +1,5 @@
 from typing import Optional
+from django.conf import settings
 from registration.models import RegulatedProduct
 from registration.schema import RegulatedProductSchema
 from django.core.cache import cache
@@ -12,8 +13,11 @@ def get_regulated_products(cls) -> QuerySet[RegulatedProduct]:
         if cached_data:
             return cached_data
         else:
-            regulated_products = RegulatedProduct.objects.exclude(name="Pulp and paper: lime recovered by kiln").only(
-                *RegulatedProductSchema.Meta.fields
-            )
+            if settings.RESTRICTED_NAICS_CODES_FOR_REPORTING:
+                regulated_products = RegulatedProduct.objects.exclude(
+                    name="Pulp and paper: lime recovered by kiln"
+                ).only(*RegulatedProductSchema.Meta.fields)
+            else:
+                regulated_products = RegulatedProduct.objects.only(*RegulatedProductSchema.Meta.fields)
             cache.set("regulated_products", regulated_products, 60 * 60 * 24 * 1)  # 1 day
             return regulated_products