def test_new_id(self):
for _ in range(100):
id1 = new_id()
id2 = new_id()
self.assertNotEqual(id1, id2)
self.assertIsNotNone(id1)
self.assertIsNotNone(id2)
def handler(event, context):
user_id = new_id()
while db.table('user').where({'id': user_id}).first():
user_id = new_id()
try:
body = json.loads(event.get('body'))
except:
body = {}
props = body.get('props') or {}
verbose = body.get('verbose')
unique_props = {k: v for k, v in props.items() if k in UNIQUE_KEYS}
custom_props = {k: v for k, v in props.items() if k not in unique_props}
if 'password' in custom_props:
custom_props['password'] = password(custom_props['password'])
try:
set_props(user_id, props=unique_props, unique=True)
set_props(user_id, props=custom_props)
db.table('user').insert(id=user_id)
except Exception as e:
return abort(400, parse_sql_error(e))
return {
'body':
dumps({
'data':
get(
user_id,
keys=props.keys(),
verbose=verbose,
is_me=True,
)
}),
'headers': {
'Access-Control-Allow-Origin': '*'
},
'statusCode':
201,
}
def test_exists(self):
set_props(new_id(), props={'username': '******'}, unique=True)
event = {'body': json.dumps({'props': {'username': '******'}})}
res = handler(event, None)
body = json.loads(res['body'])
self.assertEqual(res['statusCode'], 200)
self.assertTrue(body['data']['username']['exists'])
self.assertEqual(body['data']['username']['value'], 'riverleo')
def test_me_by_jwt(self):
user_id = new_id()
db.table('user').insert(id=user_id)
set_props(user_id, props={'key': 'value'})
encoded = jwt_encode(user_id)
res = handler({
'body': json.dumps({'props': ['key', 'anonymous']}),
'headers': {'Authorization': 'Bearer {}'.format(encoded)},
}, None)
body = json.loads(res['body'])
self.assertEqual(res['statusCode'], 200)
self.assertEqual(body['data']['id'], user_id)
self.assertEqual(body['data']['roles'], [])
self.assertEqual(body['data']['props']['key'], 'value')
self.assertIsNone(body['data']['props']['anonymous'])
def test_put_nonexistant(self):
res = handler(
{
'headers': {
'Authorization': jwt_encode(new_id()),
},
'body':
json.dumps({
'props': {
'username': '******',
'password': '******',
},
}),
}, None)
body = json.loads(res['body'])
self.assertIsNotNone(body['error'])
def test_get(self):
user_id = new_id()
db.table('user').insert(id=user_id)
set_props(user_id, props={'key': 'value'})
res = handler({
'queryStringParameters': {
'props': 'key,anonymous'
},
}, None)
body = json.loads(res['body'])
self.assertEqual(res['statusCode'], 200)
self.assertEqual(body['data'][0]['id'], user_id)
self.assertEqual(body['data'][0]['roles'], [])
self.assertEqual(body['data'][0]['props']['key'], 'value')
self.assertIsNone(body['data'][0]['props']['anonymous'])
def test_me_by_email_and_password(self):
user_id = new_id()
db.table('user').insert(id=user_id)
set_props(user_id, props={
'key': 'value',
'email': '*****@*****.**',
'password': password('password'),
}, unique=True)
res = handler({
'body': json.dumps({
'email': '*****@*****.**',
'password': '******',
'props': ['key', 'anonymous'],
}),
}, None)
body = json.loads(res['body'])
self.assertEqual(res['statusCode'], 200)
self.assertEqual(body['data']['id'], user_id)
self.assertIsNotNone(body['data']['ssid'])
self.assertEqual(body['data']['props']['key'], 'value')
self.assertIsNone(body['data']['props']['anonymous'])
res = handler({
'queryStringParameters': {
'email': '*****@*****.**',
'password': '******',
'props': 'key,anonymous',
},
}, None)
body = json.loads(res['body'])
self.assertEqual(res['statusCode'], 200)
self.assertEqual(body['data']['id'], user_id)
self.assertIsNotNone(body['data']['ssid'])
self.assertEqual(body['data']['props']['key'], 'value')
self.assertIsNone(body['data']['props']['anonymous'])
def set_connection(provider, provider_id, access_token, referer, props={}):
connection = (
db.table('userConnection')
.where({
'provider': provider,
'providerId': provider_id,
}).first()
)
if connection:
created = False
user_id = connection.get('userId')
(
db.table('userConnection')
.where('providerId', provider_id)
.update({'token': access_token})
)
else:
created = True
user_id = new_id()
conn_id = db.table('userConnection').insert_get_id({
'userId': user_id,
'provider': provider,
'providerId': provider_id,
'token': access_token,
})
db.table('user').insert(id=user_id)
set_props(user_id, props=props)
connection = db.table('userConnection').where('id', conn_id).first()
location = assign_query_params(referer, {
'ssid': jwt_encode(connection.get('userId')),
'created': int(created),
})
return connection, location
def test_put(self):
user_id = new_id()
db.table('user').insert(id=user_id)
res = handler(
{
'headers': {
'Authorization': jwt_encode(user_id),
},
'body':
json.dumps({
'props': {
'username': '******',
'password': '******',
},
}),
}, None)
body = json.loads(res['body'])
self.assertIsNotNone(body['data']['id'])
self.assertEqual(body['data']['props']['username'], 'riverleo')
self.assertFalse('password' in body['data']['props'])
