在Python单元测试中的每个请求之后模拟清除cookie

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属性来访问饼干：

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

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

Official documentation on CookieJar objects