Написать пост

Pytest-фикстуры на человеческом

Аватарка пользователя Aleksei Koledachkin

Как понять фикстуры и как начать их использовать в своих проектах. Обьясняю эту тему начинающим, ибо именно в ней часто случаются сложности.

Привет сообщество,

Я хотел бы поделиться с Вами своим обьяснением того, как понять фикстуры и как начать их использовать в своих проектах, тем самым начать радоваться жизни)

Вероятно, даже продвинутый QA Automation найдет что-то новое, но моя цель обьяснить на пальцах эту тему начинающим, ибо именно в ней зачастую происходят затыки.

Файл conftest.py

Это специальный pytest файл, в который он заглядывает еще перед тем как запустить тесты. Поэтому в основном он используется для создания фикстур внутри нашего проекта.

Данный файл обычно находиться в корне проекта, возьмем к примеру базовую структуру PageObject:

Pytest-фикстуры на человеческом 1

Что такое фикстура

Фикстура – это ресурс или объект, который можно рассматривать как набор условий или предопределенное состояние, необходимое тесту для правильного выполнения, зачастую фикстуры создаются, чтобы генерировать какие-то данные еще до теста и возвращать их для использования в тесте или перед тестом, например:

  1. Создавать подключение к базе данных перед тестом и отключаться от нее после завершения теста
  2. Инициализировать драйвер-браузера и закрывать сессию после завершения теста
  3. Авторизовываться перед запуском теста и не тратить время на логин
  4. Создавать новый аккаунт перед тестом, использовать его в тесте и по завершению теста удалять его т.д

Так же важно знать, что фикстуры могут быть общими для нескольких тестов.

Использование фикстур через return и ее передача в тест в качестве аргумента

Начнем с простого примера:

			def connect_to_database():
    connection = "Соединение с базой данных установлено"
    return connection
		

Как Вы можете видеть, у нас есть максимально простая функция, она возвращает образно, подключение к базе данных.

Что, если мы хотим использовать ее, как обертку для теста, т.е подключаться к базе данных еще перед тестом, а уже в тесте использовать эти данные?

В голову приходит идея импортировать эту функцию и вызывать ее в каждом тесте или сложнее, сделать из этой функции декоратор, обертку для тестов (как раз то, что нам нужно).

Тк вот в этом нам поможет фикстурирование, т.е превращение нашей функции в фикстуру (по сути так называется декоратор, но в контексте pytest)

Для того, чтобы функцию зарегистрировать как фикстуру, в pytest есть специальный маркер (декоратор) @pytest.fixture, его нужно прописать над нужной функцией.

Напомню все общие фикстуры мы пишем в файле conftest.py, они будут видны всем тестовым классам по умолчанию.

			import pytest

@pytest.fixture # Таким образом функция будет восприниматься, как фикстура
def connect_to_database():
    connection = "Соединение с базой данных установлено"
    return connection
		

Но как это использовать? Есть несколько путей, мы начнем с самого простого, а именно будем прокидывать фикстуру в качестве аргумента внутрь нашего теста / тестового метода.

			import pytest

@pytest.fixture # Таким образом функция будет восприниматься как фикстура
def connect_to_database():
    connection = "Соединение с базой данных установлено"
    return connection
		
			import pytest

class TestExample:
		def test_1(self, connect_to_database): # Прокидываем фикстуру в тест и она выполнится перед тестом
				print(connect_to_database) # Используем фикстуру, она вернет нам текст с подключением к БД
				self.driver.get("https://ya.ru")
		

В итоге перед запуском теста, в консоли мы увидим то самое сообщение «Соединение с базой данных установлено», что будет означать выполнение подключения к БД или любого другого кода перед тестом.

Как вымогли заметить, никаких импортов нет, тестовые файлы автоматически находят фикстуру изconftest.py без каких-либо импортов, достаточно пробросить фикстуру вкачестве аргумента втестовый метод.

Но что, если фикстура возвращает несколько объектов, например, сгенерированный логин и пароль, то как их использовать в тесте?

			import pytest
import time

@pytest.fixture
def generate_data():
    login = f"autotest_{time.time()}@hyper.org" # Генерирует логин
    password = "512" # Назначает пароль
		return login, password # Возвращает логин и пароль
		

Исходя из того, как работает python, мы знаем, что он в данном случае вернет кортеж данных:

			("autotest_12399412@hyper.org", "512") # логин, пароль
		

Соответственно, чтобы передать данные в тест, нужно так же передать фикстуру в качестве аргумента и обращаться к элементам кортежа по индексу:

			import pytest

class TestExample:
	    def test_1(self, generate_data): # Прокидываем фикстуру в тест и она выполнится перед тестом
                login = generate_data[0] # Записываем в переменную сгенерированный логин
                password = generate_data[1] # Записываем в переменную полученный логин
        ...
		

Все это классно, но не совсем удобно, потому что, чтобы узнать к чему мы обращаемся, нужно идти в саму фикстуру, так как из строчки generate_dataничего не будет понятно.

Выход следующий, можно возвращать значения в виде словаря и обращаться к элементам по ключу:

			import pytest
import time

@pytest.fixture
def generate_data():
    login = f"autotest_{time.time()}@hyper.org" # Генерирует логин
    password = "512" # Назначает пароль
		return {"login":login, "password": password} # Возвращает логин и пароль
		
			import pytest

class TestExample:
		def test_1(self, generate_data): # Прокидываем фикстуру в тест и она выполнится перед тестом
        login = generate_data[0] # Записываем в переменную сгенерированный логин
        password = generate_data[1] # Записываем в переменную полученный логин
        ...
		

Вот теперь стало намного понятнее)

Использование фикстур через request.cls и ее вызов с помощью маркера (декоратора)

Следующий способ использования фикстур — это request.cls

Постараюсь объяснить максимально простым языком, для этого возьмем тот же пример с генерацией логина и пароля:

  1. В качестве аргумента внутрь нашей фикстуры прокидываем request
  2. Назначим переменную с помощью request.cls.имя_переменной
  3. Суть в том, что, когда вы объявляете переменную через request.cls.имя_переменной, в тестовом классе автоматически будет создаваться такой атрибут. Это то же самое, что если бы вы напрямую объявили их в init тестового класса: test_example.pyrequest
			import pytest
import time

@pytest.fixture
def generate_data(request):
    request.cls.login = f"autotest_{time.time()}@hyper.org" # Генерирует логин
    request.cls.password = "512" # Назначает пароль
		
  • request — создать
  • cls — внутри класса
  • имя_переменной — атрибут класса
			создать.внутри_класса.имя_переменной = request.cls.login
		
Вслучае использования request.cls, оператор return прописывать ненужно, новсегда нужно передавать вфикстуру request.

Теперь переходим к этапу использования. В случае, если вы используете вышеописанный метод, то не нужно прокидывать фикстуру в тест в качестве аргумента, тут все интереснее, мы будем использовать специальный pytest-маркер (декоратор):

  • @pytest.mark.usefixtures(«имя фикстуры») — маркер (декоратор) для вызова фикстуры

Маркер (декоратор) прописывается либо над нужным тестом, в случае, если вы хотите генерировать данные из фикстуры только для одного теста, или над классом, если хотите генерировать новые данные для каждого теста.

			import pytest
import time

@pytest.fixture
def generate_data(request):
    request.cls.login = f"autotest_{time.time()}@hyper.org" # Генерирует логин
    request.cls.password = "512" # Назначает пароль
		
			import pytest

@pytest.mark.usefixtures("generate_data") # Вызываем фикстуру над классом
class TestExample:

		def test_1(self):
		    print(self.login) # Сюда передастся логин
		    print(self.password) # А сюда пароль
		
Если вывызвали фикстуру через маркер (декоратор) ииспользовали request.cls для создания данных вфикстуре, тологин ипароль будут доступны через self параметр напрямую!

Надеюсь разобрались) Теперь по аналогии сделаем фикстуру для инициализации драйвера:

			import pytest
import time

from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.service import Service

@pytest.fixture
def get_driver(request):
    service = Service(ChromeDriverManager().install())
    driver = webdriver.Chrome(service=service)
    request.cls.driver = driver
		

Теперь, вызвав с помощью маркера (декоратора) фикстуру непосредственно в тесте, вы можете спокойно обращаться к драйверу и его методам через self.driver.

			@pytest.mark.usefixtures**("get_driver") # Вызываем фикстуру над тестом
class TestExample:

		def test_1(self):
		    self.driver.get("https://yandex.ru") # Работаем с полученным обьектом драйвера
		

Автоматическое использование фикстур

Тут все просто, у фикстуры есть параметр:

  • autouse — данный параметр принимает в качестве значения True или False

В случае, если выставить autouse=True, фикстура будет вызываться абсолютно для каждого теста в проекте автоматически, без явного ее вызова где-либо.

			import pytest
import time

from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.service import Service


@pytest.fixture(autouse=True) # Пока не обращаем внимания на эту строку
def get_driver(request):
    service = Service(ChromeDriverManager().install())
    driver = webdriver.Chrome(service=service)
    request.cls.driver = driver
		
			import pytest

class TestExample:

		def test_1(self):
		    self.driver.get("https://vk.com") # И никаких явных вызовов фикстуры
		

Соответсвенно, чтобы избежать проблем и конфликтов между разными фикстурами, я рекомендую использовать этот атрибут исключительно для инициализации драйвера.

Пред- и постусловия в фикстурах

Для того, чтобы фикстура делала что-то перед тестом, например, инициализация драйвера, и после теста, например, закрытие браузера, существует специальная функция:

  • yield — это разделитель, все, что написано над ним, будет исполнено до теста, все, что ниже — после теста.
			import pytest
import time

from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.service import Service


@pytest.fixture(autouse=True)
def get_driver(request):
    service = Service(ChromeDriverManager().install())
    driver = webdriver.Chrome(service=service)
    request.cls.driver = driver
		yield
    driver.quit()
		

Если мы вспомним как создается стандартный декоратор, то он принимает в себя функцию. И все что над ней — это предусловия, а то, что после — это постусловия.

			def decorator(test): # Принимает в себя функцию
		def wrapper():
        print("Перед тестом") # Предусловия
        test() # Функция которую декарируем
        print("После теста") # Постусловия
		return wrapper

@decorator # Применяем декоратор
def hello():
    print("Привет")
		

Соответственно, все аналогично, во время запуска тестов, в фикстуру вместо yield подставляется наш тест.

Область видимости фикстур

Область видимости определяет, как применяется фикстура, для всего набора тестов (для тестового класса) единожды или для каждого теста (тестового метода) отдельно.

За область видимости отвечает параметр фикстуры:

  • scope — определяет область применения фикстуры

В целом, для 99% случаев, достаточно знать лишь 2 области видимости:

1. scope=»class» — фикстура будет вызвана один раз для всех тестов внутри тестового класса.

— Пример 1: Возьмем фикстуру инициализации драйвера

			import pytest
import time

from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.service import Service


@pytest.fixture(autouse=True, scope="class")
def get_driver(request):
    service = Service(ChromeDriverManager().install())
    driver = webdriver.Chrome(service=service)
    request.cls.driver = driver
		yield
    driver.quit()
		

И для примера сделаем 2 теста

			import pytest

class TestExample:

		def test_1(self):
        self.driver.get("https://vk.com")
		
		def test_2(self):
        self.driver.get("https://ya.ru")
		

Тогда браузер откроется, и в этом же окне будут проходить все тесты. Браузер закроется только тогда, когда пройдут все тесты, а это уже, как минимум, не позволит нам запустить тесты параллельно, ибо они будут друг друга перебивать.

— Пример 2: Генерация логина

			import pytest
import time

@pytest.fixture(scope="class")
def generate_login(request):
    request.cls.login = f"autotest_{time.time()}@hyper.org" # Генерирует логин
		

При использовании фикстуры, логин сгенирируется один раз и будет одинаковым для всех тестов внутри класса.

			import pytest

@pytest.mark.usefixtures("generate_login") # Вызов фикстуры генерации логина
class TestExample:
		
		def test_1(self):
        print(self.login)
		
		def test_2(self):
        print(self.login)
		

В результате, мы получим один и тот же логин в каждом тесте. Это может помочь в зависимых тестах, но тем не менее, атомарность — залог качественных тестов. Тут мы и подходим ко второй области видимости.

2. scope=»function» — фикстура будет вызываться для каждого теста по отдельности, атомарно.

Это означает, что для каждого теста, в случае фикстуры с драйвером например, будет открываться новая сессия браузера, а при генерации данных, для каждого теста будут генерироваться новые данные.

— Пример 1: Возьмем фикстуру инициализации драйвера

			import pytest
import time

from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.service import Service


@pytest.fixture(autouse=True, scope="function")
def get_driver(request):
    service = Service(ChromeDriverManager().install())
    driver = webdriver.Chrome(service=service)
    request.cls.driver = driver
		yield
    driver.quit()
		

И для примера сделаем 2 теста

			import pytest

class TestExample:

		def test_1(self):
        self.driver.get("https://vk.com")
		
		def test_2(self):
        self.driver.get("https://ya.ru")
		

Тогда браузер будет инициализироваться для каждого теста отдельно, что позволит запускать тесты параллельно, ну или просто не сталкивать их друг с другом и в прямом смысле запускать их атомарно.

— Пример 2: Генерация логина

			import pytest
import time

@pytest.fixture(scope="function")
def generate_login(request):
    request.cls.login = f"autotest_{time.time()}@hyper.org" # Генерирует логин
		

При использовании фикстуры, будет генирироваться новый логин для каждого теста

			import pytest

@pytest.mark.usefixtures("generate_login") # Вызов фикстуры генерации логина
class TestExample:

		def test_1(self):
        print(self.login)
		
		def test_2(self):
        print(self.login)
		

В результате, полностью разные данные в каждом новом тесте.

Заключение

Я надеюсь, что данная статься наконец помогла Вам разобраться с тем, как устроены фикстуры и как с ними работать.

В статье я использовал максимально человеческий язык, избегая заумных технических терминов, ибо куда важнее объяснить так, чтобы человек разобрался, а не так чтобы он прочитав пол статьи бросил ее и пошел искать другую.

Следите за новыми постами
Следите за новыми постами по любимым темам
16К открытий17К показов