@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為True時(shí)，會(huì)作用于每一條用例"""
print("\n---用例前置操作---")
yield
print("---用例后置操作---")

# func_auto函數(shù)的autouse=True時(shí)，無論是否使用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")

fixture(scope='function'，params=None，autouse=False，ids=None，name=None)

# @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沒有用修飾器修飾，故不會(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():
"""類級(jí)別fixture，作用域整個(gè)類"""
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")

def test_class_auto_fixture(class_auto):
print("class 1 print")

@pytest.fixture(scope="module")
def module_auto():
"""作用于整個(gè)py文件"""
print("\n---module級(jí)別的用例前置操作---")
yield
print("---module級(jí)別的用例后置操作---")

# 測(cè)試類外和測(cè)試類內(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-z2cbyvy" 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 = "周潤發(fā)"
print(user_name)

# 此處需傳入重命名后的fixture函數(shù)名
@pytest.mark.usefixtures("rename_get_user_info")
def test_parametrize_by_use_fixtures():
"""通過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": "周潤發(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為固定寫法

def test_fix_func(fix_func):
print(f"fixture函數(shù)fix_func的返回值為：{fix_func}")

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

params = [
{"case_id": 1, "case_title": "驗(yàn)證正常添加車輛", "car_name": "蘇C99688", "car_type": 1, "origin": 1, "expected": "200"},
{"case_id": 2, "case_title": "驗(yàn)證添加重復(fù)車輛", "car_name": "蘇C99688", "car_type": 1, "origin": 1, "expected": "500"},
{"case_id": 3, "case_title": "驗(yàn)證車牌號(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)證正常添加車輛-蘇C99688
2-驗(yàn)證添加重復(fù)車輛-蘇C99688
3-驗(yàn)證車牌號(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