def setUp(self):
self.scope = 'scope'
self.service_account_email = '*****@*****.**'
self.private_key_password = '******'
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)
fake_open = fake_filesystem.FakeFileOpen(filesystem)
key_file_path = tempfile.NamedTemporaryFile(delete=False).name
with fake_open(key_file_path, 'w') as file_handle:
file_handle.write('IT\'S A SECRET TO EVERYBODY.')
# 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 oauth2client.client.OAuth2Credentials for testing
self.oauth2_credentials = mock.Mock(spec=SignedJwtAssertionCredentials)
self.mock_oauth2_credentials = self.oauth2_credentials.return_value = (
mock.Mock())
self.mock_oauth2_credentials.access_token = 'x'
self.mock_oauth2_credentials.token_expiry = datetime.datetime(1980, 1, 1,
12)
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('__builtin__.open', fake_open):
with mock.patch('oauth2client.client'
'.SignedJwtAssertionCredentials',
self.oauth2_credentials):
self.googleads_client = googleads.oauth2.GoogleServiceAccountClient(
self.scope, self.service_account_email, 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()
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem()
self.tempfile = fake_tempfile.FakeTempfileModule(self.filesystem)
self.fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
self.fake_proxy = mock.Mock()
self.fake_proxy.return_value = mock.Mock()
self.fake_proxy.return_value.proxy_host = 'ahost'
self.fake_proxy.return_value.proxy_port = 'aport'
def setUp(self):
self.real_tempfile = sys.modules['tempfile']
self.real_os = sys.modules['os']
self.filesystem = fake_filesystem.FakeFilesystem()
sys.modules['tempfile'] = fake_tempfile.FakeTempfileModule(
self.filesystem)
sys.modules['os'] = fake_filesystem.FakeOsModule(self.filesystem)
random.seed(32452)
def _refresh(self):
'''Renew the fake file system and set the _isStale flag to `False`.'''
mock.patch.stopall()
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:
globs['path'] = fake_filesystem.FakePathModule(self.fs)
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_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 setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator='/')
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.filesystem = fake_filesystem.FakeFilesystem()
self.tempfile = fake_tempfile.FakeTempfileModule(self.filesystem)
self.fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
