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()