def test_getHistory(self):
u = User(username='******', fpl_id=229086)
u.getHistory()
a = [gw for gw in u.fpl_gameweeks]
self.assertEqual(a[0].fpl_points, 96)
self.assertEqual(a[0].user_id, 1)
self.assertLess(len(a), 38)
self.assertEqual(a[0].chip, '3xc')
def test_getPicksHistory(self):
u = User(username='******', fpl_id=229086)
u.getHistory()
u.getPicksHistory()
Player.buildPlayerTable()
a = [pick for pick in u.picks]
self.assertEqual(a[0].element_id, 93)
self.assertEqual(a[7].is_captain, 1)
self.assertEqual(a[0].user, u)
def register():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data,
email=form.email.data,
fpl_id=form.fpl_id.data)
user.set_password(form.password.data)
user.getHistory()
user.getPicksHistory()
db.session.add(user)
db.session.commit()
flash('Congratulations, you are now a registered user!')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
