def test_minimal_green_path(self, mock_hook): mock_hook.return_value.delete_product_set.return_value = {} op = CloudVisionDeleteProductSetOperator( location=LOCATION_TEST, product_set_id=PRODUCTSET_ID_TEST, task_id='id' ) op.execute(context=None) mock_hook.assert_called_once_with(gcp_conn_id=GCP_CONN_ID) mock_hook.return_value.delete_product_set.assert_called_once_with( location=LOCATION_TEST, product_set_id=PRODUCTSET_ID_TEST, project_id=None, retry=None, timeout=None, metadata=None, )
) # [END howto_operator_vision_product_set_get] # [START howto_operator_vision_product_set_update] product_set_update = CloudVisionUpdateProductSetOperator( location=GCP_VISION_LOCATION, product_set_id=product_set_create_output, product_set=ProductSet(display_name='My Product Set 2'), task_id='product_set_update', ) # [END howto_operator_vision_product_set_update] # [START howto_operator_vision_product_set_delete] product_set_delete = CloudVisionDeleteProductSetOperator( location=GCP_VISION_LOCATION, product_set_id=product_set_create_output, task_id='product_set_delete', ) # [END howto_operator_vision_product_set_delete] # [START howto_operator_vision_product_create] product_create = CloudVisionCreateProductOperator( location=GCP_VISION_LOCATION, product=product, retry=Retry(maximum=10.0), timeout=5, task_id='product_create', ) # [END howto_operator_vision_product_create] product_create_output = product_create.output
) # [END howto_operator_vision_product_set_get] # [START howto_operator_vision_product_set_update] product_set_update = CloudVisionUpdateProductSetOperator( location=GCP_VISION_LOCATION, product_set_id="{{ task_instance.xcom_pull('product_set_create') }}", product_set=ProductSet(display_name='My Product Set 2'), task_id='product_set_update', ) # [END howto_operator_vision_product_set_update] # [START howto_operator_vision_product_set_delete] product_set_delete = CloudVisionDeleteProductSetOperator( location=GCP_VISION_LOCATION, product_set_id="{{ task_instance.xcom_pull('product_set_create') }}", task_id='product_set_delete', ) # [END howto_operator_vision_product_set_delete] # [START howto_operator_vision_product_create] product_create = CloudVisionCreateProductOperator( location=GCP_VISION_LOCATION, product=product, retry=Retry(maximum=10.0), timeout=5, task_id='product_create', ) # [END howto_operator_vision_product_create] # [START howto_operator_vision_product_get]