Как дать sys.argv при использовании pytest и unittest.TestCase - PullRequest
Новая
       44

Как дать sys.argv при использовании pytest и unittest.TestCase

1 голос
/ 30 апреля 2020

У меня есть тестовый файл, который запускает некоторый двоичный исполняемый файл и проверяет вывод. Тест выглядит так: 

#! /usr/bin/env python
from subprocess import Popen, PIPE
import unittest
import sys

class MyTest(unittest.TestCase):
    PATH = './app'

    def setUp(self):
        pass

    def tearDown(self):
        pass

    @staticmethod
    def run_subprocess(cmd):
        process = Popen(cmd, stdout=PIPE, stderr=PIPE)
        stdout, stderr = process.communicate()
        return stdout.decode('ascii'), stderr

    def test_case1(self):
        stdout, stderr = self.run_subprocess([self.PATH, some_cmd])

    def test_case2(self):
        stdout, stderr = self.run_subprocess([self.PATH, some_other_cmd])



if __name__ =="__main__":
    if len(sys.argv) < 2:
        raise AttributeError("Usage: python run_test.py app_path")
    TestBingbangboom.PATH = sys.argv[1]
    unittest.main(argv=['first-arg-is-ignored'], exit=False)

Я могу запустить тест с $[python run_test.py app_path. Но как я могу сделать то же самое с pytest: 

$pytest app_path

1 Ответ

1 голос
/ 30 апреля 2020
Бегуны

pytest и unittest совершенно разные, поэтому вам придется добавить код pytest -specifi c для определения и чтения аргумента командной строки и присвоения его классу MyTest. Вот пример решения: в каталоге вашего проекта / теста root создайте файл с именем conftest.py со следующим содержимым: 

import pytest


def pytest_addoption(parser):
    parser.addoption("--executable", action="store", default="./app")


@pytest.fixture(scope="class", autouse=True)
def pass_executable(request):
    try:
        request.cls.PATH = request.config.getoption("--executable")
    except AttributeError:
        pass

Теперь при запуске, например, pytest --executable=path/to/my/binary, path/to/my/binary будет быть установлен на MyTest.PATH.

...