refactor: replace hardcoded user plan strings with CloudPlan enum (#27675)

api/tests/test_containers_integration_tests/services/test_app_generate_service.py

@@ -5,6 +5,7 @@ import pytest
 from faker import Faker
 
 from core.app.entities.app_invoke_entities import InvokeFrom
+from enums.cloud_plan import CloudPlan
 from models.model import EndUser
 from models.workflow import Workflow
 from services.app_generate_service import AppGenerateService
@@ -32,7 +33,7 @@ class TestAppGenerateService:
             patch("services.app_generate_service.dify_config") as mock_dify_config,
         ):
             # Setup default mock returns for billing service
-            mock_billing_service.get_info.return_value = {"subscription": {"plan": "sandbox"}}
+            mock_billing_service.get_info.return_value = {"subscription": {"plan": CloudPlan.SANDBOX}}
 
             # Setup default mock returns for workflow service
             mock_workflow_service_instance = mock_workflow_service.return_value
@@ -430,7 +431,7 @@ class TestAppGenerateService:
 
         # Setup billing service mock for sandbox plan
         mock_external_service_dependencies["billing_service"].get_info.return_value = {
-            "subscription": {"plan": "sandbox"}
+            "subscription": {"plan": CloudPlan.SANDBOX}
         }
 
         # Set BILLING_ENABLED to True for this test
@@ -461,7 +462,7 @@ class TestAppGenerateService:
 
         # Setup billing service mock for sandbox plan
         mock_external_service_dependencies["billing_service"].get_info.return_value = {
-            "subscription": {"plan": "sandbox"}
+            "subscription": {"plan": CloudPlan.SANDBOX}
         }
 
         # Set BILLING_ENABLED to True for this test

api/tests/test_containers_integration_tests/services/test_feature_service.py

@@ -3,6 +3,7 @@ from unittest.mock import patch
 import pytest
 from faker import Faker
 
+from enums.cloud_plan import CloudPlan
 from services.feature_service import FeatureModel, FeatureService, KnowledgeRateLimitModel, SystemFeatureModel
 
 
@@ -173,7 +174,7 @@ class TestFeatureService:
             # Set mock return value inside the patch context
             mock_external_service_dependencies["billing_service"].get_info.return_value = {
                 "enabled": True,
-                "subscription": {"plan": "sandbox", "interval": "monthly", "education": False},
+                "subscription": {"plan": CloudPlan.SANDBOX, "interval": "monthly", "education": False},
                 "members": {"size": 1, "limit": 3},
                 "apps": {"size": 1, "limit": 5},
                 "vector_space": {"size": 1, "limit": 2},
@@ -189,7 +190,7 @@ class TestFeatureService:
             result = FeatureService.get_features(tenant_id)
 
         # Assert: Verify sandbox-specific limitations
-        assert result.billing.subscription.plan == "sandbox"
+        assert result.billing.subscription.plan == CloudPlan.SANDBOX
         assert result.education.activated is False
 
         # Verify sandbox limitations

api/tests/test_containers_integration_tests/tasks/test_document_indexing_task.py

@@ -3,6 +3,7 @@ from unittest.mock import MagicMock, patch
 import pytest
 from faker import Faker
 
+from enums.cloud_plan import CloudPlan
 from extensions.ext_database import db
 from models import Account, Tenant, TenantAccountJoin, TenantAccountRole
 from models.dataset import Dataset, Document
@@ -197,7 +198,7 @@ class TestDocumentIndexingTask:
         # Configure billing features
         mock_external_service_dependencies["features"].billing.enabled = billing_enabled
         if billing_enabled:
-            mock_external_service_dependencies["features"].billing.subscription.plan = "sandbox"
+            mock_external_service_dependencies["features"].billing.subscription.plan = CloudPlan.SANDBOX
             mock_external_service_dependencies["features"].vector_space.limit = 100
             mock_external_service_dependencies["features"].vector_space.size = 50
 
@@ -442,7 +443,7 @@ class TestDocumentIndexingTask:
         )
 
         # Configure sandbox plan with batch limit
-        mock_external_service_dependencies["features"].billing.subscription.plan = "sandbox"
+        mock_external_service_dependencies["features"].billing.subscription.plan = CloudPlan.SANDBOX
 
         # Create more documents than sandbox plan allows (limit is 1)
         fake = Faker()