trying to mock firestore client

GoogleCloudPlatform · Jul 30, 2024 · fb0413c · fb0413c
1 parent 3ad96cb
commit fb0413c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -2,14 +2,13 @@
 from flask import Flask
 from flask.testing import FlaskClient, FlaskCliRunner
 from google.cloud import firestore
-from main import app as main_app
 
 
 @pytest.fixture()
 def app(mocker) -> Flask:
+    mocker.patch.object(firestore, "Client")
 
-    mocker.patch.object(firestore.Client, "__init__")
-
+    from main import app as main_app
     app = main_app
     app.config.update({
         "TESTING": True,