Тестирование HTTP-запросов — Python: Продвинутое тестирование

Инверсия зависимостей — это крайне мощная техника, которая работает не только с функциями, но и с объектами. В этом уроке мы рассмотрим ее глубже на примере HTTP-запросов. Здесь мы познакомимся с таким понятием как стаб (от английского stub — заглушка).

Предположим, что у нас есть функция, которая анализирует приватные репозитории на GitHub одной конкретной организации. При этом функция возвращает форки — репозитории, отпочкованные от основного репозитория:

# Библиотека для работы с GitHub API
from github import Github

def get_private_fork_names(username):
    client = Github('access_token')
    # Клиент выполняет запрос на GitHub и возвращает
    # список приватных репозиториев указанной организации
    repos = client.get_user(username).get_repos(type='private')
    # Оставляем только имена приватных форков
    return [repo.name for repo in repos if repo.fork == True]

Давайте ее протестируем. Что мы хотим от этой функции? В первую очередь убедиться, что она работает правильно – возвращает массив имен приватных форков. Идеальный тест выглядел бы так:

def test_get_private_fork_names():
    names = get_private_fork_names('hexlet')
    assert names ==  # ожидаемый список имен

К сожалению, не все так просто. Внутри функции выполняется HTTP-запрос. Обсудим, какие проблемы из-за этого могут возникнуть:

1. Нестабильная сеть может тормозить выполнение тестов и приводить к фантомным ошибкам. Тесты будут иногда проходить, иногда нет
2. У github.com и других подобных сервисов установлены временные ограничения на запросы — в секунду, в час и так далее. Скорее всего, тесты начнут упираться в эти лимиты. Есть вероятность, что GitHub заблокирует компьютер, с которого идут запросы
3. Реальные данные на GitHub не статичны, они могут меняться. Это опять же приведет к ошибкам и необходимости править тесты

В этом примере HTTP-запрос воспринимается как помеха к тому, чтобы протестировать нашу основную логику. Мы доверяем PyGithub и его библиотеке github — то есть нам не нужно проверять, что она работает правильно.

Из предыдущего урока мы узнали о нескольких способах выхода из этой ситуации и теперь можем применить один из них.

Инверсия зависимостей

Чтобы использовать инверсию зависимостей, добавим вторым аргументом функции сам клиент github. Так мы сможем подменить его в тестах:

from github import Github

# Клиент для работы с Github передается снаружи, его можно подменить
def get_private_fork_names(username, client=Github()):
    # Логика повторяется, за исключением работы с клиентом
    repos = client.get_user(username).get_repos(type='private')
    return [repo.name for repo in repos if repo.fork == True]

Нам придется реализовать фейковый клиент, который ведет себя примерно так же, как и реальный Github. Разница только в том, что наш клиент не выполняет сетевых запросов.

Также нам нужно описать конкретные данные, которые вернут вызовы get_user() и get_repos(). Только в таком случае мы сможем протестировать, что функция get_private_fork_names() работает корректно:

# Структура этого класса описывает только ту часть,
# которая необходима для вызова client.get_user.get_repos
class GithubFake:
    # Здесь мы описываем желаемые данные, которые должны вернуться в тесте
    def __init__(self, data):
        self.data = data

    # Возвращаем сами себя, чтобы иметь возможность вызвать get_repos() по цепочке
    def get_user(self, name):
        return self

    # Возвращаем переданные в начале данные
    def get_repos(self):
        return self.data

Посмотрим на сам тест с использованием этого клиента:

from github_fake import GithubFake

def test_get_private_fork_names():
    data =  # Данные, которые должен вернуть get_repos
    client = GithubFake(data)
    names = get_private_fork_names('hexlet', client=client)
    assert names ==  # ожидаемый список имен

В тестировании подобные фейковые объекты или функции используются часто. У них есть специальное название – стаб.

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

---

Остались вопросы? Задайте их в разделе «Обсуждение»

Вам ответят команда поддержки Хекслета или другие студенты