python-docs-samples/auth/api-client/snippets_test.py at master · git-connected/python-docs-samples · GitHub

52 lines (37 loc) · 1.52 KB

# 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

import google.auth

import mock

import snippets

def test_implicit():

snippets.implicit(os.environ['GOOGLE_CLOUD_PROJECT'])

def test_explicit():

with open(os.environ['GOOGLE_APPLICATION_CREDENTIALS']) as creds_file:

creds_file_data = creds_file.read()

open_mock = mock.mock_open(read_data=creds_file_data)

with mock.patch('io.open', open_mock):

snippets.explicit(os.environ['GOOGLE_CLOUD_PROJECT'])

def test_explicit_compute_engine():

adc, project = google.auth.default()

credentials_patch = mock.patch(

'google.auth.compute_engine.Credentials', return_value=adc)

with credentials_patch:

snippets.explicit_compute_engine(project)

def test_explicit_app_engine():

adc, project = google.auth.default()

credentials_patch = mock.patch(

'google.auth.app_engine.Credentials', return_value=adc)

with credentials_patch:

snippets.explicit_app_engine(project)

Expand file tree