Open
Show file tree
Hide file tree
Changes from all commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Failed to load files.
Original file line numberDiff line numberDiff line change
Expand Up@@ -17,10 +17,11 @@
import json
import os
import pathlib
from unittest import mock

import google.auth
from google.auth import crypt
from google.auth import exceptions

from google.auth import crypt, exceptions, impersonated_credentials
from google.oauth2 import credentials as gcredentials
from google.oauth2 import service_account
import pytest
Expand DownExpand Up@@ -191,3 +192,19 @@ def _verify_credential(self, credential):
access_token = credential.get_access_token()
assert access_token.access_token == 'mock_access_token'
assert isinstance(access_token.expiry, datetime.datetime)


class TestImpersonatedCredentials:
def test_init_from_valid_credentials(self) -> None:
mock_tcreds = mock.Mock(spec=impersonated_credentials.Credentials)
credential = credentials.ImpersonatedCredentials(mock_tcreds)
assert credential._g_credential == mock_tcreds

def test_init_from_invalid_credentials(self) -> None:
with pytest.raises(ValueError):
credentials.ImpersonatedCredentials(None)

def test_get_credential(self) -> None:
mock_tcreds = mock.Mock(spec=impersonated_credentials.Credentials)
credential = credentials.ImpersonatedCredentials(mock_tcreds)
assert credential.get_credential() == mock_tcreds