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, )