create_client_test.py

# Copyright 2021 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 json

import backoff
from google.api_core.exceptions import RetryError
from googleapiclient.errors import HttpError
import pytest

import create_client
import create_client_apiary


@pytest.fixture()
def job_filter(project_id: str):
    yield json.dumps({'project_id': project_id})


@backoff.on_exception(backoff.expo, (RetryError,), max_time=60)
def test_create_client(job_filter: str):
    client = create_client.create_transfer_client()

    # a simple test to prove a usable client has been created.
    # The output isn't relevant - just that a valid API call can be made.
    # We expect an error to be raised if this operation fails.
    client.list_transfer_jobs({'filter': job_filter, 'page_size': 1})


@backoff.on_exception(backoff.expo, (HttpError,), max_time=60)
def test_create_client_apiary(job_filter: str):
    client = create_client_apiary.create_transfer_client()

    # a simple test to prove a usable client has been created.
    # The output isn't relevant - just that a valid API call can be made.
    # We expect an error to be raised if this operation fails.
    client.transferJobs().list(
        filter=job_filter,
        pageSize=1).execute()