def test_minimal_green_path(self, mock_hook): op = CloudVisionAddProductToProductSetOperator( 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.add_product_to_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_reference_image_delete] reference_image_delete = CloudVisionDeleteReferenceImageOperator( location=GCP_VISION_LOCATION, product_id=product_create_output, reference_image_id=GCP_VISION_REFERENCE_IMAGE_ID, retry=Retry(maximum=10.0), timeout=5, task_id='reference_image_delete', ) # [END howto_operator_vision_reference_image_delete] # [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]
# [START howto_operator_vision_reference_image_delete] reference_image_delete = CloudVisionDeleteReferenceImageOperator( location=GCP_VISION_LOCATION, product_id="{{ task_instance.xcom_pull('product_create') }}", reference_image_id=GCP_VISION_REFERENCE_IMAGE_ID, retry=Retry(maximum=10.0), timeout=5, task_id='reference_image_delete', ) # [END howto_operator_vision_reference_image_delete] # [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]