Improve coverage
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user