feat: update tenant user status in periodic task

TencentBlueKing · Nov 28, 2024 · 7917deb · 7917deb
1 parent f9f77a5
commit 7917deb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/bk-user/tests/apps/tenant/conftest.py b/src/bk-user/tests/apps/tenant/conftest.py
@@ -36,7 +36,7 @@ def not_expired_tenant_user(bare_local_data_source, random_tenant):
     return TenantUser.objects.create(
         id=generate_random_string(),
         tenant=random_tenant,
-        data_source=data_source_user.data_source,
+        data_source=bare_local_data_source,
         data_source_user=data_source_user,
         status=TenantUserStatus.ENABLED,
         account_expired_at=timezone.now() + timedelta(days=1),
@@ -53,7 +53,7 @@ def expired_tenant_user(bare_local_data_source, random_tenant):
     return TenantUser.objects.create(
         id=generate_random_string(),
         tenant=random_tenant,
-        data_source=data_source_user.data_source,
+        data_source=bare_local_data_source,
         data_source_user=data_source_user,
         status=TenantUserStatus.ENABLED,
         account_expired_at=get_midnight() - timedelta(days=1),