def testChangeNarrativePlace(self): logoutCurrentUser() setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Place 1") place2 = create_place("Place 2") role1 = create_role("Role 1") role2 = create_role("Role 2") create_presence(role1, place1) create_presence(role2, place2) set_playing(role1) create_narrative(place1, "Narrative 1 from Role 1 at Place 1") rset = get_first_narrative() self.assertEqual(len(rset["narrative"]), 1) narrative = rset["narrative"][0] set_playing(role2) rset = get_first_narrative() self.assertEqual(len(rset["narrative"]), 0) set_playing(role1) create_presence(role1, place2) save_narrative(narrative["id"], place_id=place2) rset = get_first_narrative() self.assertEqual(len(rset["narrative"]), 1) set_playing(role2) rset = get_first_narrative() self.assertEqual(len(rset["narrative"]), 1)
def testSetPlaying(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') role1 = create_role("Test Role 1") role2 = create_role("Test Role 2") set_playing(role1) self.assertEqual(get_my_roles()[0]["name"], "Test Role 1") set_playing(role2) self.assertEqual(get_my_roles()[1]["name"], "Test Role 2") set_playing(None) self.assertEqual(get_my_roles()[2]["name"], "User1")
def testIsPresent(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") place2 = create_place("Test Place 2") role1 = create_role("Test Role 1") role2 = create_role("Test Role 2") create_presence(role1, place1) create_presence(role2, place2) self.assertTrue(is_present(role1, place1)) self.assertFalse(is_present(role1, place2)) self.assertTrue(is_present(role2, place2)) self.assertFalse(is_present(role2, place1))
def testDeleteOccupiedPlace(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") role1 = create_role("Test Role 1") create_presence(role1, place1) self.assertRaises(ActionException, delete_place, place1)
def testBlankRoleName(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') roleid = create_role("Test Role 1") self.assertRaises(ActionException, create_role, None) self.assertRaises(DataModelException, create_role, "") self.assertRaises(DataModelException, save_role, roleid, name="")
def testChangingSomeoneElsesProfile(self): setCurrentUser("*****@*****.**", "Administrator") register_new_user('SiteAdmin') site_admin_id = _get_my_profile().key.urlsafe() logoutCurrentUser() setCurrentUser("*****@*****.**", "NewUser1") register_new_user('NewUserOne') new_user_1_id = _get_my_profile().key.urlsafe() role_1_id = create_role("Test Role 1") set_playing(role_1_id) logoutCurrentUser() setCurrentUser("*****@*****.**", "NewUser2") ''' This code simulates an attack against the Model, using the Google Account ID to load another users Profile ''' profile = ndb.Key(urlsafe=new_user_1_id).get() profile.playing = ndb.Key(urlsafe=role_1_id) self.assertRaises(DataModelException, profile.put) profile = ndb.Key(urlsafe=site_admin_id).get() profile.playing = ndb.Key(urlsafe=role_1_id) self.assertRaises(DataModelException, profile.put) register_new_user('NewUserTwo') ''' This code simulates an attack against the Model, using the Google Account ID to load another users Profile ''' profile = ndb.Key(urlsafe=new_user_1_id).get() profile.playing = ndb.Key(urlsafe=role_1_id) self.assertRaises(DataModelException, profile.put) profile = ndb.Key(urlsafe=site_admin_id).get() profile.playing = ndb.Key(urlsafe=role_1_id) self.assertRaises(DataModelException, profile.put)
def testChangeNarrativeBody(self): logoutCurrentUser() setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Place 1") role1 = create_role("Role 1") create_presence(role1, place1) set_playing(role1) create_narrative(place1, "Narrative 1 from Role 1 at Place 1") rset = get_first_narrative() self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 1) narrative = rset["narrative"][0] self.assertEqual(narrative["role"], "Role 1") self.assertEqual(narrative["place"], "Place 1") self.assertEqual(narrative["body"], "Narrative 1 from Role 1 at Place 1") self.assertIsNotNone(narrative["created"]) self.assertIsNone(narrative["updated"]) save_narrative(narrative["id"], "Narrative 1 from Role 1 at Place 1 - updated") rset = get_first_narrative() self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 1) narrative = rset["narrative"][0] self.assertEqual(narrative["role"], "Role 1") self.assertEqual(narrative["place"], "Place 1") self.assertEqual(narrative["body"], "Narrative 1 from Role 1 at Place 1 - updated") self.assertIsNotNone(narrative["created"]) self.assertIsNotNone(narrative["updated"])
def testDeleteUnoccupiedPlace(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") place2 = create_place("Test Place 2") role1 = create_role("Test Role 1") create_presence(role1, place2) delete_place(place1)
def testSavePresence(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") role1 = create_role("Test Role 1") create_presence(role1, place1) save_presence(role1, place1, ["broken nose", "busted leg"]) self.assertEqual(2, len(get_roles_in_place(place1)[0].get("damage")))
def testCreatePresenceInSomeoneElsesPlace(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") logoutCurrentUser() setCurrentUser("*****@*****.**", "User2") register_new_user('User2') role1 = create_role("Test Role 1") self.assertRaises(DataModelException, create_presence, role1, place1)
def testDeleteRoleWithPresence(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") self.assertEqual(len(get_roles_in_place(place1)), 0) role1 = create_role("Test Role 1") create_presence(role1, place1) self.assertEqual(len(get_roles_in_place(place1)), 1) self.assertRaises(ActionException, delete_role, role1)
def testDeleteAnotherUsersRole(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') role1url = create_role("Test Role 1") logoutCurrentUser() setCurrentUser("*****@*****.**", "User2") register_new_user('User2') self.assertEquals(len(get_my_roles()), 1) self.assertRaises(DataModelException, delete_place, role1url)
def testWritableProperty(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') create_role("Role 1") myroles1 = get_my_roles() self.assertTrue(myroles1[0].get("writable")) self.assertTrue(myroles1[1].get("writable")) place1 = create_place("Place 1") create_presence(myroles1[1]["id"], place1) logoutCurrentUser() setCurrentUser("*****@*****.**", "User2") register_new_user('User2') create_role("Role 2") myroles2 = get_my_roles() self.assertTrue(myroles2[0].get("writable")) self.assertTrue(myroles2[1].get("writable")) myroles3 = get_roles_in_place(place1) self.assertFalse(myroles3[0].get("writable"))
def testDeletePresenceFromSomeoneElsesPlace(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") role1 = create_role("Test Role 1") pres1 = create_presence(role1, place1) logoutCurrentUser() setCurrentUser("*****@*****.**", "User2") register_new_user('User2') self.assertRaises(ActionException, delete_presence, role1, place1)
def testSaveAnotherUsersRole(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') roleid = create_role("Test Role 1") save_role(roleid, "New Role name") logoutCurrentUser() setCurrentUser("*****@*****.**", "User2") self.assertRaises(DataModelException, save_role, roleid, "Should Fail") register_new_user('User2') self.assertRaises(DataModelException, save_role, roleid, "Should Fail")
def testCreatePresence(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") place2 = create_place("Test Place 2") role1 = create_role("Test Role 1") self.assertEqual(len(get_roles_in_place(place1)), 0) self.assertEqual(len(get_roles_in_place(place2)), 0) create_presence(role1, place1) self.assertEqual(len(get_roles_in_place(place1)), 1) self.assertEqual(len(get_roles_in_place(place2)), 0) create_presence(role1, place2) self.assertEqual(len(get_roles_in_place(place1)), 1) self.assertEqual(len(get_roles_in_place(place2)), 1)
def testCreateNarrative(self): logoutCurrentUser() setCurrentUser("*****@*****.**", "User1") user1 = register_new_user('User1') place1 = create_place("Place 1") role1 = create_role("Role 1") place2 = create_place("Place 2") role2 = create_role("Role 2") create_presence(role1, place1) set_playing(role1) create_narrative(place1, "Narrative 1 from Role 1 at Place 1") create_presence(role2, place2) set_playing(role2) create_narrative(place2, "Narrative 1 from Role 2 at Place 2") create_presence(role2, place1) set_playing(role1) create_narrative(place1, "Narrative 2 from Role 1 at Place 1") create_presence(role1, place2) set_playing(role2) create_narrative(place2, "Narrative 2 from Role 2 at Place 2") delete_presence(role2, place1) set_playing(role1) create_narrative(place1, "Narrative 3 from Role 1 at Place 1") delete_presence(role1, place2) set_playing(role2) create_narrative(place2, "Narrative 3 from Role 2 at Place 2") set_playing(role1) rset = get_first_narrative() self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 4) set_playing(role2) rset = get_first_narrative() self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 4) set_playing(user1) rset = get_first_narrative() self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 6) setCurrentUser("*****@*****.**", "Administrator") rset = get_first_narrative() self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 0) setCurrentUser("*****@*****.**", "User1") create_presence(role1, place2) create_presence(role2, place1) set_playing(role1) create_narrative(place1, "Narrative 4 from Role 1 at Place 1") create_narrative(place1, "Narrative 5 from Role 1 at Place 1") create_narrative(place1, "Narrative 6 from Role 1 at Place 1") create_narrative(place1, "Narrative 7 from Role 1 at Place 1") create_narrative(place1, "Narrative 8 from Role 1 at Place 1") create_narrative(place1, "Narrative 9 from Role 1 at Place 1") set_playing(role2) create_narrative(place2, "Narrative 4 from Role 2 at Place 2") create_narrative(place2, "Narrative 5 from Role 2 at Place 2") create_narrative(place2, "Narrative 6 from Role 2 at Place 2") create_narrative(place2, "Narrative 7 from Role 2 at Place 2") create_narrative(place2, "Narrative 8 from Role 2 at Place 2") create_narrative(place2, "Narrative 9 from Role 2 at Place 2") set_playing(role1) rset = get_first_narrative() self.assertEqual(rset["more"], True) self.assertEqual(len(rset["narrative"]), 10) rset = get_more_narrative(rset["next"]) self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 6) set_playing(role2) rset = get_first_narrative() self.assertEqual(rset["more"], True) self.assertEqual(len(rset["narrative"]), 10) rset = get_more_narrative(rset["next"]) self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 6) set_playing(user1) rset = get_first_narrative() self.assertEqual(rset["more"], True) self.assertEqual(len(rset["narrative"]), 10) rset = get_more_narrative(rset["next"]) self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 8) setCurrentUser("*****@*****.**", "Administrator") rset = get_first_narrative() self.assertEqual(rset["more"], False) self.assertEqual(len(rset["narrative"]), 0)
def testSaveRoleName(self): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') roleid = create_role("Test Role 1") save_role(roleid, "New Role name") self.assertEqual(get_my_roles()[0]["name"], "New Role name")
def testNonAdminCreateRole(self): logoutCurrentUser() setCurrentUser("*****@*****.**", "User1") register_new_user('User1') self.assertIsInstance(create_role("Test Role 1"), str)
def testAdminCreateRole(self): self.assertIsInstance(create_role("Test Role 1"), str)
def setupLotsOfRoles(): setCurrentUser("*****@*****.**", "User1") register_new_user('User1') place1 = create_place("Test Place 1") role1 = create_role("Test Role 1") create_presence(role1, place1) role2 = create_role("Test Role 2") create_presence(role2, place1) create_role("Test Role 3") create_role("Test Role 4") create_role("Test Role 5") create_role("Test Role 6") create_role("Test Role 7") create_role("Test Role 8") create_role("Test Role 9") create_role("Test Role 10") create_role("Test Role 11") create_role("Test Role 12") create_role("Test Role 13") create_role("Test Role 14") setCurrentUser("*****@*****.**", "User2") register_new_user('User2') create_place("Test Place 2") create_role("Test Role 15") create_role("Test Role 16") return place1