def test_provider_query(self):
p = GoogleUserProfile()
qry = GoogleUserProfile.query().fetch()
self.assertEqual(len(qry), 0)
p.put()
# query UserProfile
qry = UserProfile.query().fetch()
self.assertEqual(len(qry), 1)
qry = GoogleUserProfile.query().fetch()
self.assertEqual(len(qry), 1)
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 post(self):
email = self.request.POST.get('email')
if not email:
self.request.add_message(
'Please enter a valid email address', 'error')
return self.get()
existing_account = UserProfile.get_key('password', email).get()
if not existing_account:
self.request.add_message(
'The email address you provided was not found. '
'Please try again.', 'error')
return self.get()
return self.sent(email)
def post(self):
email = self.request.POST.get('email')
if not email or "@" not in email or "." not in email:
self.request.add_message(
'Please enter a valid email address', 'error')
return self.get()
existing_account = UserProfile.get_key('password', email).get()
if not existing_account:
self.request.add_message(
'The email address you provided was not found. '
'Please try again.', 'error')
return self.get()
user = User.get_by_auth_id(existing_account.key.id())
taskqueue.add(url='/account/tasks/password-recovery-email', params={
'recipient_id': user.key.id(),
})
return self.sent(email)
def test_method_call_on_subclass(self):
p = GoogleUserProfile()
p.put()
pq = UserProfile.query().get()
v = pq.test_method_call()
self.assertEqual(v, 'it worked.')
