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)