def test_annotate_video(self):
        # Setup Expected Response
        expected_response = {}
        expected_response = video_intelligence_pb2.AnnotateVideoResponse(
            **expected_response
        )
        operation = operations_pb2.Operation(
            name="operations/test_annotate_video", done=True
        )
        operation.response.Pack(expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = videointelligence_v1p3beta1.VideoIntelligenceServiceClient()

        # Setup Request
        features_element = enums.Feature.LABEL_DETECTION
        features = [features_element]
        input_uri = "gs://cloud-samples-data/video/cat.mp4"

        response = client.annotate_video(features, input_uri=input_uri)
        result = response.result()
        assert expected_response == result

        assert len(channel.requests) == 1
        expected_request = video_intelligence_pb2.AnnotateVideoRequest(
            features=features, input_uri=input_uri
        )
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
    def annotate_video(
        self,
        input_uri=None,
        input_content=None,
        features=None,
        video_context=None,
        output_uri=None,
        location_id=None,
        retry=google.api_core.gapic_v1.method.DEFAULT,
        timeout=google.api_core.gapic_v1.method.DEFAULT,
        metadata=None,
    ):
        """
        Performs asynchronous video annotation. Progress and results can be
        retrieved through the ``google.longrunning.Operations`` interface.
        ``Operation.metadata`` contains ``AnnotateVideoProgress`` (progress).
        ``Operation.response`` contains ``AnnotateVideoResponse`` (results).

        Example:
            >>> from google.cloud import videointelligence_v1p3beta1
            >>> from google.cloud.videointelligence_v1p3beta1 import enums
            >>>
            >>> client = videointelligence_v1p3beta1.VideoIntelligenceServiceClient()
            >>>
            >>> input_uri = 'gs://cloud-samples-data/video/cat.mp4'
            >>> features_element = enums.Feature.LABEL_DETECTION
            >>> features = [features_element]
            >>>
            >>> response = client.annotate_video(input_uri=input_uri, features=features)
            >>>
            >>> def callback(operation_future):
            ...     # Handle result.
            ...     result = operation_future.result()
            >>>
            >>> response.add_done_callback(callback)
            >>>
            >>> # Handle metadata.
            >>> metadata = response.metadata()

        Args:
            input_uri (str): Input video location. Currently, only `Google Cloud
                Storage <https://cloud.google.com/storage/>`__ URIs are supported, which
                must be specified in the following format: ``gs://bucket-id/object-id``
                (other URI formats return ``google.rpc.Code.INVALID_ARGUMENT``). For
                more information, see `Request
                URIs <https://cloud.google.com/storage/docs/reference-uris>`__. A video
                URI may include wildcards in ``object-id``, and thus identify multiple
                videos. Supported wildcards: '\*' to match 0 or more characters; '?' to
                match 1 character. If unset, the input video should be embedded in the
                request as ``input_content``. If set, ``input_content`` should be unset.
            input_content (bytes): The video data bytes. If unset, the input video(s) should be specified
                via ``input_uri``. If set, ``input_uri`` should be unset.
            features (list[~google.cloud.videointelligence_v1p3beta1.types.Feature]): Requested video annotation features.
            video_context (Union[dict, ~google.cloud.videointelligence_v1p3beta1.types.VideoContext]): Additional video context and/or feature-specific parameters.

                If a dict is provided, it must be of the same form as the protobuf
                message :class:`~google.cloud.videointelligence_v1p3beta1.types.VideoContext`
            output_uri (str): Optional location where the output (in JSON format) should be stored.
                Currently, only `Google Cloud
                Storage <https://cloud.google.com/storage/>`__ URIs are supported, which
                must be specified in the following format: ``gs://bucket-id/object-id``
                (other URI formats return ``google.rpc.Code.INVALID_ARGUMENT``). For
                more information, see `Request
                URIs <https://cloud.google.com/storage/docs/reference-uris>`__.
            location_id (str): Optional cloud region where annotation should take place. Supported
                cloud regions: ``us-east1``, ``us-west1``, ``europe-west1``,
                ``asia-east1``. If no region is specified, a region will be determined
                based on video file location.
            retry (Optional[google.api_core.retry.Retry]):  A retry object used
                to retry requests. If ``None`` is specified, requests will not
                be retried.
            timeout (Optional[float]): The amount of time, in seconds, to wait
                for the request to complete. Note that if ``retry`` is
                specified, the timeout applies to each individual attempt.
            metadata (Optional[Sequence[Tuple[str, str]]]): Additional metadata
                that is provided to the method.

        Returns:
            A :class:`~google.cloud.videointelligence_v1p3beta1.types._OperationFuture` instance.

        Raises:
            google.api_core.exceptions.GoogleAPICallError: If the request
                    failed for any reason.
            google.api_core.exceptions.RetryError: If the request failed due
                    to a retryable error and retry attempts failed.
            ValueError: If the parameters are invalid.
        """
        # Wrap the transport method to add retry and timeout logic.
        if "annotate_video" not in self._inner_api_calls:
            self._inner_api_calls[
                "annotate_video"] = google.api_core.gapic_v1.method.wrap_method(
                    self.transport.annotate_video,
                    default_retry=self._method_configs["AnnotateVideo"].retry,
                    default_timeout=self._method_configs["AnnotateVideo"].
                    timeout,
                    client_info=self._client_info,
                )

        request = video_intelligence_pb2.AnnotateVideoRequest(
            input_uri=input_uri,
            input_content=input_content,
            features=features,
            video_context=video_context,
            output_uri=output_uri,
            location_id=location_id,
        )
        operation = self._inner_api_calls["annotate_video"](request,
                                                            retry=retry,
                                                            timeout=timeout,
                                                            metadata=metadata)
        return google.api_core.operation.from_gapic(
            operation,
            self.transport._operations_client,
            video_intelligence_pb2.AnnotateVideoResponse,
            metadata_type=video_intelligence_pb2.AnnotateVideoProgress,
        )