Skip to content

Commit 9143fcd

Browse files
committed
Fix patch of Firestore client
1 parent bf58759 commit 9143fcd

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/test_ingest.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
import unittest
3434
from unittest.mock import patch
3535

36-
from gcp_mocks import mock_storage_client, mock_storage_blob
36+
from gcp_mocks import mock_storage_client, mock_storage_blob, mock_firestore_client
3737

3838
sys.path.append('../ingest')
3939
from ingest_pipeline import create_parser, validate_arguments, IngestPipeline
@@ -86,7 +86,10 @@ def get_nth_gene_models(n, models, mock_dir):
8686
class IngestTestCase(unittest.TestCase):
8787
@patch('google.cloud.storage.Blob', side_effect=mock_storage_blob)
8888
@patch('google.cloud.storage.Client', side_effect=mock_storage_client)
89-
def setup_ingest(self, args, mock_storage_client, mock_storage_blob):
89+
@patch("google.cloud.firestore.Client", side_effect=mock_firestore_client)
90+
def setup_ingest(
91+
self, args, mock_storage_client, mock_storage_blob, mock_firestore_client
92+
):
9093

9194
self.maxDiff = None
9295

0 commit comments

Comments
 (0)