def test_inspect_content(self, mock_hook): mock_hook.return_value.inspect_content.return_value = {} operator = CloudDLPInspectContentOperator(project_id=PROJECT_ID, task_id="id") operator.execute(context=None) mock_hook.assert_called_once_with(gcp_conn_id=GCP_CONN_ID) mock_hook.return_value.inspect_content.assert_called_once_with( project_id=PROJECT_ID, inspect_config=None, item=None, inspect_template_name=None, retry=None, timeout=None, metadata=None, )
INSPECT_TEMPLATE = InspectTemplate(inspect_config=INSPECT_CONFIG) with DAG("example_gcp_dlp", default_args=default_args, schedule_interval=None) as dag: create_template = CloudDLPCreateInspectTemplateOperator( project_id=GCP_PROJECT, inspect_template=INSPECT_TEMPLATE, template_id=TEMPLATE_ID, task_id="create_template", do_xcom_push=True, dag=dag, ) inspect_content = CloudDLPInspectContentOperator( task_id="inpsect_content", project_id=GCP_PROJECT, item=ITEM, inspect_template_name= "{{ task_instance.xcom_pull('create_template', key='return_value')['name'] }}", dag=dag, ) delete_template = CloudDLPDeleteInspectTemplateOperator( task_id="delete_template", template_id=TEMPLATE_ID, project_id=GCP_PROJECT, dag=dag, ) create_template > inspect_content > delete_template