def test_e_permissions_get_user_element_permissions(self):
user_key = ndb.Key("uk","guep")
element_key = ndb.Key("ek","guep")
ps.add_element_permissions(user_key,element_key,True)
obj = ps.get_user_element_permissions(user_key,element_key)
self.assertEqual(obj.user_key, user_key)
self.assertEqual(obj.element_key, element_key)
self.assertEqual(obj.access, True)
def test_e_permissions_modify_user_element_permissions(self):
key = ps.add_element_permissions(ndb.Key("uk","muep"),ndb.Key("ek","muep"),True)
ps.modify_user_element_permissions(ndb.Key("uk","muep"),ndb.Key("ek","muep"), False)
retrieved = ps.get_element_permissions_by_key(key)
self.assertFalse(retrieved.access)
def test_e_permissions_revoke_element_permissions_user_key(self):
uk = ndb.Key("uk","repuk")
ps.add_element_permissions(uk,ndb.Key("ek","repuk1"),True)
ps.add_element_permissions(uk,ndb.Key("ek","repuk1"),True)
ps.add_element_permissions(uk,ndb.Key("ek","repuk1"),True)
ps.revoke_element_permissions_user_key(uk)
with self.assertRaises(StopIteration):
ps.get_element_permissions_by_user_key(uk).next()
def test_e_permissions_revoke_element_permissions_element_key(self):
ek = ndb.Key("ek","repek")
ps.add_element_permissions(ndb.Key("uk","repek1"),ek,True)
ps.add_element_permissions(ndb.Key("uk","repek2"),ek,True)
ps.add_element_permissions(ndb.Key("uk","repek3"),ek,True)
ps.revoke_element_permissions_element_key(ek)
with self.assertRaises(StopIteration):
ps.get_element_permissions_by_element_key(ek).next()
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")
def test_e_permissions_get_element_permissions_by_user_key(self):
uk = ndb.Key("uk","gepuk")
ps.add_element_permissions(uk,ndb.Key("ek","gepuk1"),True)
ps.add_element_permissions(uk,ndb.Key("ek","gepuk2"),True)
ps.add_element_permissions(uk,ndb.Key("ek","gepuk3"),True)
elements = ps.get_element_permissions_by_user_key(uk)
for element in elements:
self.assertEqual(element.user_key, uk)
def test_e_permissions_get_element_permissions_by_element_key(self):
ek = ndb.Key("ek","gepek")
ps.add_element_permissions(ndb.Key("uk","gepek1"),ek,True)
ps.add_element_permissions(ndb.Key("uk","gepek2"),ek,True)
ps.add_element_permissions(ndb.Key("uk","gepek3"),ek,True)
elements = ps.get_element_permissions_by_element_key(ek)
for element in elements:
self.assertEqual(element.element_key, ek)
def test_e_permissions_get_user_element_permissions_key(self):
user_key = ndb.Key("uk","guepk")
element_key = ndb.Key("ek","guepk")
key = ps.add_element_permissions(user_key,element_key,True)
self.assertEqual(key,ps.get_user_element_permissions_key(user_key,element_key))
def test_e_permissions_revoke_element_permissions_by_key(self):
key = ps.add_element_permissions(ndb.Key("uk","repk"),ndb.Key("ek","repk"),True)
ps.revoke_element_permissions_by_key(key)
self.assertIsNone(ps.get_element_permissions_by_key(key))
def test_e_permissions_revoke_user_element_permissions(self):
user_key = ndb.Key("uk","ruep")
element_key = ndb.Key("ek","ruep")
ps.add_element_permissions(user_key,element_key,True)
ps.revoke_user_element_permissions(user_key,element_key)
self.assertIsNone(ps.get_user_element_permissions(user_key,element_key))
def test_e_permissions_add_element_permissions(self):
ret = ps.add_element_permissions(ndb.Key("uk","aep"),ndb.Key("ek","aep"),True)
self.assertTrue(isinstance(ret, ndb.Key))
