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