fix: update deprecated methods (GoogleCloudPlatform#424)

dandhlee · web-flow · commit 63f1fa389be3 · 2022-01-23T06:12:46.000-05:00
diff --git a/background/app/main_test.py b/background/app/main_test.py
@@ -30,7 +30,7 @@
 TOPIC_NAME = "projects/{}/topics/{}".format(project_id, "translate")
 
 
-@pytest.yield_fixture
+@pytest.fixture
 def db():
     def clear_collection(collection):
         """Removes every document from the collection, to make it easy to see
@@ -54,13 +54,13 @@ def clear_collection(collection):
     yield client
 
 
-@pytest.yield_fixture
+@pytest.fixture
 def publisher():
     client = pubsub.PublisherClient()
     yield client
 
 
-@pytest.yield_fixture
+@pytest.fixture
 def subscriber():
     subscriber = pubsub.SubscriberClient()
     subscriber.create_subscription(
diff --git a/bookshelf/main_test.py b/bookshelf/main_test.py
@@ -26,7 +26,7 @@
 os.environ['GOOGLE_CLOUD_PROJECT'] = project_id
 
 
-@pytest.yield_fixture
+@pytest.fixture
 def app(request):
     """This fixture provides a Flask app instance configured for testing.
 
@@ -38,7 +38,7 @@ def app(request):
         yield app
 
 
-@pytest.yield_fixture
+@pytest.fixture
 def firestore():
 
     import firestore
diff --git a/optional-kubernetes-engine/tests/conftest.py b/optional-kubernetes-engine/tests/conftest.py
@@ -22,7 +22,7 @@
 from retrying import retry
 
 
-@pytest.yield_fixture(params=['datastore', 'mongodb'])
+@pytest.fixture(params=['datastore', 'mongodb'])
 def app(request):
     """This fixtures provides a Flask app instance configured for testing.
 
@@ -43,7 +43,7 @@ def app(request):
         yield app
 
 
-@pytest.yield_fixture
+@pytest.fixture
 def model(monkeypatch, app):
     """This fixture provides a modified version of the app's model that tracks
     all created items and deletes them at the end of the test.
