@pytest.fixture()
def login():
    print("this is login fixture")
    user = "chen"
    pwd = 123456
    return user, pwd

def test_login(login):
    """將fixture修飾的login函數(shù)作為參數(shù)傳遞給本用例"""
    print(login)
    assert login[0] == "chen"
    assert login[1] == 123456
    assert "chen" in str(login)

@pytest.fixture()
def user():
    user = "cris"
    return user

@pytest.fixture()
def pwd():
    pwd = "123456"
    return pwd

def test_trans_fixture(user, pwd):
    """同一條用例中傳入多個(gè)fixture函數(shù)"""
    print(user, pwd)
    assert "cris" in str(user)
    assert pwd == "123456"

@pytest.fixture()
def user2():
    user = "cris"
    return user

@pytest.fixture()
def login_info(user2):
    """fixture與fixture函數(shù)之間的相互傳遞"""
    pwd = "e10adc3949ba59abbe56e057f20f883e"
    return user2, pwd

def test_assert_login_info(login_info):
    print(login_info)
    print(type(login_info))
    assert login_info[0] == "cris"
    assert login_info[1] == "e10adc3949ba59abbe56e057f20f883e"

@pytest.fixture()
def run_function():
    print("run before function...")
    yield
    print("run after function...")

def test_run_1(run_function):
    print("case 1")

def test_run_2():
    print("case 2")

def test_run_3(run_function):
    print("case 3")

@pytest.fixture()
def fixture_driver():
    driver = webdriver.Chrome()
    yield driver
    driver.quit()

def test_baidu(fixture_driver):
    driver = fixture_driver
    driver.get("http://www.baidu.com")
    driver.find_element_by_id('kw').send_keys("python fixture")
    driver.find_element_by_id('su').click()

@pytest.fixture
def func_1():
    print("用例前置操作---1")
    yield
    print("用例后置操作---1")

@pytest.fixture
def func_2():
    print("用例前置操作---2")
    yield
    print("用例后置操作---2")

@pytest.fixture
def func_3():
    print("用例前置操作---3")
    yield
    print("用例后置操作---3")

@pytest.mark.usefixtures("func_3")  # 最后執(zhí)行func_3
@pytest.mark.usefixtures("func_2")  # 再執(zhí)行func_1
@pytest.mark.usefixtures("func_1")  # 先執(zhí)行func_1
def test_func():
    print("這是測(cè)試用例")

@pytest.fixture(autouse=True, scope="function")
def func_auto():
    """autouse為T(mén)rue時(shí)，會(huì)作用于每一條用例"""
    print("\n---用例前置操作---")
    yield
    print("---用例后置操作---")

# func_auto函數(shù)的autouse=True時(shí)，無(wú)論是否使用usefixtures引用func_auto，都會(huì)執(zhí)行func_auto
@pytest.mark.usefixtures("func_auto")
def test_01():
    print("case 1")

def test_02():
    print("case 2")

class Test:
    def test_03(self):
        print("case 3")

# @pytest.fixture(scope="function")等價(jià)于@pytest.fixture()
@pytest.fixture(scope="function")
def func_auto():
    """用例級(jí)別fixture，作用域單個(gè)用例"""
    print("\n---function級(jí)別的用例前置操作---")
    yield
    print("---function級(jí)別的用例后置操作---")

# test_01會(huì)引用func_auto函數(shù)，test_02沒(méi)有用修飾器修飾，故不會(huì)引用
def test_func_auto_fixture_1(func_auto):
    print("func 1 print")

def test_func_auto_fixture_2():
    print("func 2 print")

@pytest.fixture(scope="class")
def class_auto():
    """類(lèi)級(jí)別fixture，作用域整個(gè)類(lèi)"""
    print("\n---class級(jí)別的用例前置操作---")
    yield
    print("---class級(jí)別的用例后置操作---")

class TestClassAutoFixture:
    # class級(jí)別的fixture任意一個(gè)用例引用即可
    def test_class_auto_fixture_1(self, class_auto):
        print("class 1 print")

    def test_class_auto_fixture_2(self):
        print("class 1 print")

@pytest.fixture(scope="module")
def module_auto():
    """作用于整個(gè)py文件"""
    print("\n---module級(jí)別的用例前置操作---")
    yield
    print("---module級(jí)別的用例后置操作---")
    
# 測(cè)試類(lèi)外和測(cè)試類(lèi)內(nèi)的函數(shù)方法都調(diào)用了module級(jí)別的fixture，但整個(gè)py文件只會(huì)生效一次fixture。
def test_module_scope_out_class(module_auto):
    print("case scope 01")
    
class TestScope1:
    def test_scope_01(self):
        print("case scope 01")

    def test_scope_02(self, module_auto):
        print("case scope 02")

    def test_scope_03(self):
        print("case scope 03")

# 順序在前面fixture會(huì)先執(zhí)行
def test_scope_01(self, module_auto, class_auto): 
    print("case scope 01")

# 順序在前面fixture會(huì)先執(zhí)行
def test_scope_02(self, module_auto, class_auto, func_auto):  
    print("case scope 02")

@pytest.fixture(scope="session", )
def session_auto():
    """session級(jí)別的fixture,針對(duì)該目錄下的所有用例都生效"""
    print("\n---session級(jí)別的用例前置操作---")
    yield
    print("---session級(jí)別的用例后置操作---")

class TestSessionAutoFixture:
    # session級(jí)別的fixture任意一個(gè)用例引用即可
    def test_session_auto_fixture_1(self, session_auto):
        print("session 1 print")

    def test_session_auto_fixture_2(self):
        print("session 1 print")


def test_session_auto_fixture():
    print("session 1 print")

stars = ["劉德華", "張學(xué)友", "黎明", "郭富城"]
# 利用列表生成式生成一個(gè)用例名稱的列表
ids = [f"test-case-n5n3t3z" for d in range(len(stars))]

@pytest.mark.parametrize("name", stars, ids=ids)
def test_multi_param(name):
    print(f"my name is {name}")

@pytest.fixture(name="rename_get_user_info")
def get_user_info():
    user_name = "周潤(rùn)發(fā)"
    print(user_name)

# 此處需傳入重命名后的fixture函數(shù)名
@pytest.mark.usefixtures("rename_get_user_info")
def test_parametrize_by_use_fixtures():
    """通過(guò)usefixtures裝飾器傳入fixture"""
    print(f"test parametrize use fixtures")
    
def test_parametrize_by_fixture_name(rename_get_user_info):
    """將fixture函數(shù)名作為形參傳入"""
    print(f"test parametrize use fixtures")

@pytest.fixture(params=[{"name": "周潤(rùn)發(fā)"}, {"age": 61}, {"height": 183}])
def fix_func(request):  # request為內(nèi)建fixture
    # 使用request.param作為返回值供測(cè)試函數(shù)調(diào)用，params的參數(shù)列表中包含了做少元素，該fixture就會(huì)被調(diào)用幾次，分別作用在每個(gè)測(cè)試函數(shù)上
    return request.param  # request.param為固定寫(xiě)法

def test_fix_func(fix_func):
    print(f"fixture函數(shù)fix_func的返回值為：{fix_func}")
    
    """打印結(jié)果如下：
    fixture函數(shù)fix_func的返回值為：{'name': '周潤(rùn)發(fā)'}
    fixture函數(shù)fix_func的返回值為：{'age': 61}
    fixture函數(shù)fix_func的返回值為：{'height': 183}
    """

params = [
    {"case_id": 1, "case_title": "驗(yàn)證正常添加車(chē)輛", "car_name": "蘇C99688", "car_type": 1, "origin": 1, "expected": "200"},
    {"case_id": 2, "case_title": "驗(yàn)證添加重復(fù)車(chē)輛", "car_name": "蘇C99688", "car_type": 1, "origin": 1, "expected": "500"},
    {"case_id": 3, "case_title": "驗(yàn)證車(chē)牌號(hào)為空", "car_name": "", "car_type": 2, "origin": 1, "expected": "500"}]

@pytest.fixture(params=params)
def add_car_params(request):
    return request.param

def test_add_car(add_car_params):
    print(f"{add_car_params['case_id']}-{add_car_params['case_title']}-{add_car_params['car_name']}")
    
    """
    運(yùn)行結(jié)果如下：
    1-驗(yàn)證正常添加車(chē)輛-蘇C99688
    2-驗(yàn)證添加重復(fù)車(chē)輛-蘇C99688
    3-驗(yàn)證車(chē)牌號(hào)為空-
    """

def test_tmpdir(tmpdir):
    # tmpdir already has a path name associated with it
    # join() extends the path to include a filename
    # the file is created when it's written to
    a_file = tmpdir.join('something.txt')

    # you can create directories
    a_sub_dir = tmpdir.mkdir('anything')

    # you can create files in directories (created when written)
    another_file = a_sub_dir.join('something_else.txt')

    # this write creates 'something.txt'
    a_file.write('contents may settle during shipping')

    # this write creates 'anything/something_else.txt'
    another_file.write('something different')

    # you can read the files as well
    assert a_file.read() == 'contents may settle during shipping'
    assert another_file.read() == 'something different'


def test_tmpdir_factory(tmpdir_factory):
    # you should start with making a directory
    # a_dir acts like the object returned from the tmpdir fixture
    a_dir = tmpdir_factory.mktemp('mydir')

    # base_temp will be the parent dir of 'mydir'
    # you don't have to use getbasetemp()
    # using it here just to show that it's available
    base_temp = tmpdir_factory.getbasetemp()
    print('base:', base_temp)

    # the rest of this test looks the same as the 'test_tmpdir()'
    # example except I'm using a_dir instead of tmpdir

    a_file = a_dir.join('something.txt')
    a_sub_dir = a_dir.mkdir('anything')
    another_file = a_sub_dir.join('something_else.txt')

    a_file.write('contents may settle during shipping')
    another_file.write('something different')

    assert a_file.read() == 'contents may settle during shipping'
    assert another_file.read() == 'something different'

def pytest_addoption(parser):
    """"添加一個(gè)命令行選項(xiàng)"""
    parser.addoption(
        "--env", default="test", choices=["dev", "test", "pre"], help="enviroment parameter")

def test_option(pytestconfig):
    print('the current environment is:', pytestconfig.getoption('env'))
# 運(yùn)行測(cè)試
pytest -s -q test_config.py::test_option