def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem()
self.tempfile = fake_tempfile.FakeTempfileModule(self.filesystem)
self.fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
self.host1 = 'h1'
self.port1 = 1
self.host2 = 'h2'
self.port2 = 2
self.uplink_host = '127.0.0.1'
self.uplink_port = 999
self.username = '******'
self.password = '******'
self.test1_name = 'Test1'
self.test2_name = 'Test2'
self.test2_version = 'testing'
@googleads.common.RegisterUtility(self.test1_name)
class Test1(object):
def test(self):
pass
@googleads.common.RegisterUtility(self.test2_name,
{'test': self.test2_version})
class Test2(object):
def test(self):
pass
self.test1 = Test1
self.test2 = Test2
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem()
self.tempfile = fake_tempfile.FakeTempfileModule(self.filesystem)
self.fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
self.uri1 = 'http://*****:*****@h1:1'
self.uri2 = 'http://h2:2'
self.uplink_uri = 'http://127.0.0.1:999'
self.test1_name = 'Test1'
self.test2_name = 'Test2'
self.test2_version = 'testing'
self.fake_version = 'ignored'
locale_patcher = mock.patch('googleads.common.locale.getdefaultlocale',
return_value=('en_us', 'UTF-8'))
self.locale_patcher = locale_patcher.start()
@googleads.common.RegisterUtility(self.test1_name)
class Test1(object):
def test(self):
pass
@googleads.common.RegisterUtility(self.test2_name,
{'test': self.test2_version})
class Test2(object):
def test(self):
pass
self.test1 = Test1
self.test2 = Test2
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator='/')
self.os = fake_filesystem.FakeOsModule(self.filesystem)
self.tempfile = fake_tempfile.FakeTempfileModule(self.filesystem)
self.orig_logging = fake_tempfile.logging
self.fake_logging = FakeLogging(self)
fake_tempfile.logging = self.fake_logging
def setUp(self):
self.scope = 'scope'
self.private_key = b'IT\'S A SECRET TO EVERYBODY.'
self.delegated_account = '*****@*****.**'
# Mock out filesystem and file for testing.
filesystem = fake_filesystem.FakeFilesystem()
tempfile = fake_tempfile.FakeTempfileModule(filesystem)
self.fake_open = fake_filesystem.FakeFileOpen(filesystem)
self.key_file_path = tempfile.NamedTemporaryFile(delete=False).name
self.cert_file_path = tempfile.NamedTemporaryFile(delete=False,
prefix='cert_',
suffix='.pem').name
with self.fake_open(self.key_file_path, 'wb') as file_handle:
file_handle.write(self.private_key)
self.access_token_unrefreshed = 'a'
self.access_token_refreshed = 'b'
# Mock out google.auth.transport.Request for testing.
self.mock_req = mock.Mock(spec=Request)
self.mock_req.return_value = mock.Mock()
self.mock_req_instance = self.mock_req.return_value
# Mock out service account credentials for testing.
self.mock_credentials = mock.Mock()
self.mock_credentials.from_service_account_file.return_value = mock.Mock(
)
self.mock_credentials_instance = (
self.mock_credentials.from_service_account_file.return_value)
self.mock_credentials_instance.token = 'x'
self.mock_credentials_instance.expiry = datetime.datetime(
1980, 1, 1, 12)
self.mock_credentials_instance.expired = True
def apply(headers, token=None):
headers['authorization'] = ('Bearer %s' %
self.mock_credentials_instance.token)
def refresh(request):
self.mock_credentials_instance.token = (
self.access_token_unrefreshed
if self.mock_credentials_instance.token == 'x' else
self.access_token_refreshed)
self.mock_credentials_instance.token_expiry = datetime.datetime.utcnow(
)
self.mock_credentials_instance.apply = mock.Mock(side_effect=apply)
self.mock_credentials_instance.refresh = mock.Mock(side_effect=refresh)
with mock.patch('builtins.open', self.fake_open):
with mock.patch('google.oauth2.service_account.Credentials',
self.mock_credentials):
self.sa_client = googleads.oauth2.GoogleServiceAccountClient(
self.key_file_path, self.scope)
# Undo the call count for the auto-refresh
self.mock_credentials_instance.refresh.reset_mock()
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem()
self.tempfile = fake_tempfile.FakeTempfileModule(self.filesystem)
self.fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
self.host1 = 'h1'
self.port1 = 1
self.host2 = 'h2'
self.port2 = 2
self.uplink_host = '127.0.0.1'
self.uplink_port = 999
self.username = '******'
self.password = '******'
def _refresh(self):
'''Renew the fake file system and set the _isStale flag to `False`.'''
if self._stubs is not None:
self._stubs.SmartUnsetAll()
self._stubs = mox3.stubout.StubOutForTesting()
self.fs = fake_filesystem.FakeFilesystem()
self.fake_os = fake_filesystem.FakeOsModule(self.fs)
self.fake_glob = fake_filesystem_glob.FakeGlobModule(self.fs)
self.fake_path = self.fake_os.path
self.fake_shutil = fake_filesystem_shutil.FakeShutilModule(self.fs)
self.fake_tempfile_ = fake_tempfile.FakeTempfileModule(self.fs)
self.fake_open = fake_filesystem.FakeFileOpen(self.fs)
self._isStale = False
def replaceGlobs(self, globs_):
globs = globs_.copy()
if self._isStale:
self._refresh()
if 'os' in globs:
globs['os'] = fake_filesystem.FakeOsModule(self.fs)
if 'glob' in globs:
globs['glob'] = fake_filesystem_glob.FakeGlobModule(self.fs)
if 'path' in globs:
fake_os = globs['os'] if 'os' in globs \
else fake_filesystem.FakeOsModule(self.fs)
globs['path'] = fake_os.path
if 'shutil' in globs:
globs['shutil'] = fake_filesystem_shutil.FakeShutilModule(self.fs)
if 'tempfile' in globs:
globs['tempfile'] = fake_tempfile.FakeTempfileModule(self.fs)
return globs
def _refresh(self):
"""Renew the fake file system and set the _isStale flag to `False`."""
if self._stubs is not None:
self._stubs.SmartUnsetAll()
self._stubs = mox3.stubout.StubOutForTesting()
self.fs = fake_filesystem.FakeFilesystem()
self.fake_os = fake_filesystem.FakeOsModule(self.fs)
self.fake_path = self.fake_os.path
if self.HAS_PATHLIB:
self.fake_pathlib = fake_pathlib.FakePathlibModule(self.fs)
self.fake_shutil = fake_filesystem_shutil.FakeShutilModule(self.fs)
self.fake_tempfile_ = fake_tempfile.FakeTempfileModule(self.fs)
self.fake_open = fake_filesystem.FakeFileOpen(self.fs)
self.fake_io = fake_filesystem.FakeIoModule(self.fs)
self._isStale = False
def setUp(self):
self.scope = 'scope'
self.service_account_email = '*****@*****.**'
self.private_key = b'IT\'S A SECRET TO EVERYBODY.'
self.private_key_password = '******'
self.delegated_account = '*****@*****.**'
https_proxy_host = 'myproxy.com'
https_proxy_port = 443
https_proxy = googleads.common.ProxyConfig.Proxy(
https_proxy_host, https_proxy_port)
self.proxy_config = googleads.common.ProxyConfig(
https_proxy=https_proxy)
self.https_proxy = '%s:%s' % (https_proxy_host, https_proxy_port)
self.access_token_unrefreshed = 'a'
self.access_token_refreshed = 'b'
# Mock out filesystem and file for testing.
filesystem = fake_filesystem.FakeFilesystem()
tempfile = fake_tempfile.FakeTempfileModule(filesystem)
self.fake_open = fake_filesystem.FakeFileOpen(filesystem)
self.key_file_path = tempfile.NamedTemporaryFile(delete=False).name
with self.fake_open(self.key_file_path, 'wb') as file_handle:
file_handle.write(self.private_key)
# Mock out httplib2.Http for testing.
self.http = mock.Mock(spec=httplib2.Http)
self.opener = self.http.return_value = mock.Mock()
self.opener.proxy_info = self.proxy_config.proxy_info
self.opener.ca_certs = self.proxy_config.cafile
self.opener.disable_ssl_certificate_valiation = (
self.proxy_config.disable_certificate_validation)
# Mock out service account credentials for testing.
self.oauth2_credentials = mock.Mock()
self.oauth2_credentials.return_value = mock.Mock()
self.mock_oauth2_credentials = self.oauth2_credentials.return_value
self.mock_oauth2_credentials.access_token = 'x'
self.mock_oauth2_credentials.token_expiry = datetime.datetime(
1980, 1, 1, 12)
# Also mock out instantiation methods for newer oauth2client versions.
self.oauth2_credentials.from_p12_keyfile.return_value = (
self.mock_oauth2_credentials)
self.oauth2_credentials.from_json_keyfile_name.return_value = (
self.mock_oauth2_credentials)
def apply(headers):
headers['Authorization'] = (
'Bearer %s' % self.mock_oauth2_credentials.access_token)
def refresh(mock_http):
self.mock_oauth2_credentials.access_token = (
self.access_token_unrefreshed
if self.mock_oauth2_credentials.access_token is 'x' else
self.access_token_refreshed)
self.mock_oauth2_credentials.token_expiry = datetime.datetime.utcnow(
)
self.mock_oauth2_credentials.apply = mock.Mock(side_effect=apply)
self.mock_oauth2_credentials.refresh = mock.Mock(side_effect=refresh)
with mock.patch('%s.open' % _BUILTIN_PATH, self.fake_open):
with mock.patch(_SA_CRED_PATH, self.oauth2_credentials):
self.googleads_client = googleads.oauth2.GoogleServiceAccountClient(
self.scope,
self.service_account_email,
self.key_file_path,
self.private_key_password,
proxy_config=self.proxy_config)
# Undo the call count for the auto-refresh
self.mock_oauth2_credentials.refresh.reset_mock()
