mirror of
https://github.com/PaiGramTeam/SIMNet.git
synced 2024-11-23 14:57:45 +00:00
126 lines
3.2 KiB
Python
126 lines
3.2 KiB
Python
import asyncio
|
|
import os
|
|
import warnings
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
import pytest
|
|
from dotenv import load_dotenv
|
|
|
|
from simnet.client.cookies import Cookies
|
|
from simnet.utils.cookies import parse_cookie
|
|
from simnet.utils.enums import Region
|
|
|
|
env_path = Path(".env")
|
|
if env_path.exists():
|
|
load_dotenv()
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def event_loop(): # skipcq: PY-D0003
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore")
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
yield loop
|
|
loop.close()
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def cookies() -> "Cookies": # skipcq: PY-D0003
|
|
cookies_str = os.environ.get("COOKIES")
|
|
if not cookies_str:
|
|
pytest.exit("No cookies set", 1)
|
|
|
|
_cookies = Cookies(parse_cookie(cookies_str))
|
|
if _cookies.account_id is None:
|
|
warnings.warn("can not found account id in cookies")
|
|
|
|
return _cookies
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def starrail_cookies() -> Optional["Cookies"]: # skipcq: PY-D0003
|
|
cookies_str = os.environ.get("STARRAIL_COOKIES")
|
|
if not cookies_str:
|
|
return None
|
|
_cookies = Cookies(parse_cookie(cookies_str))
|
|
if _cookies.account_id is None:
|
|
warnings.warn("can not found account id in starrail cookies")
|
|
|
|
return _cookies
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def zzz_cookies() -> Optional["Cookies"]:
|
|
cookies_str = os.environ.get("ZZZ_COOKIES")
|
|
if not cookies_str:
|
|
return None
|
|
_cookies = Cookies(parse_cookie(cookies_str))
|
|
if _cookies.account_id is None:
|
|
warnings.warn("can not found account id in zzz cookies")
|
|
|
|
return _cookies
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def genshin_player_id() -> Optional[int]: # skipcq: PY-D0003
|
|
_player_id = os.environ.get("GENSHIN_PLAYER_ID")
|
|
if not _player_id:
|
|
warnings.warn("No genshin player id set")
|
|
return None
|
|
return int(_player_id)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def starrail_player_id() -> Optional[int]: # skipcq: PY-D0003
|
|
_player_id = os.environ.get("STARRAIL_PLAYER_ID")
|
|
if not _player_id:
|
|
warnings.warn("No starrail player id set")
|
|
return None
|
|
return int(_player_id)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def zzz_player_id() -> Optional[int]:
|
|
_player_id = os.environ.get("ZZZ_PLAYER_ID")
|
|
if not _player_id:
|
|
warnings.warn("No zzz player id set")
|
|
return None
|
|
return int(_player_id)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def account_id() -> Optional[int]: # skipcq: PY-D0003
|
|
_account_id = os.environ.get("ACCOUNT_ID")
|
|
if not _account_id:
|
|
return None
|
|
return int(_account_id)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def region() -> Region: # skipcq: PY-D0003
|
|
_region = os.environ.get("REGION")
|
|
if not _region:
|
|
return Region.CHINESE
|
|
return Region(_region)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def stoken() -> Optional[str]: # skipcq: PY-D0003
|
|
_stoken = os.environ.get("STOKEN")
|
|
return _stoken
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def login_ticket() -> Optional[str]: # skipcq: PY-D0003
|
|
_login_ticket = os.environ.get("LOGIN_TICKET")
|
|
return _login_ticket
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def if_test_build() -> bool: # skipcq: PY-D0003
|
|
_test_build = bool(os.environ.get("TEST_BUILD", False))
|
|
return _test_build
|