def test_minimal_green_path(self, mock_hook): mock_hook.return_value.get_product_set.return_value = {} op = CloudVisionGetProductSetOperator( 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.get_product_set.assert_called_once_with( location=LOCATION_TEST, product_set_id=PRODUCTSET_ID_TEST, project_id=None, retry=None, timeout=None, metadata=None, )
# [START howto_operator_vision_product_set_create] product_set_create = CloudVisionCreateProductSetOperator( location=GCP_VISION_LOCATION, product_set=product_set, retry=Retry(maximum=10.0), timeout=5, task_id='product_set_create', ) # [END howto_operator_vision_product_set_create] product_set_create_output = product_set_create.output # [START howto_operator_vision_product_set_get] product_set_get = CloudVisionGetProductSetOperator( location=GCP_VISION_LOCATION, product_set_id=product_set_create_output, task_id='product_set_get', ) # [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(
# ################################## # # [START howto_operator_vision_product_set_create] product_set_create = CloudVisionCreateProductSetOperator( location=GCP_VISION_LOCATION, product_set=product_set, retry=Retry(maximum=10.0), timeout=5, task_id='product_set_create', ) # [END howto_operator_vision_product_set_create] # [START howto_operator_vision_product_set_get] product_set_get = CloudVisionGetProductSetOperator( location=GCP_VISION_LOCATION, product_set_id="{{ task_instance.xcom_pull('product_set_create') }}", task_id='product_set_get', ) # [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(