选择文件名取决于运行pytest的命令行中的参数

Question

I have a project with structure the same as this: https://github.com/RomanBaggins/Test_automatisation_final_project . In conftest.py i get parameters from cmd. In one of the functions in mane_page.py I need to assign a value(= parameter in cmd) to a variable. For example:

cmd: pytest -s -v --language=en --status=empty test_main_page.py

import pytest
from selenium import webdriver
from selenium.webdriver.chrome.options import Options


def pytest_addoption(parser):
    parser.addoption('--language', action='store', default=None, help="Choose language")
    parser.addoption('--status', action='store', default=None, help="Choose status")



@pytest.fixture(scope="function")
def browser(request):
    user_language = request.config.getoption("language")
    if user_language:
        options = Options()
        options.add_experimental_option('prefs', {'intl.accept_languages': user_language})
        browser = webdriver.Chrome(options=options)
        browser.implicitly_wait(5)
        yield browser
    else:
        raise pytest.UsageError("--choose your language again")
    browser.quit()

@pytest.fixture
def choose_status(request):
    return request.config.getoption("status")

Then in main_page.py I want to do this:

from .base_page import BasePage
from selenium.webdriver.common.by import By
from .locators import MainPageLocators

class MainPage(BasePage): 
    def go_to_login_page(self):
        login_link = self.browser.find_element(By.CSS_SELECTOR, "#login_link")
        login_link.click()
        **chosen status = status** <-----------
        def should_be_login_link(self):
        assert self.is_element_present(*MainPageLocators.LOGIN_LINK)

Finally I need this result: chosen_satus = "empty"

How can I do it?

Answer 1

You can send it as a parameter to go_to_login_page() from the test

@pytest.fixture
def choose_status(request):
    return request.config.getoption("status")

def test_main_page(choose_status):
    MainPage().go_to_login_page(choose_status)


class MainPage(BasePage): 
    def go_to_login_page(self, status):
        # status = empty

Answer 2

Here is the final code, that solve my problem: conftest.py

@pytest.fixture
def choose_status(request):
    return request.config.getoption("status")


def test_main_page(choose_status):
    MainPage().go_to_login_page(choose_status)

In main_page.py

class MainPage(BasePage): 
    def go_to_login_page(self,choose_status):
        if choose_status == "clear":
            #do smth

In test_main_page:

def test_guest_can_go_to_login_page(browser,choose_status):
    link = "someurl"

    page = MainPage(browser, link)   
    page.open()                      
    page.go_to_login_page(choose_status)   
    page.should_be_login_link()