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

时间:2018-09-03 15:59:05

标签: python selenium docker flask pytest-selenium

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

  • firefox
  • pytest-硒
  • pytest-flask

这是我的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固定装置似乎无法完成设置。有更好的方法吗?

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

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:

我是否需要以其他方式启动实时服务器?还是应该在某个地方更改主机+端口配置?

3 个答案:

答案 0 :(得分:1)

所引用的pytest-selenium问题具有:

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

请注意-headless之前的add_argument()(单破折号)

Source

答案 1 :(得分: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配置?你可以分享吗?

答案 2 :(得分:0)

对于后来者来说,值得一看Xvfb,更有用的是tutorial

然后(在Linux shell中)您可以输入:

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

这为应用程序提供了虚拟帧缓冲区(假屏幕),并在那里输出所有图形内容。

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