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 up
create_product(PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_DISPLAY_NAME,
PRODUCT_CATEGORY)
yield None
# tear down
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
def test_delete_product(capsys, product):
list_products(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_ID in out
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
list_products(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_ID not in out
def product():
# set up
create_product(
PROJECT_ID, LOCATION, PRODUCT_ID,
PRODUCT_DISPLAY_NAME, PRODUCT_CATEGORY)
yield None
# tear down
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
def test_delete_product(capsys, product):
list_products(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_ID in out
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
list_products(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_ID not in out
def test_create_reference_image(capsys, product):
list_reference_images(PROJECT_ID, LOCATION, PRODUCT_ID)
out, _ = capsys.readouterr()
assert REFERENCE_IMAGE_ID not in out
create_reference_image(PROJECT_ID, LOCATION, PRODUCT_ID,
REFERENCE_IMAGE_ID, GCS_URI)
list_reference_images(PROJECT_ID, LOCATION, PRODUCT_ID)
out, _ = capsys.readouterr()
assert REFERENCE_IMAGE_ID in out
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
def test_update_product_labels(capsys, product):
get_product(PROJECT_ID, LOCATION, PRODUCT_ID)
out, _ = capsys.readouterr()
assert KEY not in out
assert VALUE not in out
update_product_labels(PROJECT_ID, LOCATION, PRODUCT_ID, KEY, VALUE)
out, _ = capsys.readouterr()
assert KEY in out
assert VALUE in out
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
def test_create_product(capsys):
list_products(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_ID not in out
create_product(PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_DISPLAY_NAME,
PRODUCT_CATEGORY)
list_products(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_ID in out
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
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_update_product_labels(capsys, product):
get_product(PROJECT_ID, LOCATION, PRODUCT_ID)
out, _ = capsys.readouterr()
assert KEY not in out
assert VALUE not in out
update_product_labels(PROJECT_ID, LOCATION, PRODUCT_ID, KEY, VALUE)
out, _ = capsys.readouterr()
assert KEY in out
assert VALUE in out
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
def test_create_reference_image(capsys, product):
list_reference_images(PROJECT_ID, LOCATION, PRODUCT_ID)
out, _ = capsys.readouterr()
assert REFERENCE_IMAGE_ID not in out
create_reference_image(
PROJECT_ID, LOCATION, PRODUCT_ID, REFERENCE_IMAGE_ID,
GCS_URI)
list_reference_images(PROJECT_ID, LOCATION, PRODUCT_ID)
out, _ = capsys.readouterr()
assert REFERENCE_IMAGE_ID in out
delete_product(PROJECT_ID, LOCATION, PRODUCT_ID)
def test_create_product(capsys):
list_products(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_ID not in out
create_product(
PROJECT_ID, LOCATION, PRODUCT_ID,
PRODUCT_DISPLAY_NAME, PRODUCT_CATEGORY)
list_products(PROJECT_ID, LOCATION)
out, _ = capsys.readouterr()
assert PRODUCT_ID in out
delete_product(PROJECT_ID, LOCATION, PRODUCT_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(PROJECT_ID, LOCATION, PRODUCT_ID)
