From 79a8747c1b9c218f66825f761037fea1ef3fa585 Mon Sep 17 00:00:00 2001
From: Niels Kaspers <153818647+nielskaspers@users.noreply.github.com>
Date: Sat, 14 Mar 2026 09:14:34 +0200
Subject: [PATCH] refactor(tests): use enum values instead of string literals
 for status fields (#33441)

---
 .../core/rag/pipeline/test_queue_integration.py      | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/api/tests/test_containers_integration_tests/core/rag/pipeline/test_queue_integration.py b/api/tests/test_containers_integration_tests/core/rag/pipeline/test_queue_integration.py
index cdf390b327..a60159c66a 100644
--- a/api/tests/test_containers_integration_tests/core/rag/pipeline/test_queue_integration.py
+++ b/api/tests/test_containers_integration_tests/core/rag/pipeline/test_queue_integration.py
@@ -18,7 +18,7 @@ from faker import Faker
 
 from core.rag.pipeline.queue import TaskWrapper, TenantIsolatedTaskQueue
 from extensions.ext_redis import redis_client
-from models import Account, Tenant, TenantAccountJoin, TenantAccountRole
+from models import Account, AccountStatus, Tenant, TenantAccountJoin, TenantAccountRole, TenantStatus
 
 
 @dataclass
@@ -47,7 +47,7 @@ class TestTenantIsolatedTaskQueueIntegration:
             email=fake.email(),
             name=fake.name(),
             interface_language="en-US",
-            status="active",
+            status=AccountStatus.ACTIVE,
         )
         db_session_with_containers.add(account)
         db_session_with_containers.commit()
@@ -55,7 +55,7 @@ class TestTenantIsolatedTaskQueueIntegration:
         # Create tenant
         tenant = Tenant(
             name=fake.company(),
-            status="normal",
+            status=TenantStatus.NORMAL,
         )
         db_session_with_containers.add(tenant)
         db_session_with_containers.commit()
@@ -101,7 +101,7 @@ class TestTenantIsolatedTaskQueueIntegration:
         # Create second tenant
         tenant2 = Tenant(
             name=fake.company(),
-            status="normal",
+            status=TenantStatus.NORMAL,
         )
         db_session_with_containers.add(tenant2)
         db_session_with_containers.commit()
@@ -410,7 +410,7 @@ class TestTenantIsolatedTaskQueueCompatibility:
             email=fake.email(),
             name=fake.name(),
             interface_language="en-US",
-            status="active",
+            status=AccountStatus.ACTIVE,
         )
         db_session_with_containers.add(account)
         db_session_with_containers.commit()
@@ -418,7 +418,7 @@ class TestTenantIsolatedTaskQueueCompatibility:
         # Create tenant
         tenant = Tenant(
             name=fake.company(),
-            status="normal",
+            status=TenantStatus.NORMAL,
         )
         db_session_with_containers.add(tenant)
         db_session_with_containers.commit()