suguo
/
dify
mirror of https://github.com/langgenius/dify.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor(tests): use enum values instead of string literals for status fields (#33441)

This commit is contained in:
Niels Kaspers 2026-03-14 09:14:34 +02:00 committed by GitHub
parent e64f4d6039
commit 79a8747c1b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/tests/test_containers_integration_tests/core/rag/pipeline/test_queue_integration.py
View File

@ -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()