]j[ R d Z ddlmZmZ ddlmZ ddlmZ ddlm Z G d de Z
y) zTest data_manager.api module functionality.
This file tests the TaskPagination class optimizations that prevent
loading heavy task.data fields during pagination.
) MagicMockpatch)TaskPagination)TestCase)PageNumberPaginationc ( e Zd ZdZd Zd Zd Zd Zy)$TestTaskPaginationMemoryOptimizationa Test TaskPagination uses .only('id') to avoid loading heavy fields.
The optimization prevents loading full task objects (with potentially
multi-megabyte data fields) during pagination. Only task IDs are loaded;
full task objects are loaded later with proper annotations.
Critical validation: The queryset passed to the parent paginate_queryset
must have .only('id') applied to defer loading of heavy fields.
c h t | _ t | _ ddd| j _ y)zSet up test fixtures.130)page page_sizeN)r
paginationr mock_request query_params)selfs R/root/env/lib/python3.12/site-packages/label_studio/data_manager/tests/test_api.pysetUpz*TestTaskPaginationMemoryOptimization.setUp s) (* %K25D)I& c t }t }||j _ t d 5 }t d 5 }d|j j
j j _ d|j j
j j _ t j t dg 5 }| j j || j |j j d |j |j }|d d |u sJ d d d d d d d d d y # 1 sw Y xY w# 1 sw Y xY w# 1 sw Y y xY w)
a5 Test sync_paginate_queryset applies .only('id') optimization.
This test validates:
- The queryset passed to parent's paginate_queryset has .only('id') applied
- Count queries still work with the original queryset
- The optimization is always applied (no feature flag)
data_manager.api.Predictiondata_manager.api.Annotation
paginate_queryset return_valueidr N)r onlyr r objectsfiltercountobjectr r sync_paginate_querysetr assert_called_once_withassert_called_once call_args)r
mock_querysetmock_id_only_querysetmock_predictionmock_annotationmock_parent_paginater' s r (test_sync_paginate_queryset_uses_only_idzMTestTaskPaginationMemoryOptimization.test_sync_paginate_queryset_uses_only_id s@ "
) *?
'
0
1 D_45
D QS''..;;AANQR''..;;AAN \\"68KZ\] DauOO::=$J[J[\ "&&>>tD );;= 4 > >I$Q<?.CCCC D
D D D D D
D
D D Ds< EA;E8A*D<#E+E<E EE
EE c t }t }||j _ ddd|j j j _ t j t dg 5 }| j j || j |j j d |j |j }|d d |u sJ ddd y# 1 sw Y yxY w) a Test paginate_totals_queryset applies .only('id') optimization.
This test validates:
- The optimized path (with fflag_fix_back_optic_1407_optimize_tasks_api_pagination_counts)
also uses .only('id')
- Aggregate queries work correctly
r r )total_annotationstotal_predictionsr r r r N)r r r values aggregater r# r r paginate_totals_querysetr r% r& r' )r r( r) r, r' s r *test_paginate_totals_queryset_uses_only_idzOTestTaskPaginationMemoryOptimization.test_paginate_totals_queryset_uses_only_id< s "
) *?
'!#!"D
))33@
\\.0CRT
U /F/F/M/M/W/WXY/Z,3MBmSSS#++22EEG/>/F/F/M/M/W/WXY/Z,3MBmSSS
T
T T T
T
T
T
T T Ts< F1A;F%8CF F%F1F" F%%F. *F11F:N)__name__
__module__ __qualname____doc__r r- r4 r: r r r r
s J D<<6Tr r N) r>
unittest.mockr r data_manager.apir django.testr rest_framework.paginationr r r? r r rD s(
+ + :eT8 eTr