]jGdZddlZddlmZmZddlmZddlZddlmZddl m Z ddl m Z ddl mZmZmZdd lmZmZdd lmZmZmZdd lmZmZmZdd lmZdd lmZddl m!Z!ddl"m#Z#m$Z$ddl%m&Z&e Z'ejPe)Z*ejVjXGddZ-ejVjXGddZ.y)z Label Studio Open Source FSM Integration Tests. Tests the core FSM functionality with real Django models in LSO, focusing on coverage of state_manager.py and utils modules. N)datetimetimezone)patchCurrentContext)get_user_model)cache)AnnotationStateChoicesProjectStateChoicesTaskStateChoices) StateManagerStateManagerError)AnnotationState ProjectState TaskState)get_current_state_safeis_fsm_enabledresolve_organization_id)OrganizationFactory)ProjectFactory) Annotation)AnnotationFactory TaskFactory) UserFactoryceZdZdZej ddZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZy)TestLSOFSMIntegrationz Test LSO FSM integration with real models. Focuses on improving coverage of state_manager.py and utils.py by testing error paths, cache behavior, and bulk operations. Tautousec#XKtjt|_t |_t j|j t j|jjdtjt jywzSet up test data.N r clearrorgruserrset_userset_organization_ididselfs U/root/env/lib/python3.12/site-packages/label_studio/fsm/tests/test_lso_integration.pysetup_test_dataz%TestLSOFSMIntegration.setup_test_data'd  &(M  ***488;;7  B(B*ct|j}tj|}|tj k(s Jd|t tjj|jd}t|dk(sJ|djtj k(sJ|djdk(sJy) z Test that creating a project automatically generates a state record. Validates: - Project model extends FsmHistoryStateModel - Automatic state transition on creation - State is CREATED for new projects  organizationExpected CREATED, got project created_atrproject_createdN)rr#r get_current_state_valuer CREATEDlistrobjectsfilterorder_bylenstatetransition_name)r)r3r>historys r*%test_project_creation_generates_statez;TestLSOFSMIntegration.test_project_creation_generates_state3s!dhh744W=+333U7MeW5UU3|++2272CLL\Z[7|q   qz#6#>#>>>>qz))->>>>ct|j}t|}tj|}|t j k(s Jd|ttjj|jd}t|dk(sJ|djt j k(sJy) z Test that creating a task automatically generates a state record. Validates: - Task model extends FsmHistoryStateModel - Automatic state transition on creation - State is CREATED for new tasks r/r2r1)taskr4r5rN)rr#rr r7r r8r9rr:r;r<r=r>)r)r3rDr>r@s r*"test_task_creation_generates_statez8TestLSOFSMIntegration.test_task_creation_generates_stateHs!dhh77+44T:(000R4J5'2RR0y((//T/:CCLQR7|q   qz#3#;#;;;;rBct|j}t|}t||j}t j |}|tjk(s Jd|ttjj|jd}t|dk\sJ|djtjk(sJy ) a  Test that creating an annotation automatically generates a state record. Validates: - Annotation model extends FsmHistoryStateModel - Automatic state transition on creation - State is CREATED for new annotations in LSO r/r2)rD completed_byr1) annotationr4r5rN)rr#rrr$r r7r r8r9rr:r;r<r=r>)r)r3rDrHr>r@s r*(test_annotation_creation_generates_statez>TestLSOFSMIntegration.test_annotation_creation_generates_state]s!dhh77+&DtyyI 44Z@.666X:PQVPW8XX6..555LUUVbcd7|q   qz#9#A#AAAArBct|j}tjt j |}|t jk(sJt j |}|t jk(sJ||k(sJy)z Test that StateManager caching works correctly. Validates: - State retrieval works consistently - Cache doesn't cause incorrect state returns - Multiple accesses return same state r/N)rr#r r"r r7r r8)r)r3state1state2s r*test_cache_functionalityz.TestLSOFSMIntegration.test_cache_functionalityssy!dhh7  55g>,4444455g>,44444rBct|j}tjj |j t jt|}|Jy)z Test get_current_state_safe returns None for entities without states. Validates: - Utility function handles entities with no state records - No exceptions raised - Returns None gracefully r/r2N) rr#rr:r;deleter r"r)r)r3r>s r*)test_get_current_state_safe_with_no_statez?TestLSOFSMIntegration.test_get_current_state_safe_with_no_statesQ!dhh7##G#4;;= 'w/}}rBct|j}t|}t|}||jjk(sJt|}||jjk(sJy)a Test resolve_organization_id utility function. Validates: - Extracts organization_id from entity with direct attribute - Extracts organization_id from entity with project relation - Falls back to CurrentContext r/r2N)rr#rrr')r)r3rDorg_ids r*(test_resolve_organization_id_from_entityz>TestLSOFSMIntegration.test_resolve_organization_id_from_entitys`!dhh77+)1$$$).$$$rBc<t|j}|dusJy)z Test is_fsm_enabled checks feature flag correctly. Validates: - Feature flag check works - Returns True when enabled r$TN)rr$)r)results r*test_is_fsm_enabled_in_lsoz0TestLSOFSMIntegration.test_is_fsm_enabled_in_lsos TYY/~~rBc@ddlm}|}||_d|j_d|j_d|_t jt5}tj|ddddtjvsJy#1swY#xYw)z Test StateManager error handling for invalid entities. Validates: - Proper error raised for entities without state model - Error includes helpful message rMocknonexistent_modelztest.nonexistentr5NzNo state model found) unittest.mockrZ_meta model_name label_lowerpkpytestraisesrr r7strvalue)r)rZ mock_entityexc_infos r*!test_state_manager_error_handlingz7TestLSOFSMIntegration.test_state_manager_error_handlings 'f  F ': $(: % ]], - >  0 0 = >&X^^)<<<< > >s BBc2t|j}tdDcgc]}t|}}t j t j||D],}t j|}|tjk(r,Jycc}w)z Test bulk cache warming for multiple entities. Validates: - Warm cache operation works for multiple entities - Subsequent state retrievals are faster (cached) - Correct states for all entities r/r2N) rr#rangerr r"r warm_cacher7r r8)r)r3_tasksrDr>s r*test_warm_cache_bulk_operationz4TestLSOFSMIntegration.test_warm_cache_bulk_operations!dhh77E,444 44 5AsBct|j}tj|}t |dk\sJ|dj t jk(sJ|djdk(sJy)z Test retrieving state history for an entity. Validates: - History retrieval works - States are in chronological order - All transition metadata captured r/r5rr6N) rr#r get_state_historyr=r>r r8r?r)r3r@s r*test_get_state_historyz,TestLSOFSMIntegration.test_get_state_historysp!dhh70097|q   qz#6#>#>>>>qz))->>>>rBct|j}tj|}t |dk\sJ|Dcgc]}|j }}|t |k(sJycc}w)z Test that state history is returned in chronological order. Validates: - History is ordered by creation time - Oldest states appear first - All state records are included r/r5N)rr#r rpr=r4sorted)r)r3r@h timestampss r*test_get_state_history_orderingz5TestLSOFSMIntegration.test_get_state_history_orderingsi!dhh70097|q   -44qall4 4VJ////5sA&ct|j}t|}t||jddigd}|j t j|}|tjk(sJy)z Test annotation created from draft has correct state. Validates: - Draft-based annotation workflow - Correct transition triggered - State is CREATED in LSO r/r2testdataF)rDrGrV was_cancelledN) rr#rrr$saver r7r r8)r)r3rDrHr>s r*#test_annotation_from_draft_workflowz9TestLSOFSMIntegration.test_annotation_from_draft_workflowss!dhh77+ V$%  44Z@.66666rBc\t|j}tj|d|_|j t tjj|jd}t|dk(sJ|djtjk(sJy)z Test that multiple state transitions are recorded correctly. Validates: - Multiple transitions create multiple records - History ordering is correct - Each transition has correct metadata r/rir2r4r5rN)rr#r r7maximum_annotationsr|r9rr:r;r<r=r>r r8rqs r*,test_state_manager_with_multiple_transitionszBTestLSOFSMIntegration.test_state_manager_with_multiple_transitions(s!dhh7 ,,W5'(# |++2272CLL\Z[7|q   qz#6#>#>>>>rBN)__name__ __module__ __qualname____doc__rafixturer+rArErIrMrPrSrWrgrnrrrwr}rrBr*rrshV^^D! " ?*<*B, .$%( =.5*?&0*72?rBrceZdZdZej ddZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZy)TestLSOFSMUtilitiesz` Test LSO FSM utility functions. Focuses on improving coverage of utils.py module. Trc#XKtjt|_t |_t j|j t j|jjdtjt jywr r!r(s r*r+z#TestLSOFSMUtilities.setup_test_dataKr,r-cft|j}t||j}|Jy)z Test resolve_organization_id with user parameter. Validates: - User parameter takes precedence - Correct organization extracted r/rUN)rr#rr$)r)r3rRs r*&test_resolve_organization_id_with_userz:TestLSOFSMUtilities.test_resolve_organization_id_with_userWs.!dhh7(tyyA!!!rBctj|jjddlm}|g}t |}||jjk(sJy)z Test resolve_organization_id falls back to CurrentContext. Validates: - CurrentContext used when no other source available - Correct ID returned rrY)specN)rr&r#r'r\rZr)r)rZrerRs r*0test_resolve_organization_id_fallback_to_contextzDTestLSOFSMUtilities.test_resolve_organization_id_fallback_to_contextdsH **488;;7 'm (5$$$rBct|j}tjt |}|J|t j k(sJy)z Test get_current_state_safe returns correct state value. Validates: - Function returns state string (not None) - State value is correct - Works correctly with database query r/N)rr#r r"rr r8)r)r3 state_values r*&test_get_current_state_safe_with_statez:TestLSOFSMUtilities.test_get_current_state_safe_with_statevsH!dhh7 -W5 &&&199999rBct|j}tdDcgc]}tj|}}t d|DsJycc}w)z Test StateManager handles concurrent access correctly. Validates: - No race conditions in state retrieval - Cache consistency - Multiple simultaneous requests work correctly r/ c3BK|]}|tjk(yw)N)r r8).0ss r* zSTestLSOFSMUtilities.test_state_manager_handles_concurrent_access..sD1+333DsN)rr#rjr r7all)r)r3rlstatess r*,test_state_manager_handles_concurrent_accessz@TestLSOFSMUtilities.test_state_manager_handles_concurrent_accesssT!dhh7JOrSA,66w?SSDVDDDDTsAct|j}tjtdd5t j |}|J dddy#1swYyxYw)z Test get_current_state_value returns None when FSM is disabled. Validates: - Returns None instead of raising when feature flag is off - Handles disabled state gracefully r/_is_fsm_enabledF return_valueN)rr#robjectr r7r)r3rVs r*.test_get_current_state_value_when_fsm_disabledzBTestLSOFSMUtilities.test_get_current_state_value_when_fsm_disabledsR!dhh7 \\,(9 N "!99'BF> !> " " "s AA c*ddlm}t|j}tjt j |dz }tjt j }tj|||}t|dk\sJy)z Test get_states_in_time_range for time-based queries. Validates: - UUID7-based time range queries work - Returns states within specified time range r) timedeltar/r5)daysN) rrrr#nowrutcr get_states_in_time_ranger=)r)rr3 start_timeend_timers r*test_get_states_in_time_rangez1TestLSOFSMUtilities.test_get_states_in_time_rangesm ' dhh7\\(,,/)2CC << -66w HU6{arBct|j}tj|}|tj k(sJtj |tj|}|tj k(sJy)z Test cache invalidation for entity state. Validates: - Cache is cleared for specific entity - Subsequent lookups hit database r/N)rr#r r7r r8invalidate_cache)r)r3r> state_afters r*test_invalidate_cachez)TestLSOFSMUtilities.test_invalidate_cachesp!dhh744W=+33333 %%g.#::7C 199999rBct|j}tj|}|J|jt j k(sJt|dsJt|dsJy)z Test get_current_state_object returns full state record. Validates: - Returns BaseState instance with full audit information - Contains all expected fields r/N triggered_byr?)rr#r get_current_state_objectr>r r8hasattr)r)r3 state_objects r*test_get_current_state_objectz1TestLSOFSMUtilities.test_get_current_state_objectsl!dhh7$< !>  " " "s *A''A0ctdDcgc]}t|j}}tj||D],}tj |}|t jk(r,Jycc}w)z Test warm_cache with multiple entities for bulk operations. Validates: - Cache is populated for all entities - Subsequent get_current_state_value calls are fast (from cache) r/N)rjrr#r rkr7r r8)r)rlprojectsr3r>s r*!test_warm_cache_multiple_entitiesz5TestLSOFSMUtilities.test_warm_cache_multiple_entitiessrDI8LaN9LL )  8G 88AE/777 77 8 MsA4cddlm}tdusJ|jdtdusJ|jdusJ|jdtdusJ|jdusJy)z Test CurrentContext.set_fsm_disabled() directly. Validates: - Can disable FSM via CurrentContext - is_fsm_enabled() respects the flag - State is properly restored rrTFN)core.current_requestrrset_fsm_disabledis_fsm_disabled)r)rs r*%test_fsm_disabled_via_current_contextz9TestLSOFSMUtilities.test_fsm_disabled_via_current_contexts 84''' (''-5(((-~--/4777 (''.4'''-~--/5888rBN)rrrrrarr+rrrrrrrrrrrrrBr*rrCs^ V^^D! "  "%$:"E" " *:*8$" 8$9rBr)/rloggingrrr\rrarrdjango.contrib.authrdjango.core.cacher fsm.state_choicesr r r fsm.state_managerr rfsm.state_modelsrrr fsm.utilsrrrorganizations.tests.factoriesrprojects.tests.factoriesr tasks.modelsrtasks.tests.factoriesrrusers.tests.factoriesrUser getLoggerrloggermark django_dbrrrrBr*rs' /.#[[=EEUU=3#@-   8 $a?a?a?H V9V9V9rB