def test_authenticate(self):
r = User.register(username="******",
password="******",
email="*****@*****.**")
db.session.add(r)
db.session.commit()
u = User.authenticate(username="******", password="******")
self.assertEqual(r.username, "test1")
self.assertEqual(u.username, "test1")
def test_playlist_song(self):
u = User(username="******", password="******", email="*****@*****.**")
db.session.add(u)
db.session.commit()
s = Song(title='Test Song',
artist='Testerson',
track_id="testid1235432")
db.session.add(s)
db.session.commit()
p = Playlist(name='Test Playlist',
description="Testing",
username=u.username)
db.session.add(p)
db.session.commit()
ps = PlaylistSong(song_id=s.id, playlist_id=p.id)
db.session.add(ps)
db.session.commit()
self.assertEqual(ps.song_id, s.id)
self.assertEqual(ps.playlist_id, p.id)
def register_user():
if g.user:
flash("You are already registered", "danger")
return redirect("/")
form = UserForm()
if form.validate_on_submit():
try:
user = User.register(username=form.username.data,
password=form.password.data,
email=form.email.data)
db.session.add(user)
db.session.commit()
except IntegrityError:
form.username.errors.append("Username/Email is already taken")
return render_template('register.html', form=form)
do_login(user)
flash('Welcome! Account created successfully!', 'success')
return redirect('/')
else:
return render_template('register.html', form=form)
def test_create_user(self):
u = User.register(username="******",
password="******",
email="*****@*****.**")
db.session.add(u)
db.session.commit()
self.assertEqual(u.username, "test1")
def setUp(self):
"""Add sample users."""
User.query.delete()
u1 = User.register(username="******",
password="******",
email="*****@*****.**")
u2 = User(username="******",
password="******",
email="*****@*****.**",
is_admin=True)
db.session.add_all([u1, u2])
db.session.commit()
self.user_id = u1.id
self.username = u1.username
def test_create_playlist(self):
u = User.register(username="******",
password="******",
email="*****@*****.**")
db.session.add(u)
db.session.commit()
p = Playlist(name='Test Playlist',
description="Testing",
username=u.username)
db.session.add(p)
db.session.commit()
self.assertEqual(p.name, 'Test Playlist')
self.assertEqual(p.description, 'Testing')
self.assertEqual(p.username, 'test1')
def login_user():
if g.user:
flash("You are already logged in", "danger")
return redirect("/")
form = LoginForm()
if form.validate_on_submit():
user = User.authenticate(form.username.data, form.password.data)
if user:
do_login(user)
flash(f"Welcome back, {user.username}!", "success")
if user.is_admin:
session["is_admin"] = True
else:
session["is_admin"] = False
return redirect('/')
else:
form.username.errors = ['Invalid username/password']
return render_template('login.html', form=form)