def post(self):
args = self.reqparse.parse_args()
user = User.query.filter_by(username=args['username']).first()
session.clear()
if user:
if User.verify_password(user, args['password']):
session['user_id'] = user.id
return {'user_id': user.id, 'message': 'Logged in successfully'}, 201
return 'Invalid username or password', 400
def post(self):
args = self.reqparse.parse_args()
user = User.query.filter_by(username=args['username']).first()
session.clear()
if user:
if User.verify_password(user, args['password']):
session['user_id'] = user.id
return {
'user_id': user.id,
'message': 'Logged in successfully'
}, 201
return 'Invalid username or password', 400
def test_create_user(self):
"""create a user in database"""
session.clear()
response = self.client.post(
url_for('users'),
data={
'username': '******',
'email': '*****@*****.**',
'password': '******'
}
)
self.assertEqual(response.status_code, 201)
self.assertIn(b'User created successfully', response.data)
self.assertEqual(db.session.query(User).count(), 2)
def logout():
session.clear()
return redirect("/")
def delete(self):
if 'user_id' in session:
session.clear()
return 'Logged out successfully', 200
return 'You are not logged in', 400
def test_current_user(self):
session.clear()
user = current_user()
self.assertEqual(user, None)
def delete(self):
if 'user_id' in session:
session.clear()
return 'Logged out successfully', 200
return 'You are not logged in', 400
