helix.fhir.client.sdk 4.2.14__py3-none-any.whl → 4.2.16__py3-none-any.whl

{helix_fhir_client_sdk-4.2.14.dist-info → helix_fhir_client_sdk-4.2.16.dist-info}/RECORD

@@ -1,14 +1,14 @@
 helix_fhir_client_sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 helix_fhir_client_sdk/dictionary_parser.py,sha256=WrGkVAxMlUvVycRVrX7UZt2oP2e_Vk4-E8QibwTpHLM,3401
 helix_fhir_client_sdk/dictionary_writer.py,sha256=V7Bx9Z69s0LRYF6Lc6Xp0d-Gj0BnAVKA1vBuwf3JORE,1486
-helix_fhir_client_sdk/fhir_auth_mixin.py,sha256=p2QIYrCv7ZktutY7SzYizmHADahxBddRTSCqX1EKyHc,14465
+helix_fhir_client_sdk/fhir_auth_mixin.py,sha256=L-_fwQbyojv6AoRUYkPSXWEsW7NRInvJ3wwYgINEtJs,14351
 helix_fhir_client_sdk/fhir_bundle_appender.py,sha256=t1hs7p_vXKC9MUFyUnN9dTuDhRF-kw-kkgVFtGHv9QQ,11749
-helix_fhir_client_sdk/fhir_client.py,sha256=P8xhTVNnJuo5UGbg5nrFOlAR4DAOlEstH27YtFHub00,35201
-helix_fhir_client_sdk/fhir_delete_mixin.py,sha256=1YiKddTJTUzzrRvG7WgSisXY8rfJAHPAEXTsOevrom8,6521
-helix_fhir_client_sdk/fhir_merge_mixin.py,sha256=YTUODvc2rqXhk5_kwNMFPykm1I9_omww2WUcDExMESg,15264
-helix_fhir_client_sdk/fhir_merge_resources_mixin.py,sha256=d5x41uL__yxMcaAnYDdmbpt529ZsOZcO0EBvnV1HSGI,36002
-helix_fhir_client_sdk/fhir_patch_mixin.py,sha256=YGcCPStoqVxWCYOaNI8vCKSFkAyRzP0YM_UWp4UKYro,5858
-helix_fhir_client_sdk/fhir_update_mixin.py,sha256=7psQTBsGPY1izuwn3yD4MGLjLVWQjqA_15_IeaUspew,6230
+helix_fhir_client_sdk/fhir_client.py,sha256=dHfkruJZ1uPM4cmKcnn4HqI0BCDP3IwIqXSLvCu4IsY,37346
+helix_fhir_client_sdk/fhir_delete_mixin.py,sha256=IRcJ5AJ7yrsc1HbjqYW5-jAcIEwG__9nEqwbvvC8qJ0,7532
+helix_fhir_client_sdk/fhir_merge_mixin.py,sha256=h2lvpLDs5nvt9F1bcpd110LFuFXI2HW-uKrUs5TbW7I,17783
+helix_fhir_client_sdk/fhir_merge_resources_mixin.py,sha256=QO2SeZa4Co69f_2YCz0_ss2dJb4xhVMkRdzbjLIJ-iI,35799
+helix_fhir_client_sdk/fhir_patch_mixin.py,sha256=QLTsqhFLGi4gC_qQblnCA4skdiLw8h4QRlzvaeyFkl4,7249
+helix_fhir_client_sdk/fhir_update_mixin.py,sha256=_Yx9yg809N8EXY07XlwHxSJLp-xIzpNg5erL92LOeww,7370
 helix_fhir_client_sdk/function_types.py,sha256=x95j6ix3Xa9b276Q741xX1jguqBuFT6EBLDw35_EoVM,3916
 helix_fhir_client_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 helix_fhir_client_sdk/well_known_configuration.py,sha256=hwKpqZoJHkHuCEOowoXk07ywEMMhr_rcmQHNKCUEgVk,221
@@ -38,13 +38,13 @@ helix_fhir_client_sdk/graph/test/test_graph_mixin.py,sha256=LNd4LVjryVLgzWeTXMDp
 helix_fhir_client_sdk/graph/test/test_simulate_graph_processor_mixin.py,sha256=EQDfhqJfUrP6SptXRP7ayEN7g5cZQMA00ccXzeXiSXM,46312
 helix_fhir_client_sdk/graph/test/test_simulate_graph_processor_mixin_caching.py,sha256=WFzKKHtKDcOOXjXRMKkRyJ64whoJoo9M9ST88ayvEbY,16176
 helix_fhir_client_sdk/open_telemetry/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-helix_fhir_client_sdk/open_telemetry/attribute_names.py,sha256=iUxm82t5-_qONL3Ld6LkzOwjuuCwcOptaJVBgCE4tps,169
-helix_fhir_client_sdk/open_telemetry/span_names.py,sha256=Y-k29sniK5beSgkEz335eYFpybaw3K_3VU-ayxLdrW0,380
+helix_fhir_client_sdk/open_telemetry/attribute_names.py,sha256=mcPcgpaRe-hZDmPu8gLQo51E_-rllAk2OXMC9uK6EmM,328
+helix_fhir_client_sdk/open_telemetry/span_names.py,sha256=sEuzUXxE9pSoAZti2YVifBqbo3r4SLTPlIUW4F2EuP0,548
 helix_fhir_client_sdk/queue/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-helix_fhir_client_sdk/queue/request_queue_mixin.py,sha256=L5HgWWARj-JoM7v4cradNonDK1aO1BQyl2MWZknP_PY,21806
+helix_fhir_client_sdk/queue/request_queue_mixin.py,sha256=Q5ZyadT2nMX6TATsiy0FLXzzCpTbVjN0Gh-B_F0RGCk,21684
 helix_fhir_client_sdk/responses/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 helix_fhir_client_sdk/responses/bundle_expander.py,sha256=ilR5eMgciSgzsdQvKB6bHtn9jtpvn3uS-EBz-hrahzo,1065
-helix_fhir_client_sdk/responses/fhir_client_protocol.py,sha256=QbkKa3x6czhy-QCLN1icctBSXBlLnll1Lj9kMdFm914,7002
+helix_fhir_client_sdk/responses/fhir_client_protocol.py,sha256=IWM7LNQ1ZbgaySLGqpCwWAKyArT5HgBdNhkmSEivNMo,7100
 helix_fhir_client_sdk/responses/fhir_delete_response.py,sha256=0K11vyfZ0LtL-G61NHzDqHrZgEHjMVZr00VWpWcktZA,2941
 helix_fhir_client_sdk/responses/fhir_get_response.py,sha256=3PXvFoMZ7ix2ZzucIjY-49RL3foLR07dD57BdulyMGI,17657
 helix_fhir_client_sdk/responses/fhir_merge_response.py,sha256=uvUjEGJgMDlAkcc5_LjgAsTFZqREQMlV79sbxUZwtwE,2862
@@ -104,7 +104,7 @@ helix_fhir_client_sdk/utilities/hash_util.py,sha256=YNUy7-IC_OtC0l-T45UO9UkA-_ps
 helix_fhir_client_sdk/utilities/list_chunker.py,sha256=2h2k5CCFmOhICaugOx6UI-9dh4q5w1lVdF7WQLX0LCM,1456
 helix_fhir_client_sdk/utilities/ndjson_chunk_streaming_parser.py,sha256=3TCYfWVCEpJbqRxqlSDsGnFnraO4T9bxzYdShvu6Pos,1954
 helix_fhir_client_sdk/utilities/practitioner_generator.py,sha256=gneCAXNDNEphBY-Nc2nMQBbEWJgHcjvv3S8JQ75yiJI,3778
-helix_fhir_client_sdk/utilities/retryable_aiohttp_client.py,sha256=yjZ0hUWNG4JD8sWgPGXYmYsY_wzaJM0cWD_Vo2LG8Sg,21095
+helix_fhir_client_sdk/utilities/retryable_aiohttp_client.py,sha256=8DdzqTmIY1hWlxmEXOxycBpa72wTo-vsAtcLO4hpnmQ,22605
 helix_fhir_client_sdk/utilities/retryable_aiohttp_response.py,sha256=DvNX6WO1m2Hz6LoI5CwSPDECPd8oDsqRCVsyq_Oxf-0,3542
 helix_fhir_client_sdk/utilities/retryable_aiohttp_url_result.py,sha256=Gdmvn6qIM2JF0YOhobQUHY41fCxvYyaths_CZs0iJfo,616
 helix_fhir_client_sdk/utilities/url_checker.py,sha256=_JRSIvu7WNXh2OA79HJbEEiomGT-quGhAUGh44-9824,3580
@@ -130,10 +130,10 @@ helix_fhir_client_sdk/validators/async_fhir_validator.py,sha256=Bgiw5atbc5YzBYpk
 helix_fhir_client_sdk/validators/fhir_validator.py,sha256=HWBldSEB9yeKIcnLcV8R-LoTzwT_OMu8SchtUUBKzys,2331
 helix_fhir_client_sdk/validators/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 helix_fhir_client_sdk/validators/test/test_async_fhir_validator.py,sha256=RmSowjPUdZee5nYuYujghxWyqJ20cu7U0lJFtFT-ZBs,3285
-helix_fhir_client_sdk-4.2.14.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+helix_fhir_client_sdk-4.2.16.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/logger_for_test.py,sha256=UC-7F6w6fDsUIYf37aRnvUdiUUVk8qkJEUSuO17NQnI,1525
-tests/test_fhir_client_clone.py,sha256=tKL-Kz9sUc1mTkaGO0QQtgO5_8ZuBdDICm_oNTV12sY,4030
+tests/test_fhir_client_clone.py,sha256=c5y1rWJ32nBSUnK1FfyymY005dNowd4Nf1xrbuQolNk,5368
 tests/test_get_nested_property.py,sha256=dA7eNmPJuwzQTViORRmJkcn9RAZzxeajSxUghobHpAo,2381
 tests/async/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/async/test_async_fhir_client_fetch_response_in_chunks.py,sha256=Ql95DuOqaidFL6u553f-Bkp4Rsd662C__NA-5sw0KL0,3993
@@ -153,6 +153,7 @@ tests/async/test_async_real_fhir_server_get_patients.py,sha256=0oMnUJg1KEspJ5_4e
 tests/async/test_async_real_fhir_server_get_patients_error.py,sha256=_s7chLogAg0yKgGpsq1o9_dDHBrzGaRWBAo8agFTN6U,1914
 tests/async/test_benchmark_compress.py,sha256=q1gDG7qXvof-3uVAqJlZAW7uO8cR0vEeDfzl-iwIEtY,16470
 tests/async/test_benchmark_merge.py,sha256=ME0Pow_IXpIaVGWvq3ii7dGltXcz-3DGxz2gGF4LmYQ,19830
+tests/async/test_retryable_client_session_management.py,sha256=cOAE0wGkh3cv0AS187nujeial3gGEu1VlOJ5b9-LaCI,5360
 tests/async/fhir_server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/async/fhir_server/test_async_real_fhir_server_get_graph_large.py,sha256=fM2MPF47nDF2Qwj2AkdTZ2CfvgUYGN4AVIS253KC9MQ,9430
 tests/async/fhir_server/test_async_real_fhir_server_get_patients_large.py,sha256=rXRF8E8Al7XANCmef1d_WqxSA9TVQjVC7B41OZaEQlY,5583
@@ -212,7 +213,7 @@ tests_integration/test_emr_server_auth.py,sha256=2I4QUAspQN89uGf6JB2aVuYaBeDnRJz
 tests_integration/test_firely_fhir.py,sha256=ll6-plwQrKfdrEyfbw0wLTC1jB-Qei1Mj-81tYTl5eQ,697
 tests_integration/test_merge_vs_smart_merge_behavior.py,sha256=LrIuyxzw0YLaTjcRtG0jzy0M6xSv9qebmdBtMPDcacQ,3733
 tests_integration/test_staging_server_graph.py,sha256=5RfMxjhdX9o4-n_ZRvze4Sm8u8NjRijRLDpqiz8qD_0,7132
-helix_fhir_client_sdk-4.2.14.dist-info/METADATA,sha256=k5OlcbCO8MQ7ButJErmJBVXEV7nOSZ9oDHlXxomX_a4,5845
-helix_fhir_client_sdk-4.2.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-helix_fhir_client_sdk-4.2.14.dist-info/top_level.txt,sha256=BRnDS6ceQxs-4u2jXznATObgP8G2cGAerlH0ZS4sJ6M,46
-helix_fhir_client_sdk-4.2.14.dist-info/RECORD,,
+helix_fhir_client_sdk-4.2.16.dist-info/METADATA,sha256=AHcaKMqC3ICIbjD_Fr8CXs8MxOXLzdyh3A76J-SVYMk,7210
+helix_fhir_client_sdk-4.2.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+helix_fhir_client_sdk-4.2.16.dist-info/top_level.txt,sha256=BRnDS6ceQxs-4u2jXznATObgP8G2cGAerlH0ZS4sJ6M,46
+helix_fhir_client_sdk-4.2.16.dist-info/RECORD,,

tests/async/test_retryable_client_session_management.py

@@ -0,0 +1,155 @@
+"""
+Tests for RetryableAioHttpClient session lifecycle management.
+
+This module tests that sessions are properly closed or kept open depending on
+whether they were created internally or provided by the user.
+"""
+
+import aiohttp
+import pytest
+
+from helix_fhir_client_sdk.utilities.retryable_aiohttp_client import RetryableAioHttpClient
+
+
+@pytest.mark.asyncio
+async def test_internal_session_is_closed_after_exit() -> None:
+    """Test that internally created sessions are closed when context exits"""
+    client = RetryableAioHttpClient(
+        retries=1,
+        refresh_token_func=None,
+        tracer_request_func=None,
+        fn_get_session=None,  # No custom factory - SDK will create session
+        use_data_streaming=False,
+        access_token=None,
+        access_token_expiry_date=None,
+    )
+
+    async with client:
+        # Session should be created
+        assert client.session is not None
+        assert not client.session.closed
+        session_ref = client.session
+
+    # After exiting context, the internal session should be closed
+    assert session_ref.closed
+
+
+@pytest.mark.asyncio
+async def test_user_provided_session_is_not_closed_after_exit() -> None:
+    """Test that user-provided sessions are NOT closed when context exits"""
+    # User creates their own session
+    user_session = aiohttp.ClientSession()
+
+    try:
+        # Provide a factory that returns the user's session
+        # Since fn_get_session is provided, SDK will NOT close the session
+        client = RetryableAioHttpClient(
+            retries=1,
+            refresh_token_func=None,
+            tracer_request_func=None,
+            fn_get_session=lambda: user_session,  # User provides a custom factory
+            use_data_streaming=False,
+            access_token=None,
+            access_token_expiry_date=None,
+        )
+
+        async with client:
+            # Session should be the user's session
+            assert client.session is user_session
+            assert not client.session.closed
+
+        # After exiting context, the user's session should still be open
+        # because fn_get_session was provided (caller manages session lifecycle)
+        assert not user_session.closed
+
+    finally:
+        # User closes their own session
+        await user_session.close()
+        assert user_session.closed
+
+
+@pytest.mark.asyncio
+async def test_multiple_clients_can_share_user_session() -> None:
+    """Test that multiple RetryableAioHttpClient instances can share the same user session"""
+    # User creates a persistent session
+    shared_session = aiohttp.ClientSession()
+
+    try:
+        # Multiple clients share the same session
+        async with RetryableAioHttpClient(
+            retries=1,
+            refresh_token_func=None,
+            tracer_request_func=None,
+            fn_get_session=lambda: shared_session,
+            use_data_streaming=False,
+            access_token=None,
+            access_token_expiry_date=None,
+        ) as client1:
+            assert client1.session is shared_session
+            assert not shared_session.closed
+
+        # Session should still be open after the first client exits
+        assert not shared_session.closed
+
+        # The second client can reuse the same session
+        async with RetryableAioHttpClient(
+            retries=1,
+            refresh_token_func=None,
+            tracer_request_func=None,
+            fn_get_session=lambda: shared_session,
+            use_data_streaming=False,
+            access_token=None,
+            access_token_expiry_date=None,
+        ) as client2:
+            assert client2.session is shared_session
+            assert not shared_session.closed
+
+        # Session should still be open after the second client exits
+        assert not shared_session.closed
+
+    finally:
+        # User closes the shared session when done
+        await shared_session.close()
+        assert shared_session.closed
+
+
+@pytest.mark.asyncio
+async def test_user_can_recreate_closed_session_via_factory() -> None:
+    """Test that a user's factory can be called multiple times if session gets closed"""
+    call_count = 0
+
+    def session_factory() -> aiohttp.ClientSession:
+        nonlocal call_count
+        call_count += 1
+        return aiohttp.ClientSession()
+
+    created_sessions = []
+
+    try:
+        # First client call
+        async with RetryableAioHttpClient(
+            retries=1,
+            refresh_token_func=None,
+            tracer_request_func=None,
+            fn_get_session=session_factory,
+            use_data_streaming=False,
+            access_token=None,
+            access_token_expiry_date=None,
+        ) as client1:
+            assert client1.session is not None
+            created_sessions.append(client1.session)
+            assert call_count == 1  # Factory called once in __aenter__
+
+        # SDK doesn't close session (caller provided fn_get_session)
+        assert created_sessions[0] is not None
+        assert not created_sessions[0].closed
+
+        # User could manually close and recreate via factory if needed
+        # (This demonstrates the pattern, though in practice the factory
+        # would handle closed session detection)
+
+    finally:
+        # Clean up all created sessions
+        for session in created_sessions:
+            if session is not None and not session.closed:
+                await session.close()

tests/test_fhir_client_clone.py

@@ -110,3 +110,46 @@ def test_clone_preserves_max_concurrent_requests() -> None:
     cloned_client = fhir_client.clone()
 
     assert cloned_client._max_concurrent_requests == 10
+
+
+def test_use_http_session() -> None:
+    """Test that use_http_session sets the callable correctly"""
+    from aiohttp import ClientSession
+
+    def custom_session_factory() -> ClientSession:
+        return ClientSession()
+
+    fhir_client = FhirClient().url("http://example.com")
+
+    # Initially None
+    assert fhir_client._fn_create_http_session is None
+
+    # Set the callable
+    result = fhir_client.use_http_session(custom_session_factory)
+
+    # Returns self for chaining
+    assert result is fhir_client
+
+    # Callable is stored
+    assert fhir_client._fn_create_http_session is custom_session_factory
+
+    # Can set to None
+    fhir_client.use_http_session(None)
+    assert fhir_client._fn_create_http_session is None
+
+
+def test_clone_preserves_fn_create_http_session() -> None:
+    """Test that clone() preserves the _fn_create_http_session callable"""
+    from aiohttp import ClientSession
+
+    def custom_session_factory() -> ClientSession:
+        return ClientSession()
+
+    fhir_client = FhirClient().url("http://example.com").use_http_session(custom_session_factory)
+
+    assert fhir_client._fn_create_http_session is custom_session_factory
+
+    # Clone and verify fn_create_http_session is preserved
+    cloned_client = fhir_client.clone()
+
+    assert cloned_client._fn_create_http_session is custom_session_factory