fix: Increase tests coverage

Author: gencurrent

tests/mongo_tests/test_async_mongo_core.py

@@ -106,6 +106,12 @@ async def test_async_mongo_core_collection_resolution_and_index_branches():
     core._index_verified = True
     assert await core._ensure_collection_async() is collection
 
+    # Reset _index_verified to force re-entry into the index-check block.
+    # The index already exists in the collection → covers the False branch at line 94.
+    core._index_verified = False
+    assert await core._ensure_collection_async() is collection
+    assert core._index_verified is True
+
 
 @pytest.mark.mongo
 @pytest.mark.asyncio

tests/mongo_tests/test_mongo_core.py

@@ -341,6 +341,37 @@ def _params_with_dataframe(*args, **kwargs):
     assert value_a == value_b  # same content --> same key
 
 
+@pytest.mark.mongo
+def test_mongo_core_ensure_collection_collection_set_not_verified():
+    """Cover line 69 False branch: collection already set when entering lock."""
+    core = _MongoCore(hash_func=None, mongetter=_test_mongetter, wait_for_calc_timeout=10)
+    core.set_func(lambda x: x)
+
+    # Set mongo_collection manually so it is not None, but leave _index_verified as False.
+    # Outer guard: mongo_collection is not None AND _index_verified is False → enters lock.
+    # Line 69: False (collection already set) → skips fetching collection.
+    # Line 72: True (not verified) → verifies index.
+    core.mongo_collection = _test_mongetter()
+    collection = core._ensure_collection()
+    assert collection is not None
+    assert core._index_verified is True
+
+
+@pytest.mark.mongo
+def test_mongo_core_ensure_collection_index_verified_inside_lock():
+    """Cover line 72 False branch: _index_verified is True when inside lock."""
+    core = _MongoCore(hash_func=None, mongetter=_test_mongetter, wait_for_calc_timeout=10)
+    core.set_func(lambda x: x)
+
+    # Set _index_verified = True, but leave mongo_collection as None.
+    # Outer guard: mongo_collection is None → enters lock.
+    # Line 69: True (collection is None) → fetches collection.
+    # Line 72: False (_index_verified is already True) → skips index check.
+    core._index_verified = True
+    collection = core._ensure_collection()
+    assert collection is not None
+
+
 @pytest.mark.mongo
 def test_mongo_core_set_entry_should_not_store():
     core = _MongoCore(hash_func=None, mongetter=_test_mongetter, wait_for_calc_timeout=10)