使用 pytest-flask + pytest-selenium (docker) 测试 Flask 应用程序

Question

我正在尝试在 docker 容器中测试 Flask Web 应用程序，这对我来说是新的。 我的堆栈如下：

• 火狐
• 硒
• pytest-硒
• pytest-烧瓶

这是我的 Flask 应用程序文件：

from flask import Flask

def create_app():
    app = Flask(__name__)
    return app

app = create_app()

@app.route('/')
def index():
    return render_template('index.html')

现在，我的测试文件验证了我的索引页的标题：

import pytest
from app import create_app

# from https://github.com/pytest-dev/pytest-selenium/issues/135
@pytest.fixture
def firefox_options(request, firefox_options):
    firefox_options.add_argument('--headless')
    return firefox_options

# from https://pytest-flask.readthedocs.io/en/latest/tutorial.html#step-2-configure
@pytest.fixture
def app():
    app = create_app()
    return app

# from https://pytest-flask.readthedocs.io/en/latest/features.html#start-live-server-start-live-server-automatically-default
@pytest.mark.usefixtures('live_server')
class TestLiveServer:

    def test_homepage(self, selenium):
        selenium.get('http://0.0.0.0:5000')
        h1 = selenium.find_element_by_tag_name('h1')
        assert h1 == 'title'

当我运行我的测试时：

pytest --driver Firefox --driver-path /usr/local/bin/firefox test_app.py

我收到以下错误（这似乎是由于 Firefox 未处于无头模式）。

 selenium.common.exceptions.WebDriverException: Message: Service /usr/local/bin/firefox unexpectedly exited. Status code was: 1 Error: no DISPLAY environment variable specified

我能够运行firefox --headless但似乎我的 pytest 夹具无法进行设置。 有一个更好的方法吗？

现在，如果我用urlopen替换selenium.get()只是为了尝试正确初始化应用程序及其连接：

def test_homepage(self):
    res = urlopen('http://0.0.0.0:5000')
    assert b'OK' in res.read()
    assert res.code == 200

我收到错误：

urllib.error.URLError:

我需要以不同的方式启动实时服务器吗？ 或者我应该在某处更改我的主机 + 端口配置？

Answer 1

关于使用 urllib 直接调用的问题：

Pytest 的实时服务器默认使用随机端口。 您可以将此参数添加到 pytest 调用中：

--live-server-port 5000

或者没有此参数，您可以直接调用实时服务器，例如：

import pytest
import requests

from flask import url_for


@pytest.mark.usefixtures('live_server')
def test_something():
    r = requests.get(url_for('index', _external=True))
    assert r.status_code == 200

我想你有名为index视图函数。 它会自动添加正确的端口号。

但这与docker没有任何关系，你如何运行它？

关于 Selenium 本身的问题 - 我可以想象与 docker 网络相关的问题。 你如何使用它？ 你有没有例如。 docker-compose配置？ 你能分享一下吗？

Answer 2

引用的 pytest-selenium 问题有：

@pytest.fixture
def firefox_options(firefox_options, pytestconfig):
    if pytestconfig.getoption('headless'):
        firefox_options.add_argument('-headless')
    return firefox_options

注意add_argument() headless前面的- （单破折号add_argument()

（ 来源）

Answer 3

对于后来者，可能值得一看 Xvfb 并且本教程更有帮助

然后（在 Linux shell 中）您可以输入：

Xvfb :99 &
export DISPLAY=:99
pytest --driver Firefox --driver-path /usr/local/bin/firefox test_app.py

这为应用程序提供了一个虚拟帧缓冲区（假屏幕），并在那里输出所有图形内容。

请注意，我没有遇到此问题，只是提供了一个解决方案，帮助我使用另一个应用程序克服了上述错误。