Тестирование приложения Flask с помощью pytest-flask + pytest-selenium (докер)
Я пытаюсь протестировать веб-приложение Flask в док-контейнере, что является новым для меня. Мой стек следующий:
- Fire Fox
- селен
- 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 не удалось выполнить настройку. Есть лучший способ сделать это?
Теперь, если я заменю 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:
Нужно ли загружать живой сервер по-другому? Или я должен где-нибудь изменить свой хост + порт конфигурации?
2 ответа
Что касается проблемы с прямым вызовом с 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
, Это добавит правильный номер порта автоматически.
Но это не имеет ничего общего с докером, как вы его запускаете?
Что касается проблемы с самим Selenium - я могу представить проблему, связанную с сетями докеров. Как вы используете его? Есть ли у вас, например. docker-compose
конфигурация? Вы можете поделиться этим?
Упомянутый выпуск 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()
( Источник)
Для опоздавших, возможно, стоит взглянуть на Xvfb, и этот учебник может оказаться еще более полезным.
Затем (в оболочке Linux) вы можете ввести:
Xvfb :99 &
export DISPLAY=:99
pytest --driver Firefox --driver-path /usr/local/bin/firefox test_app.py
Это обеспечивает виртуальный буфер кадра (поддельный экран) для приложения и выводит туда весь графический контент.
Обратите внимание, что я не сталкивался с этой проблемой, просто предоставил решение, которое помогло мне преодолеть указанную ошибку с помощью другого приложения.