def test_init_initializes_members(self):
user = User('foo', '*****@*****.**')
self.assertEqual(user.name, 'foo')
self.assertEqual(user.email, '*****@*****.**')
self.assertIsNone(user.id)
self.assertEqual(user.library.library, [-1])
self.assertTrue(isinstance(user.created_date, dt))
self.assertIsNone(user.deleted_date)
def test_get_found(self, mock_query):
mock_query.return_value = {
'id': '0000',
'name': 'foo',
'email': '*****@*****.**'
}
user = User.get('0000')
self.assertTrue(isinstance(user, User))
self.assertEqual(user.id, '0000')
def login_callback():
# Get authorization code Google sent back to you
code = request.args.get("code")
# Find out what URL to hit to get tokens that allow you to ask for
# things on behalf of a user
google_provider_cfg = get_google_provider_cfg()
token_endpoint = google_provider_cfg["token_endpoint"]
# Prepare and send a request to get tokens! Yay tokens!
token_url, headers, body = client.prepare_token_request(
token_endpoint,
authorization_response=request.url,
redirect_url=request.base_url,
code=code)
token_response = requests.post(
token_url,
headers=headers,
data=body,
auth=(GOOGLE_CLIENT_ID, GOOGLE_CLIENT_SECRET),
)
# Parse the tokens!
client.parse_request_body_response(json.dumps(token_response.json()))
# Now that you have tokens (yay) let's find and hit the URL
# from Google that gives you the user's profile information,
# including their Google profile image and email
userinfo_endpoint = google_provider_cfg["userinfo_endpoint"]
uri, headers, body = client.add_token(userinfo_endpoint)
userinfo_response = requests.get(uri, headers=headers, data=body)
# You want to make sure their email is verified.
# The user authenticated with Google, authorized your
# app, and now you've verified their email through Google!
if userinfo_response.json().get("email_verified"):
unique_id = userinfo_response.json()["sub"]
users_email = userinfo_response.json()["email"]
picture = userinfo_response.json()["picture"]
users_name = userinfo_response.json()["given_name"]
else:
return "User email not available or not verified by Google.", 400
# Doesn't exist? Add it to the database.
# if not User.get(unique_id):
# user = User(unique_id, users_name, users_email)
# Begin user session by logging the user in
# replace with lookup from Google oauth response
user = User.get_by_email(users_email)
if not user:
# flash something here
return redirect(url_for('auth.login'))
login_user(user, remember=True)
# update user's library with latest catalog
user.library.update(loa.catalog)
return redirect(url_for('home.home'))
def login_post():
# use when AUTH_ENABLED = False else login_callback
user = User.get_by_email(request.form.get('email'))
if not user:
# flash something here
return redirect(url_for('auth.login'))
login_user(user, remember=True)
# update user's library with latest catalog
user.library.update(loa.catalog)
return redirect(url_for('home.home'))
def test_get_not_found(self, mock_query):
mock_query.return_value = None
user = User.get_by_email('foo')
self.assertIsNone(user)
def test_repr_returns_dict(self):
user = User('foo', '*****@*****.**')
repr = user.__repr__()
assert (isinstance(repr, dict))
for f in User.fields:
self.assertTrue(f in repr)
def test_init_missing_required_args_raises(self):
with self.assertRaises(TypeError):
user = User()
def load_user(user_id):
return User.get(user_id)