Pull to refresh

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

Level of difficultyMedium
Reading time9 min
Views24K

Вступление

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

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

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

Файл conftest.py

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

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

Базовый PageObject
Базовый PageObject

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

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

  • Создавать подключение к базе данных перед тестом и отключаться от нее после завершения теста

  • Инициализировать драйвер-браузера и закрывать сессию после завершения теста

  • Авторизовываться перед запуском теста и не тратить время на логин

  • Создавать новый аккаунт перед тестом, использовать его в тесте и по завершению теста удалять его т.д

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

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

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

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

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

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

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

В этом нам поможет фикстурирование, т.е превращение нашей функции в фикстуру.

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

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

import pytest


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

conftest.py

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

import pytest


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

conftest.py

import pytest


class TestExample:

    def test_1(self, connect_to_database): # Прокидываем фикстуру в тест и она выполнится перед тестом
          print(connect_to_database) # Используем фикстуру, она вернет нам текст с подключением к БД

test_example.py

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

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

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

import pytest
import time


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

conftest.py

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

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

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

import pytest


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

test_example.py

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

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

import pytest
import time


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

conftest.py

import pytest


class TestExample:

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

test_example.py

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

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

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

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

  1. В качестве аргумента внутрь нашей фикстуры прокидываем request

  2. Назначим переменную с помощью request.cls.имя_переменной

import pytest
import time


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

conftest.py

Суть в том, что, когда вы объявляете переменную через request.cls.имя_переменной, в тестовом классе автоматически будет создаваться такой атрибут. Это то же самое, что если бы вы напрямую объявили их в init тестового класса:

import pytest


class TestExample:
    # Данный кусок кода = requst.cls.login и request.cls.password в фикстуре
	def __init__(self):
        self.login = f"autotest_{time.time()}@hyper.org"
        self.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" # Назначает пароль

conftest.py

import pytest

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

    def test_1(self):
        print(self.login) # Сюда передастся логин
        print(self.password) # А сюда пароль

test_example.py

Если вы вызвали фикстуру через маркер (декоратор) и использовали 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

conftest.py

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

import pytest


@pytest.mark.usefixtures("get_driver") # Вызываем фикстуру над тестом
class TestExample:
  
	def test_1(self):
	    self.driver.get("https://yandex.ru") # Работаем с полученным обьектом драйвера

test_example.py

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

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

  • 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

conftest.py

import pytest


class TestExample:

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

test_example.py

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

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

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

  • 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()

conftest.py

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

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

example.py

Соответственно, принцип работы фикстур +- аналогичнен, во время запуска тестов, в фикстуру вместо 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()

      conftest.py

      И для примера сделаем 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")

      test_example.py

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

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

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

      conftest.py

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

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

      test_example.py

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

  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()

      conftest.py

      И для примера сделаем 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")

      test_example.py

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

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

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

      conftest.py

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

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

      test_example.py

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

Заключение

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

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

Мой Telegram: https://t.me/qa_proka4

Tags:
Hubs:
Total votes 15: ↑12 and ↓3+14
Comments38

Articles