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))