def test_minimal_green_path(self, mock_hook): op = CloudVisionRemoveProductFromProductSetOperator( location=LOCATION_TEST, product_set_id=PRODUCTSET_ID_TEST, product_id=PRODUCT_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.remove_product_from_product_set.assert_called_once_with( product_set_id=PRODUCTSET_ID_TEST, product_id=PRODUCT_ID_TEST, location=LOCATION_TEST, project_id=None, retry=None, timeout=None, metadata=None, )
# [START howto_operator_vision_add_product_to_product_set] add_product_to_product_set = CloudVisionAddProductToProductSetOperator( location=GCP_VISION_LOCATION, product_set_id=product_set_create_output, product_id=product_create_output, retry=Retry(maximum=10.0), timeout=5, task_id='add_product_to_product_set', ) # [END howto_operator_vision_add_product_to_product_set] # [START howto_operator_vision_remove_product_from_product_set] remove_product_from_product_set = CloudVisionRemoveProductFromProductSetOperator( location=GCP_VISION_LOCATION, product_set_id=product_set_create_output, product_id=product_create_output, retry=Retry(maximum=10.0), timeout=5, task_id='remove_product_from_product_set', ) # [END howto_operator_vision_remove_product_from_product_set] # Product path product_create >> product_get >> product_update >> product_delete # ProductSet path product_set_get >> product_set_update >> product_set_delete # ReferenceImage path reference_image_create >> reference_image_delete >> product_delete # Product/ProductSet path
# [START howto_operator_vision_add_product_to_product_set] add_product_to_product_set = CloudVisionAddProductToProductSetOperator( location=GCP_VISION_LOCATION, product_set_id="{{ task_instance.xcom_pull('product_set_create') }}", product_id="{{ task_instance.xcom_pull('product_create') }}", retry=Retry(maximum=10.0), timeout=5, task_id='add_product_to_product_set', ) # [END howto_operator_vision_add_product_to_product_set] # [START howto_operator_vision_remove_product_from_product_set] remove_product_from_product_set = CloudVisionRemoveProductFromProductSetOperator( location=GCP_VISION_LOCATION, product_set_id="{{ task_instance.xcom_pull('product_set_create') }}", product_id="{{ task_instance.xcom_pull('product_create') }}", retry=Retry(maximum=10.0), timeout=5, task_id='remove_product_from_product_set', ) # [END howto_operator_vision_remove_product_from_product_set] # Product path product_create >> product_get >> product_update >> product_delete # ProductSet path product_set_create >> product_set_get >> product_set_update >> product_set_delete # ReferenceImage path product_create >> reference_image_create >> reference_image_delete >> product_delete # Product/ProductSet path