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)