def test_detect_video_explicit_content_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
CloudVideoIntelligenceDetectVideoExplicitContentOperator(
input_uri=INPUT_URI,
task_id="id",
gcp_conn_id=GCP_CONN_ID,
impersonation_chain=IMPERSONATION_CHAIN,
).execute(context={"task_instance": mock.Mock()})
mock_hook.assert_called_once_with(
gcp_conn_id=GCP_CONN_ID,
impersonation_chain=IMPERSONATION_CHAIN,
)
mock_hook.return_value.annotate_video.assert_called_once_with(
input_uri=INPUT_URI,
features=[enums.Feature.EXPLICIT_CONTENT_DETECTION],
input_content=None,
video_context=None,
location=None,
retry=None,
timeout=None,
)
def test_detect_video_shots_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
CloudVideoIntelligenceDetectVideoShotsOperator(
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.SHOT_CHANGE_DETECTION],
input_content=None,
video_context=None,
location=None,
retry=None,
)