test: Fix last failing tests

Author: KShivendu

tests/sessions/claims/test_get_claim_value.py

@@ -45,9 +45,7 @@ async def test_should_get_the_right_value_using_session_handle():
     assert isinstance(res, GetClaimValueOkResult)
     assert res.value is True
 
-import pytest
 
-@pytest.mark.skip
 async def test_should_work_for_non_existing_handle():
     new_st_init = {
         **st_init_common_args,
@@ -58,5 +56,5 @@ async def test_should_work_for_non_existing_handle():
     init(**new_st_init)  # type: ignore
     start_st()
 
-    res = await get_claim_value("non_existing_handle", TrueClaim)
+    res = await get_claim_value("non-existing-handle", TrueClaim)
     assert isinstance(res, SessionDoesNotExistError)

tests/test_session.py

@@ -202,7 +202,7 @@ async def test_creating_many_sessions_for_one_user_and_looping():
 
     assert len(session_handles) == 7
 
-    for i, handle in enumerate(session_handles):
+    for handle in session_handles:
         info = await get_session_information(handle)
         assert info is not None
         assert info.user_id == "someUser"
@@ -224,19 +224,22 @@ async def test_creating_many_sessions_for_one_user_and_looping():
         assert info.custom_claims_in_access_token_payload == {"someKey2": "someValue"}
         assert info.session_data_in_database == {"foo": "bar"}
 
+    regenerated_session_handles: List[str] = []
     # Regenerate access token with new access_token_payload
-    for i, token in enumerate(access_tokens):
+    for token in access_tokens:
         result = await regenerate_access_token(token, {"bar": "baz"})
         assert result is not None
-        assert (
-            result.session.handle == session_handles[i]
-        )  # Session handle should remain the same
+        regenerated_session_handles.append(result.session.handle)
 
         # Confirm that update worked:
         info = await get_session_information(result.session.handle)
         assert info is not None
         assert info.custom_claims_in_access_token_payload == {"bar": "baz"}
 
+    # Session handle should remain the same session handle should remain the same
+    # but order isn't guaranteed so we should sort them
+    assert sorted(regenerated_session_handles) == sorted(session_handles)
+
     # Try updating invalid handles:
     is_updated = await merge_into_access_token_payload("invalidHandle", {"foo": "bar"})
     assert is_updated is False