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 core.rag.pipeline.queue import TaskWrapper, TenantIsolatedTaskQueue
from extensions.ext_redis import redis_client from extensions.ext_redis import redis_client
from models import Account, Tenant, TenantAccountJoin, TenantAccountRole from models import Account, AccountStatus, Tenant, TenantAccountJoin, TenantAccountRole, TenantStatus
@dataclass @dataclass
@ -47,7 +47,7 @@ class TestTenantIsolatedTaskQueueIntegration:
email=fake.email(), email=fake.email(),
name=fake.name(), name=fake.name(),
interface_language="en-US", interface_language="en-US",
status="active", status=AccountStatus.ACTIVE,
) )
db_session_with_containers.add(account) db_session_with_containers.add(account)
db_session_with_containers.commit() db_session_with_containers.commit()
@ -55,7 +55,7 @@ class TestTenantIsolatedTaskQueueIntegration:
# Create tenant # Create tenant
tenant = Tenant( tenant = Tenant(
name=fake.company(), name=fake.company(),
status="normal", status=TenantStatus.NORMAL,
) )
db_session_with_containers.add(tenant) db_session_with_containers.add(tenant)
db_session_with_containers.commit() db_session_with_containers.commit()
@ -101,7 +101,7 @@ class TestTenantIsolatedTaskQueueIntegration:
# Create second tenant # Create second tenant
tenant2 = Tenant( tenant2 = Tenant(
name=fake.company(), name=fake.company(),
status="normal", status=TenantStatus.NORMAL,
) )
db_session_with_containers.add(tenant2) db_session_with_containers.add(tenant2)
db_session_with_containers.commit() db_session_with_containers.commit()
@ -410,7 +410,7 @@ class TestTenantIsolatedTaskQueueCompatibility:
email=fake.email(), email=fake.email(),
name=fake.name(), name=fake.name(),
interface_language="en-US", interface_language="en-US",
status="active", status=AccountStatus.ACTIVE,
) )
db_session_with_containers.add(account) db_session_with_containers.add(account)
db_session_with_containers.commit() db_session_with_containers.commit()
@ -418,7 +418,7 @@ class TestTenantIsolatedTaskQueueCompatibility:
# Create tenant # Create tenant
tenant = Tenant( tenant = Tenant(
name=fake.company(), name=fake.company(),
status="normal", status=TenantStatus.NORMAL,
) )
db_session_with_containers.add(tenant) db_session_with_containers.add(tenant)
db_session_with_containers.commit() db_session_with_containers.commit()