def teardown():
# no set up, tear down only
yield None
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID_1)
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID_2)
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
def product_set():
# set up
create_product_set(
PROJECT_ID, LOCATION, PRODUCT_SET_ID, PRODUCT_SET_DISPLAY_NAME)
yield None
# tear down
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
def product_set():
# set up
create_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID,
PRODUCT_SET_DISPLAY_NAME)
yield None
# tear down
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
def test_delete_product_set(capsys):
list_product_sets(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_SET_ID in out
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
list_product_sets(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_SET_ID not in out
def test_delete_product_set(capsys, product_set):
list_product_sets(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_SET_ID in out
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
list_product_sets(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_SET_ID not in out
def setup_teardown():
# set up
create_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID,
PRODUCT_SET_DISPLAY_NAME)
create_product(PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_DISPLAY_NAME,
PRODUCT_CATEGORY)
yield
# tear down
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
def test_create_product_set(capsys):
list_product_sets(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_SET_ID not in out
create_product_set(
PROJECT_ID, LOCATION, PRODUCT_SET_ID,
PRODUCT_SET_DISPLAY_NAME)
list_product_sets(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_SET_ID in out
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
def test_resources():
# set up
product_set_id = f'test_set_{uuid.uuid4()}'
product_id = f'test_product_{uuid.uuid4()}'
create_product_set(PROJECT_ID, LOCATION, product_set_id,
PRODUCT_SET_DISPLAY_NAME)
create_product(PROJECT_ID, LOCATION, product_id, PRODUCT_DISPLAY_NAME,
PRODUCT_CATEGORY)
yield product_set_id, product_id
# tear down
delete_product(PROJECT_ID, LOCATION, product_id)
delete_product_set(PROJECT_ID, LOCATION, product_set_id)
def setup_teardown():
# Create the product set csv file locally and upload it to GCS
# This is so that there is a unique product set ID for all python version
# tests.
client = storage.Client(project=PROJECT_ID)
bucket = client.get_bucket(PROJECT_ID)
blob = storage.Blob("vision/{}.csv".format(FILENAME), bucket)
blob.upload_from_string(
'"gs://cloud-samples-data/vision/product_search/shoes_1.jpg",' +
'"{}",'.format(IMAGE_URI_1) + '"{}",'.format(PRODUCT_SET_ID) +
'"{}",'.format(PRODUCT_ID_1) +
'"apparel",,"style=womens","0.1,0.1,0.9,0.1,0.9,0.9,0.1,0.9"')
yield
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID_1)
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
# Delete the created file
blob.delete(client)
def teardown():
yield
# tear down
delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
