def test_handle_request(self): # No User or Profile p_count0 = UserProfile.query().count() u_count0 = User.query().count() self.assertEqual(p_count0, 0) self.assertEqual(u_count0, 0) # Create New User provider = 'gmail.com' req = Request.blank('/auth/appengine_openid?provider=' + provider) resp = req.get_response(application) self.assertEqual(resp.location, 'https://www.google.com/accounts/' 'Login?continue=http%3A//localhost/' 'auth/appengine_openid/callback')
def test_post(self): email = '*****@*****.**' password = '******' # No User u_count0 = User.query().count() self.assertEqual(u_count0, 0) # Create New User req = Request.blank('/auth/password', POST={ 'password': password, 'email': email, 'name': 'Kyle Finley', 'givenName': 'Kyle', 'additionalName': '"Danger"', 'familyName': 'Finley', 'gender': 'male', 'affiliations-1.name': 'Aff 1 name', 'affiliations-1.department': 'Aff 1 department', 'affiliations-1.description': 'Aff 1 description', 'affiliations-1.location': 'Aff 1 location', 'affiliations-2.name': 'Aff 2 name', 'affiliations-2.department': 'Aff 2 department', 'affiliations-2.description': 'Aff 2 description', 'affiliations-2.location': 'Aff 2 location', }) req._load_session() resp = req.get_response(application) # Retrieve user from datastore user = User.query().get() p = user.data self.assertEqual(p.name, 'Kyle Finley') self.assertEqual(p.givenName, 'Kyle') self.assertEqual(p.additionalName, '"Danger"') self.assertEqual(p.familyName, 'Finley') self.assertEqual(p.gender, 'male') self.assertEqual(p.affiliations[0].name, 'Aff 1 name') self.assertEqual(p.affiliations[0].department, 'Aff 1 department') self.assertEqual(p.affiliations[0].description, 'Aff 1 description') self.assertEqual(p.affiliations[0].location.name, 'Aff 1 location') self.assertEqual(p.affiliations[1].name, 'Aff 2 name') self.assertEqual(p.affiliations[1].department, 'Aff 2 department') self.assertEqual(p.affiliations[1].description, 'Aff 2 description') self.assertEqual(p.affiliations[1].location.name, 'Aff 2 location') u_count1 = User.query().count() self.assertEqual(p_count1, 1) self.assertEqual(u_count1, 1) # Login User req = Request.blank('/auth/password', POST={'password': password, 'email': email}) resp = req.get_response(application) # Make sure a new User is not created. p_count2 = UserProfile.query().count() u_count2 = User.query().count() self.assertEqual(p_count2, 1) self.assertEqual(u_count2, 1) # Wrong password req = Request.blank('/auth/password', POST={'password': '******', 'email': email}) resp = req.get_response(application)