def setUp(self):
super(OauthTest, self).setUp()
key = ndb.Key(Client, TEST_CLIENT_ID)
data = {
'client_id': TEST_CLIENT_ID,
'redirect_uris': [TEST_REDIRECT_URI],
'name': TEST_CLIENT_NAME,
'uri': TEST_CLIENT_URI,
'logo_uri': TEST_LOGO_URI,
'scope': ['data'],
'secret': TEST_CLIENT_SECRET,
'secret_expires_at': 0,
'registration_access_token': TEST_REGISTRATION_ACCESS_TOKEN
}
self.client = Client(key=key, **data)
self.client.put()
def get_or_create_client(self):
client = self.client
if client is None:
from oauth import Client, authorization_provider
existing_client = True
while existing_client:
random_int = ''.join(
[random.choice(UNICODE_ASCII_DIGITS) for x in xrange(5)])
client_id = Client.get_key_name("{0}-{1}".format(
CONTROLLER_CLIENT_ID, random_int))
client_key = Client.get_key(client_id)
existing_client = client_key.get()
client = Client(
key=client_key,
client_id=client_id,
instance_key=self.key,
redirect_uris=['/'],
scope=['controller'],
secret=authorization_provider.generate_client_secret())
client.put()
self.client_key = client.key
self.put()
return client
def create(cls, **kwargs):
instance = cls(**kwargs)
instance.put()
if instance.agent_key is None:
from oauth import Client, authorization_provider
existing_agent_client = True
while existing_agent_client:
random_int = ''.join(
[random.choice(UNICODE_ASCII_DIGITS) for x in xrange(5)])
client_id = Client.get_key_name("{0}-{1}".format(
AGENT_CLIENT_ID, random_int))
agent_client_key = Client.get_key(client_id)
existing_agent_client = agent_client_key.get()
agent_client = Client(
key=agent_client_key,
client_id=client_id,
server_key=instance.key,
redirect_uris=['/'],
scope=['agent'],
secret=authorization_provider.generate_client_secret())
agent_client.put()
instance.agent_key = agent_client.key
instance.put()
return instance
