def test_detect_video_labels_green_path(self, mock_hook):
mocked_operation = mock.Mock()
mocked_operation.result = mock.Mock(return_value=AnnotateVideoResponse(annotation_results=[]))
mock_hook.return_value.annotate_video.return_value = mocked_operation
CloudVideoIntelligenceDetectVideoLabelsOperator(
input_uri=INPUT_URI, task_id="id", gcp_conn_id=GCP_CONN_ID
).execute(context={"task_instance": mock.Mock()})
mock_hook.assert_called_once_with(gcp_conn_id=GCP_CONN_ID)
mock_hook.return_value.annotate_video.assert_called_once_with(
input_uri=INPUT_URI,
features=[enums.Feature.LABEL_DETECTION],
input_content=None,
video_context=None,
location=None,
retry=None,
)
# [START howto_operator_video_intelligence_other_args]
INPUT_URI = "gs://{}/video.mp4".format(GCP_BUCKET_NAME)
# [END howto_operator_video_intelligence_other_args]
with models.DAG(
"example_gcp_video_intelligence",
default_args=default_args,
schedule_interval=None, # Override to match your needs
) as dag:
# [START howto_operator_video_intelligence_detect_labels]
detect_video_label = CloudVideoIntelligenceDetectVideoLabelsOperator(
input_uri=INPUT_URI,
output_uri=None,
video_context=None,
timeout=5,
task_id="detect_video_label",
)
# [END howto_operator_video_intelligence_detect_labels]
# [START howto_operator_video_intelligence_detect_labels_result]
detect_video_label_result = BashOperator(
bash_command="echo {{ task_instance.xcom_pull('detect_video_label')"
"['annotationResults'][0]['shotLabelAnnotations'][0]['entity']}}",
task_id="detect_video_label_result",
)
# [END howto_operator_video_intelligence_detect_labels_result]
# [START howto_operator_video_intelligence_detect_explicit_content]
detect_video_explicit_content = CloudVideoIntelligenceDetectVideoExplicitContentOperator(