def callback(self):
session = ndb.Key(urlsafe=self.request.params['state']).get()
credentials = session.flow.step2_exchange(self.request.params['code'])
OAuth2UserCredentials.create(user=self.user,
scopes=session.scopes,
credentials=credentials,
admin=session.admin)
session.key.delete() # No need for the session any longer
return self.redirect(str(session.redirect))
def callback(self):
session = ndb.Key(urlsafe=self.request.params['state']).get()
credentials = session.flow.step2_exchange(self.request.params['code'])
OAuth2UserCredentials.create(
user=self.user,
scopes=session.scopes,
credentials=credentials,
admin=session.admin
)
session.key.delete() # No need for the session any longer
return self.redirect(str(session.redirect))
def test_credential_storage(self):
user1 = users.User("*****@*****.**")
user2 = users.User("*****@*****.**")
assert UserCredentials.query().count() == 0
UserCredentials.create(user=user1,
scopes=['a', 'c', 'b'],
credentials=None,
admin=False)
UserCredentials.create(user=user2,
scopes=['e', 'f', 'g'],
credentials=None,
admin=True)
assert UserCredentials.query().count() == 2
assert find_credentials(user=user1, scopes=[
'a', 'b', 'c'
]), "Should find out of order scopes credentials"
assert find_credentials(user=user1,
scopes=['a', 'c',
'b']), "Should find credentials"
assert not find_credentials(user=user1, scopes=[
'a', 'b', 'd'
]), "Should not find credentials when scopes don't match"
assert not find_credentials(
user=user1, scopes=['a', 'b', 'c'],
admin=True), "Should not find credentials that aren't admin"
UserCredentials.create(user=user1,
scopes=['a', 'c', 'b'],
credentials=None,
admin=False)
assert UserCredentials.query().count(
) == 2, "Should overwrite existing credentials"
assert find_credentials(scopes=['e', 'g', 'f'],
admin=True), "Should find admin scopes"
def test_credential_storage(self):
user1 = users.User("*****@*****.**")
user2 = users.User("*****@*****.**")
assert UserCredentials.query().count() == 0
UserCredentials.create(
user=user1,
scopes=['a', 'c', 'b'],
credentials=None,
admin=False
)
UserCredentials.create(
user=user2,
scopes=['e', 'f', 'g'],
credentials=None,
admin=True
)
assert UserCredentials.query().count() == 2
assert find_credentials(user=user1, scopes=['a', 'b', 'c']), "Should find out of order scopes credentials"
assert find_credentials(user=user1, scopes=['a', 'c', 'b']), "Should find credentials"
assert not find_credentials(user=user1, scopes=['a', 'b', 'd']), "Should not find credentials when scopes don't match"
assert not find_credentials(user=user1, scopes=['a', 'b', 'c'], admin=True), "Should not find credentials that aren't admin"
UserCredentials.create(
user=user1,
scopes=['a', 'c', 'b'],
credentials=None,
admin=False
)
assert UserCredentials.query().count() == 2, "Should overwrite existing credentials"
assert find_credentials(scopes=['e', 'g', 'f'], admin=True), "Should find admin scopes"
