如何在 Flask 中测试 HTTPS

Question

以下简化代码在我的烧瓶应用中定义。

def select_url_for():
    if request.is_secure:
        return 's3'        
    return 'local'

我尝试过这样的测试，它适用于 HTTP。

def test_select_url_for(self):
    with self.app.test_request_context(): 
        self.assertTrue(select_url_for() == 'local')

如何使用 Flask 对 HTTPS 执行类似的测试？

我找到了this，但问题没有得到解答。我需要在 HTTPS 测试模式下运行 Flask。

Answer 1

request.is_secure 检查wsgi.url_scheme 环境变量。通过使用https 覆盖它，使request.is_secure 返回True：

def test_select_url_for(self):
    with self.app.test_request_context(environ_overrides={
            'wsgi.url_scheme': 'https'
    }):
        self.assertEqual(select_url_for(), 'local')

顺便说一句，使用 assertEqual(a, b) 而不是 assertTrue(a == b)，它会为您提供更具可读性的断言失败消息。