Improve coverage

This commit is contained in:
2026-02-20 06:02:57 +00:00
parent f848bbf0f2
commit fd6b698077
28 changed files with 2719 additions and 387 deletions

View File

@@ -635,14 +635,16 @@ class TestErrorHandling:
async def mock_stream():
mock_entry = MagicMock()
mock_entry.reference = AsyncMock()
mock_reference = MagicMock()
mock_reference.delete = AsyncMock()
mock_entry.reference = mock_reference
yield mock_entry
mock_collection.stream.return_value = mock_stream()
mock_doc_ref = AsyncMock()
mock_doc_ref = MagicMock()
mock_doc_ref.collection.return_value = mock_collection
mock_doc_ref.delete.side_effect = Exception("Database error")
mock_doc_ref.delete = AsyncMock(side_effect=Exception("Database error"))
original_session_ref = clean_firestore._session_ref
clean_firestore._session_ref = MagicMock(return_value=mock_doc_ref)