def call(tok=None): headers = {'X-Delegation-Token-V1': tok} if tok else None self.call('127.0.0.1', '*****@*****.**', headers) return { 'cur_id': api.get_current_identity().to_bytes(), 'peer_id': api.get_current_identity().to_bytes(), }
def continuation(request, context, call_details): state.append(CapturedState( current_identity=api.get_current_identity().to_bytes(), is_superuser=api.is_superuser(), peer_identity=api.get_peer_identity().to_bytes(), peer_ip=api.get_peer_ip(), delegation_token=api.get_delegation_token(), )) return empty_pb2.Empty()
def call_with_tokens(self, delegation_tok=None, luci_project=None): headers = {} if delegation_tok: headers['X-Delegation-Token-V1'] = delegation_tok if luci_project: headers[check.X_LUCI_PROJECT] = luci_project self.call('127.0.0.1', '*****@*****.**', headers) return { 'cur_id': api.get_current_identity().to_bytes(), 'peer_id': api.get_peer_identity().to_bytes(), }
def call(self, remote_address, email, headers=None): """Mocks current user calls initialize_request_auth.""" class User(object): def email(self): return email self.mock( endpoints_support.endpoints, 'get_current_user', lambda: User() if email else None) api.reset_local_state() endpoints_support.initialize_request_auth(remote_address, headers or {}) return api.get_current_identity().to_bytes()
def call(self, remote_address, email, headers=None): """Mocks current user calls initialize_request_auth.""" class User(object): def email(self): return email self.mock(endpoints_support.endpoints, 'get_current_user', lambda: User() if email else None) api.reset_local_state() endpoints_support.initialize_request_auth(remote_address, headers or {}) return api.get_current_identity().to_bytes()
def call(self, remote_address, email, headers=None): """Mocks current user in initialize_request_auth.""" headers = (headers or {}).copy() if email: headers['Authorization'] = 'Bearer %s' % email # Mock ours auth. ident = model.Anonymous if email: ident = model.Identity(model.IDENTITY_USER, email) self.mock(api, 'check_oauth_access_token', lambda _: (ident, False)) # Mock auth implemented by the Cloud Endpoints. class User(object): def email(self): return email self.mock(endpoints_support.endpoints, 'get_current_user', lambda: User() if email else None) api.reset_local_state() endpoints_support.initialize_request_auth(remote_address, headers) return api.get_current_identity().to_bytes()
def get(self): test.assertEqual(model.Anonymous, api.get_current_identity()) self.response.write('OK')
def get(self): self.response.write( json.dumps({ 'peer_id': api.get_peer_identity().to_bytes(), 'cur_id': api.get_current_identity().to_bytes(), }))
def get(self): test.assertEqual(ident, api.get_current_identity()) self.response.write('OK')
def get(self): self.response.write(api.get_current_identity().to_bytes())
def test_get_current_identity(self): """Ensure get_current_identity returns whatever was put in request cache.""" api.get_request_cache().set_current_identity(model.Anonymous) self.assertEqual(model.Anonymous, api.get_current_identity())
def get(self): self.response.write(json.dumps({ 'peer_id': api.get_peer_identity().to_bytes(), 'cur_id': api.get_current_identity().to_bytes(), }))
def test_get_current_identity_unitialized(self): """If request cache is not initialized, returns Anonymous.""" self.assertEqual(api.get_current_identity(), model.Anonymous)
def test_get_current_identity_unitialized(self): """If set_current_identity wasn't called raises an exception.""" with self.assertRaises(api.UninitializedError): api.get_current_identity()
def test_get_current_identity(self): """Ensure get_current_identity returns whatever was put in request cache.""" ident = model.Identity.from_bytes("user:[email protected]") api.get_request_cache().current_identity = ident self.assertEqual(ident, api.get_current_identity())
def test_get_current_identity(self): """Ensure get_current_identity returns whatever was put in request cache.""" ident = model.Identity.from_bytes('user:[email protected]') api.get_request_cache().current_identity = ident self.assertEqual(ident, api.get_current_identity())
def get(self): test.assertEqual(ident, api.get_current_identity()) test.assertIs(auth_details, api.get_auth_details()) self.response.write('OK')