fix(retail): fix retail fulfillment places tests (GoogleCloudPlatform#10495)

chenlei1216 · gcf-owl-bot[bot] · web-flow · commit 30c71f231f60 · 2023-08-03T19:16:44.000-05:00
* Fix add_fulfillment_places.py * Fix build dependencies * Update remove_fulfillment_places.py * Use sync client for add_fulfillment_places * Use sync client for remove_fulfillment_places * Update client import * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * Fix set_inventory sample --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
diff --git a/retail/interactive-tutorials/product/add_fulfillment_places.py b/retail/interactive-tutorials/product/add_fulfillment_places.py
@@ -17,10 +17,11 @@
 import asyncio
 import random
 import string
+import time
 
 from google.api_core.exceptions import GoogleAPICallError
 import google.auth
-from google.cloud.retail import AddFulfillmentPlacesRequest, ProductServiceAsyncClient
+from google.cloud.retail import AddFulfillmentPlacesRequest, ProductServiceClient
 
 from setup_product.setup_cleanup import create_product, delete_product, get_product
 
@@ -57,12 +58,13 @@ def get_add_fulfillment_request(
 async def add_places(product_name: str):
     print("------add fulfillment places-----")
     add_fulfillment_request = get_add_fulfillment_request(product_name, "store2")
-    operation = await ProductServiceAsyncClient().add_fulfillment_places(
-        add_fulfillment_request
-    )
+    operation = ProductServiceClient().add_fulfillment_places(add_fulfillment_request)
     # This operation doesn't have result or errors. So GoogleAPICallError will be raised.
     try:
-        await operation.result()
+        while not operation.done():
+            print("---please wait till operation is done---")
+            time.sleep(30)
+        print("---add fulfillment places operation is done---")
     except GoogleAPICallError:
         pass
 
diff --git a/retail/interactive-tutorials/product/remove_fulfillment_places.py b/retail/interactive-tutorials/product/remove_fulfillment_places.py
@@ -17,10 +17,11 @@
 import asyncio
 import random
 import string
+import time
 
 from google.api_core.exceptions import GoogleAPICallError
 from google.cloud.retail import (
-    ProductServiceAsyncClient,
+    ProductServiceClient,
     RemoveFulfillmentPlacesRequest,
 )
 
@@ -52,12 +53,15 @@ def get_remove_fulfillment_request(
 async def remove_places(product_name: str):
     print("------remove fulfillment places-----")
     remove_fulfillment_request = get_remove_fulfillment_request(product_name, "store0")
-    operation = await ProductServiceAsyncClient().remove_fulfillment_places(
+    operation = ProductServiceClient().remove_fulfillment_places(
         remove_fulfillment_request
     )
     # This operation doesn't have result or errors. So GoogleAPICallError will be raised.
     try:
-        await operation.result()
+        while not operation.done():
+            print("---please wait till operation is done---")
+            time.sleep(30)
+        print("---remove fulfillment places operation is done---")
     except GoogleAPICallError:
         pass
 
diff --git a/retail/interactive-tutorials/product/set_inventory.py b/retail/interactive-tutorials/product/set_inventory.py
@@ -17,14 +17,15 @@
 import asyncio
 import random
 import string
+import time
 
 from google.api_core.exceptions import GoogleAPICallError
 import google.auth
 from google.cloud.retail import (
     FulfillmentInfo,
     PriceInfo,
     Product,
-    ProductServiceAsyncClient,
+    ProductServiceClient,
     SetInventoryRequest,
 )
 from google.protobuf.field_mask_pb2 import FieldMask
@@ -86,14 +87,17 @@ def get_set_inventory_request(product_name: str) -> SetInventoryRequest:
 # set inventory to product
 def set_inventory(product_name: str):
     set_inventory_request = get_set_inventory_request(product_name)
-    return ProductServiceAsyncClient().set_inventory(set_inventory_request)
+    return ProductServiceClient().set_inventory(set_inventory_request)
 
 
 async def set_inventory_and_remove_product(product_name: str):
-    operation = await set_inventory(product_name)
+    operation = set_inventory(product_name)
     # This operation doesn't have result or errors. So GoogleAPICallError will be raised.
     try:
-        await operation.result()
+        while not operation.done():
+            print("---please wait till operation is done---")
+            time.sleep(30)
+        print("---set inventory operation is done---")
     except GoogleAPICallError:
         pass
 
