def test_database_cascading(client):
db.create_user('foobar', 'foobar', '*****@*****.**', False)
u = models.User.get_user_by_name('foobar')
models.Post.create(u.id, 'foo', None)
models.Post.create(u.id, 'foo bar', None)
# Deleting the user needs to delete all depending rows in other tables
u.delete()
assert len(models.Post.get_posts_by_user_id(u.id)) == 0
def addUser():
error = None
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
if db.create_user(username, password):
session['username'] = username
return redirect(url_for('home'))
else:
session.clear()
error = 'Invalid Credentials. Please try again.'
return render_template('login.html', error=error)
def post(self):
api_key = utils.generate_api_key()
args = parser.parse_args()
data = {
"username": args["username"],
"email": args["email"],
"password": args["password"],
"api_key": api_key
}
if db.create_user(data):
res = {
"message": "Success",
"api_key": api_key
}
return jsonify(**res)
return jsonify("Fail")
def test_create_user(self):
res = db.create_user(user_test_data)
self.assertTrue(res)