def test_minimal_green_path(self, mock_hook): mock_hook.return_value.update_product.return_value = {} op = CloudVisionUpdateProductOperator(location=LOCATION_TEST, product=PRODUCT_TEST, task_id='id') op.execute(context=None) mock_hook.assert_called_once_with(gcp_conn_id=GCP_CONN_ID) mock_hook.return_value.update_product.assert_called_once_with( location=LOCATION_TEST, product=PRODUCT_TEST, product_id=None, project_id=None, retry=None, timeout=None, metadata=None, update_mask=None, )
product_create_output = product_create.output # [START howto_operator_vision_product_get] product_get = CloudVisionGetProductOperator( location=GCP_VISION_LOCATION, product_id=product_create_output, task_id='product_get', ) # [END howto_operator_vision_product_get] # [START howto_operator_vision_product_update] product_update = CloudVisionUpdateProductOperator( location=GCP_VISION_LOCATION, product_id=product_create_output, product=Product(display_name='My Product 2', description='My updated description'), task_id='product_update', ) # [END howto_operator_vision_product_update] # [START howto_operator_vision_product_delete] product_delete = CloudVisionDeleteProductOperator( location=GCP_VISION_LOCATION, product_id=product_create_output, task_id='product_delete', ) # [END howto_operator_vision_product_delete] # [START howto_operator_vision_reference_image_create] reference_image_create = CloudVisionCreateReferenceImageOperator(
) # [END howto_operator_vision_product_create] # [START howto_operator_vision_product_get] product_get = CloudVisionGetProductOperator( location=GCP_VISION_LOCATION, product_id="{{ task_instance.xcom_pull('product_create') }}", task_id='product_get', ) # [END howto_operator_vision_product_get] # [START howto_operator_vision_product_update] product_update = CloudVisionUpdateProductOperator( location=GCP_VISION_LOCATION, product_id="{{ task_instance.xcom_pull('product_create') }}", product=Product(display_name='My Product 2', description='My updated description'), task_id='product_update', ) # [END howto_operator_vision_product_update] # [START howto_operator_vision_product_delete] product_delete = CloudVisionDeleteProductOperator( location=GCP_VISION_LOCATION, product_id="{{ task_instance.xcom_pull('product_create') }}", task_id='product_delete', ) # [END howto_operator_vision_product_delete] # [START howto_operator_vision_reference_image_create] reference_image_create = CloudVisionCreateReferenceImageOperator(