def test_user_remove_user_by_id_succeed(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodyruis')
usr.set_user_id('SomeUserid1234ruis')
ps.add_user(usr)
self.assertTrue(ps.remove_user_by_id('SomeUserid1234ruis'))
def test_user_get_user_by_id_succeed(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodyguis')
usr.set_user_id('SomeUserid1234guis')
ps.add_user(usr)
self.assertEqual(ps.get_user_by_id('SomeUserid1234guis').email(), usr.email())
def test_user_modify_user_by_id_succeed(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodymids')
usr.set_user_id('SomeUserid1234mids')
ps.add_user(usr)
newusr = User('*****@*****.**')
newusr.set_nickname('SomebodyElsemids')
newusr.set_user_id('SomeUotherId3878mids')
key = ps.modify_user_by_id('SomeUserid1234mids',newusr)
self.assertEqual(ps.get_user_by_key(key).email(), newusr.email())
def test_user_add_user(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodyau')
usr.set_user_id('SomeUserid1234au')
ret = ps.add_user(usr)
self.assertTrue(isinstance(ret, ndb.Key))
def test_user_get_user_key_by_id(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodyguki')
usr.set_user_id('SomeUserid1234guki')
key = ps.add_user(usr)
self.assertEqual(key,ps.get_user_key_by_id("SomeUserid1234guki"))
def signup_handler(request):
authed_user = auth.get_current_user()
if authed_user is None:
return __unauthed_response()
user_key = ps.get_user_key_by_id(authed_user.user_id())
if user_key is None:
user_key = ps.add_user(authed_user)
response_part = {
'success' : False
}
try:
file_req = json.loads(request.raw_post_data)
except ValueError:
response_part.update({'error' : 'Invalid request payload.'})
return HttpResponse(json.dumps(response_part), content_type="application/json")
if 'action' not in file_req:
response_part.update({'error' : 'Incomplete request.'})
return HttpResponse(json.dumps(response_part), content_type="application/json")
elem_key = ps.get_element_key_by_ref('canLogIn')
if elem_key is None:
elem_key = ps.add_element('canLogIn')
user_elem = ps.get_user_element_permissions(user_key, elem_key)
if user_elem is not None:
response_part.update({'error' : 'Already signed up.'})
return HttpResponse(json.dumps(response_part), content_type="application/json")
else:
ps.add_element_permissions(user_key, elem_key, True)
response_part.update({'success' : True})
return HttpResponse(json.dumps(response_part), content_type="application/json")
