Question

所以我有网络应用程序，如果用户登录，他无法注册。相反，他被重定向，并且显示他无法注册的消息被闪现。

问题是我出于某种原因无法测试它。

这是视图

@users_template.route('/signup', methods=['GET', 'POST'])
def signup():
    form = RegisterationForm()
    if current_user.is_authenticated and current_user.is_active:  # prevent the user from seeing the register page if he's already logged in
        flash('You cannot sign up while you\'re logged in')
        return redirect(url_for('main.home'))

    if request.method == 'POST' and form.validate_on_submit():
        username = form.username.data
        password = form.password.data
        user = User(username=username, password=password)
        if db.session.query(User).filter(User.username == username).first() is None:
            current_app.logger.info('<{}> did not register before. Registering him/her now'.format(username))
            db.session.add(user)
            db.session.commit()
            flash('Thank you for registering. Please login now')
            return redirect(url_for('users.login'))
        else:  # the user name has been registered before
            flash('You already registered')
    return render_template('signup.html', form=form)

现在我的测试模块中有这个方法

def test_signup_page_requires_logout(self):
    data_to_signup = {'username': 'mustafa1', 'password': 'mustafa1',
            'confirm_password': 'mustafa1'}
    data_to_login = {'username': 'mustafa1', 'password': 'mustafa1'}
    # with statement preserves the request context for the current_user
    with self.client:
        self.client.post('/signup', data=data_to_signup)
        self.client.post('/login', data=data_to_login)
        response = self.client.get('/signup', follow_redirects=True)
        assert b'You cannot sign up while you\'re logged in' in response.data

更令人困惑的是，这种结构适用于其他方法

def test_headquarter_after_login(self):
    data_to_signup = {'username': 'mustafa1', 'password': 'mustafa1',
            'confirm_password': 'mustafa1'}
    data_to_login = {'username': 'mustafa1', 'password': 'mustafa1'}
    with self.client:
        self.client.post('/signup', data=data_to_signup)
        self.client.post('/login', data=data_to_login)
        response = self.client.get('/headquarter')
        assert b'Headquarter page' in response.data

编辑：这是登录视图

from .. import bcrypt
@users_template.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm(request.form)
    if form.validate_on_submit() and request.method == 'POST':
        username = form.username.data
        password = form.password.data

        user = User.query.filter(User.username == username).first()

        if user is not None and bcrypt.check_password_hash(user.password, password):  # authenticate
            login_user(user)  # then log the user in
            flash('You are logged in')
            return redirect(url_for('main.home'))

        else:
            flash('Incorrect username or password')  # if username is wrong, it will be caught by the error handling above
    return render_template('login.html', form=form)  # for a GET request

这是错误（我正在使用py.test fyi）

            assert current_user.is_active() == True
            assert current_user.is_authenticated() == True
>           assert b'You cannot sign up while you\'re logged in' in response.data
E           AssertionError: assert b"You cannot sign up while you're logged in" in b'<!DOCTYPE html>\n<html lang="en">\n  <head>\n    <meta charset="utf-8">\n    <meta http-equiv="X-UA-Compatible" cont...jquery/1.12.4/jquery.min.js"></script>\n    <script src="/static/js/bootstrap.min.js"> </script>\n  </body>\n</html>\n'
E            +  where b'<!DOCTYPE html>\n<html lang="en">\n  <head>\n    <meta charset="utf-8">\n    <meta http-equiv="X-UA-Compatible" cont...jquery/1.12.4/jquery.min.js"></script>\n    <script src="/static/js/bootstrap.min.js"> </script>\n  </body>\n</html>\n' = <TestResponse 3874 bytes [200 OK]>.

这是为了在我的所有模板中显示闪烁的消息（它位于我继承的“base.html”中）

    {% if get_flashed_messages() %}
        {% for message in get_flashed_messages() %}
            <div class='container'>
                    <span class='alert alert-info'>
                        <small>{{ message }}</small>
                    </span>
            </div>
        {% endfor %}
    {% endif %}

我使用闪烁的消息进行了更多测试，并且工作正常。像这个用于测试用户注册的那个

assert b'Thank you for registering. Please login now' in response.data

Answer 1

建议号：

由于这一行：
```
if current_user.is_authenticated and current_user.is_active:
```
current_user将is_active标记设置为False。如果我不对，请粘贴测试错误。

另一条线索可能是主页模板中缺少flash消息。将此代码放在例如flashMessages.html：

中

{% macro render_flashMessages() %}
    {% with messages = get_flashed_messages() %}
        {% if messages %}
        <ul class="list-unstyled">
            {% for message in messages %}
            <li><div class="alert alert-info">{{ message|safe }}</div></li>
            {% endfor %}
        </ul>
        {% endif %}
    {% endwith %}
{% endmacro %}

然后您可以在home.html：

中以这种方式使用它

{% from "flashMessages.html" import render_flashMessages %}

{{ render_flashMessages() }}

接下来要检查的是使用assert而不是response.data，如果您使用Flask-WebTest，可以尝试：
```
assert b'You cannot sign up while you\'re logged in' in response.flashes[0][1]
```
flashes的位置：

包含请求期间闪烁的消息的元组（类别，消息）列表。
但如果您不使用FlaskWebTest，则可以查看会话：
```
from flask import session

session['_flashes'][0][1]
```

烧瓶测试始终失败

1 个答案: