diff --git a/samples/snippets/batch_process_documents_sample_bad_input_test.py b/samples/snippets/batch_process_documents_sample_bad_input_test.py deleted file mode 100644 index 78c40bd6..00000000 --- a/samples/snippets/batch_process_documents_sample_bad_input_test.py +++ /dev/null @@ -1,49 +0,0 @@ -# Copyright 2020 Google LLC -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. -# - -import os -from uuid import uuid4 - -from google.api_core.exceptions import InternalServerError, RetryError -from samples.snippets import batch_process_documents_sample - -location = "us" -project_id = os.getenv("GOOGLE_CLOUD_PROJECT") -processor_id = "90484cfdedb024f6" -gcs_input_uri = "gs://cloud-samples-data/documentai/invoice.pdf" -input_mime_type = "application/pdf" -# following bucket contains .csv file which will cause the sample to fail. -gcs_output_full_uri_with_wrong_type = "gs://documentai-beta-samples" -gcs_output_uri_prefix = "test/" -BUCKET_NAME = f"document-ai-python-{uuid4()}" - - -def test_batch_process_documents_with_bad_input(capsys): - try: - batch_process_documents_sample.batch_process_documents( - project_id=project_id, - location=location, - processor_id=processor_id, - gcs_input_uri=gcs_input_uri, - input_mime_type=input_mime_type, - gcs_output_bucket=gcs_output_full_uri_with_wrong_type, - gcs_output_uri_prefix=gcs_output_uri_prefix, - timeout=450, - ) - except ValueError: - out, _ = capsys.readouterr() - assert "Failed" in out or "error" in out - except (InternalServerError, RetryError) as e: - assert "error" in e.message diff --git a/samples/snippets/requirements.txt b/samples/snippets/requirements.txt index cb18a665..9e5470d6 100644 --- a/samples/snippets/requirements.txt +++ b/samples/snippets/requirements.txt @@ -1,2 +1,2 @@ google-cloud-documentai==2.4.0 -google-cloud-storage==2.6.0 +google-cloud-storage==2.7.0