def test_get_current():
email = get_random_email()
create_user(email, 'pass')
token = login_user(email, 'pass')
user = get_user(token=token)
assert user['email'] == email
assert user['balance'] == 2.5
def test_send_sms():
email = get_random_email()
create_user(email, 'pass')
token = login_user(email, 'pass')
verify_user(token)
send_sms(token, '+123', 'text', 'god')
last_message = query_last_message('+123')
assert last_message['text'] == 'text'
assert last_message['from'] == 'god'
def test_balance_sms_send_balance_depleted():
email = get_random_email()
create_user(email, 'pass')
token = login_user(email, 'pass')
verify_user(token)
for dummy in xrange(5):
send_sms(token, '+123', 'text', 'god')
with pytest.raises(requests.HTTPError):
send_sms(token, '+123', 'text', 'god')
def register(self, form): """ Implement user-registration logic here. """ User = UserModel() user = User.objects.create_user( username = form.cleaned_data['username'], first_name = form.cleaned_data['first_name'], last_name = form.cleaned_data['last_name'], email=form.cleaned_data['email'], password=form.cleaned_data['password1'] ) user.groups.add(Group.objects.get(name='account_holder')) Client.objects.create( user=user, cpf = form.cleaned_data['cpf'], created_at = models.DateTimeField(auto_now_add=True) ) login_user(self.request, user)
def login():
if request.method == 'GET':
return render_template('login.html')
else:
username = request.form['username'].lower()
password = request.form['password']
log = login_user(username, password)
if log == True:
session['username'] = username
return redirect(url_for('home'))
else:
return render_template("login.html", error="Wrong username and/or password.")
def login():
'''
* GET request returns the login page.
POST request logs the user in if inputed credentials match
db records.
\n Args:
1. user inputs (str): email & password taken in from
the login form.
\n Returns:
* If GET: Renders the login template.
* If POST: succesful, then logs the user in, creates a new
session and redirects the user to their profile page.
* IF POST: unsuccessful, then displays a flash message to the
user to tell them why, and reloads the login template.
'''
if request.method == "POST":
email = request.form.get("email").lower()
password = request.form.get("password")
url = login_user(email, password, mongo)
return url
return render_template("login.html")
def login_with_id(request, userId):
user = get_object_or_404(User, pk=userId)
return login_user(request, user)
def login_with_username(request, username):
user = get_object_or_404(User, username=username)
return login_user(request, user)
def login_with_email(request, email):
user = get_object_or_404(User, email=email)
return login_user(request, user)
def login_with_id(request, userId):
user = get_object_or_404(User, pk=userId)
return login_user(request, user)
def login_with_username(request, username):
user = get_object_or_404(User, username=username)
return login_user(request, user)
def login_with_email(request, email):
user = get_object_or_404(User, email=email)
return login_user(request, user)
def login_with_id(request, userId):
if isinstance(userId, int):
return HttpResponseBadRequest('userId must be an integer value.')
user = get_object_or_404(User, pk=userId)
return login_user(request, user)
def test_login():
email = get_random_email()
create_user(email, 'pass')
token = login_user(email, 'pass')
token = refresh_token(token)
def test_token_guard():
email = get_random_email()
create_user(email, 'pass')
token = login_user(email, 'pass')
with pytest.raises(HTTPError):
token = refresh_token(token + '1')
def test_send_sms_without_verify_throws_exception():
email = get_random_email()
create_user(email, 'pass')
token = login_user(email, 'pass')
with pytest.raises(requests.HTTPError):
send_sms(token, '+123', 'text', 'god')
def test_login_with_bad_password():
email = get_random_email()
create_user(email, 'pass')
with pytest.raises(HTTPError):
login_user(email, 'wrong_pass')
