Question

我需要编写单元测试来确定速率限制方法是否有效，即使用户在每次请求后清除cookie也是如此。现有的速率限制方法使用会话来计算请求。我们的安全专家表示，每次请求后清除cookie都可以轻松绕过这个问题。我不知道如何在单元测试中模拟新会话或清除cookie。服务器是使用Flask的Python2.7。现有代码：

def test_retry_protection(self):
    #
    prev1 = app.config['RETRY_PROTECTION_COUNT']
    app.config['RETRY_PROTECTION_COUNT'] = 3
    prev2 = app.config['RETRY_PROTECTION_SECONDS']
    app.config['RETRY_PROTECTION_SECONDS'] = 5
    username = 'retry.me'
    password = 'ke52u&%y!jfueQ'
    result = self.app.post('/add_password',
                           data=dict(username=username, password1=password,
                                     password2=password))
    assert '": "ok"' in result.data, result.data

    for x in range(app.config['RETRY_PROTECTION_COUNT'] - 1):
        result = self.app.post('/update_password',
                               data=dict(username=username, password1=password,
                                         password2=password, password_old=password + 'wrong'))
        assert 'User and password not correct' in result.data, result.data

    result = self.app.post('/update_password',
                           data=dict(username=username, password1=password + 'change',
                                     password2=password + 'change', password_old=password))
    assert 'Excessive usage - wait 10 minutes to try again' in result.data, result.data
    time.sleep(app.config['RETRY_PROTECTION_SECONDS'] + 1)
    result = self.app.post('/update_password',
                           data=dict(username=username, password1=password + 'change',
                                     password2=password + 'change', password_old=password))
    assert '": "ok"' in result.data, result.data
    app.config['RETRY_PROTECTION_COUNT'] = prev1
    app.config['RETRY_PROTECTION_SECONDS'] = prev2

Answer 1

如果您使用Flask.test_client编写单元测试，则可以使用其cookie_jar属性来访问Cookie：

app = flask.Flask(__name__)
client = app.test_client()

client.cookie_jar.clear()
# or
client.cookie_jar.clear_session_cookies()

Official documentation on CookieJar objects

在Python unittest中的每个请求之后模拟清除cookie

1 个答案: