Python Flask单元测试中的登录测试失败

Question

I am trying to test the login functionality for my Python Flask app.

I create a user and add them to the database in the test_adduser() test and that passed fine. However, the test_login() test fails with an assertion error. I have debugged the code and the test loads the login page ok (so I know the problem is not with the app or the routes etc) but it does not log in and redirect the user.

I realise that this post is a bit short on description ie the error trace, however the error is literally "AssertionError".
Maybe I am doing something obviously wrong, like the way I am passing in the user credentials to the test and someone could point this out to me?

Thanks!

from flask.ext.testing import TestCase
from flask import Flask
from Shares import db, app as appy
from Shares.models import User
import manage


class test(TestCase):

def create_app(self):
    app = Flask(__name__)
    app.config['TESTING'] = True
    return appy

SQLALCHEMY_DATABASE_URI = "sqlite://"
TESTING = True

def setUp(self):

    manage.initdb()
    #print self.login('lucas', 'test').data

def tearDown(self):

    db.session.remove()
    db.drop_all()

def login(self, username, password):
    return self.client.post('/login', data=dict(
        username=username,
        password=password
    ), follow_redirects=True)

def logout(self):
    return self.client.get('/logout', follow_redirects=True)

def test_adduser(self):

    lucas=User(username="lucas", email="lucas@example.com", password="test")
    user2 = User(username="lucas", email="lucas@test.com")

    db.session.add(lucas)
    db.session.commit()

    assert lucas in db.session
    assert user2 not in db.session

def test_login(self):
    lucas=User(username="lucas", email="lucas@example.com", password="test")
    db.session.add(lucas)

    rv = self.login('lucas', 'test')
    assert 'Welcome' in rv.data

Answer 1

Ok so I worked it out in the end. I needed to deactivate the CSRF protection. This did the trick:

def create_app(self):
    app = Flask(__name__)
    appy.config['TESTING'] = True
    appy.config['WTF_CSRF_ENABLED'] = False
    return appy