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)