60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
# @Time : 2021/8/14 12:33
|
|||
|
|
# @Author : Flora.Chen
|
|||
|
|
# @File : draft_bk.py.py
|
|||
|
|
# @Software: PyCharm
|
|||
|
|
# @Desc:
|
|||
|
|
|
|||
|
|
import pytest
|
|||
|
|
from loguru import logger
|
|||
|
|
from config.settings import EnvConfig
|
|||
|
|
from api.login_api import TrustieLogin
|
|||
|
|
from api.project_api import ProjectApi
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ------------------------------------- START: 配置运行环境 ---------------------------------------#
|
|||
|
|
|
|||
|
|
def pytest_addoption(parser):
|
|||
|
|
"""
|
|||
|
|
pytest_addoption 可以让用户注册一个自定义的命令行参数，方便用户将数据传递给 pytest；
|
|||
|
|
这个 Hook 方法一般和 内置 fixture pytestconfig 配合使用，pytest_addoption 注册命令行参数，pytestconfig 通过配置对象读取参数的值；
|
|||
|
|
:param parser:
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
parser.addoption(
|
|||
|
|
# action="store" 默认，只存储参数的值，可以存储任何类型的值，此时 default 也可以是任何类型的值，而且命令行参数多次使用也只能生效一个，最后一个值覆盖之前的值；
|
|||
|
|
"--env", action="store",
|
|||
|
|
default="test",
|
|||
|
|
choices=["test", "live"], # choices 只允许输入的值的范围
|
|||
|
|
help="通过自定义命令行参数-env设置当前运行的环境"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 从配置对象中读取自定义参数的值
|
|||
|
|
@pytest.fixture(scope="session", autouse=True)
|
|||
|
|
def env(request):
|
|||
|
|
env = request.config.getoption("--env")
|
|||
|
|
env_mapping = EnvConfig.env_mapping.value
|
|||
|
|
env = env_mapping.get(f"{env}")
|
|||
|
|
forge_env = env.get("forge")
|
|||
|
|
logger.debug(f"当前的执行环境：{forge_env}")
|
|||
|
|
return {"forge_env": forge_env}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@pytest.fixture(scope="session")
|
|||
|
|
def get_user(request):
|
|||
|
|
env = request.config.getoption("--env")
|
|||
|
|
users_mapping = EnvConfig.users_mapping.value
|
|||
|
|
users = users_mapping.get(f"{env}")
|
|||
|
|
return users
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ------------------------------------- END: 配置运行环境 ---------------------------------------#
|
|||
|
|
|
|||
|
|
@pytest.fixture(scope="session")
|
|||
|
|
def get_cookies(env, get_user):
|
|||
|
|
login = TrustieLogin(host=env)
|
|||
|
|
return login.login_api(user=get_user.get("user"), pwd=get_user.get("pwd")).cookies
|
|||
|
|
|