def configs(config_name, default=''):
# config from settings
settings_obj = Settings()
settings_module_path = os.environ.get(ENV_VAR, 'app.settings')
settings_obj.setmodule(settings_module_path, priority='project')
return configs_from_os(config_name, settings_obj, default)
pass
def get_settings_override():
return Settings(
testing=1,
database_url=os.environ.get("DATABASE_TEST_URL"),
auth_secret_key=AUTH_SECRET_KEY,
token_algorithm="HS256",
)
def get_settings_override_tacacs():
return Settings(
AUTH_KEY=AUTH_KEY,
TOKEN_ALGORITHM="HS256",
AUTH_MODE="tacacs",
TACACS_SVR="tacacs",
TACACS_KEY="tac_plus_key",
)
def test_ping(test_app_tacacs):
response = test_app_tacacs.get("/ping")
assert response.status_code == 200
assert response.json() == {
"ping": "pong",
"project": Settings().PROJECT,
"environment": "dev",
}
def main() -> None:
settings = Settings()
region_path = [(Region.NA, settings.na_gamedata),
(Region.JP, settings.jp_gamedata)]
export_path = Path(__file__).resolve().parents[1] / "export"
for region, gamedata in region_path:
print(region)
export_constant(region, gamedata, export_path)
export_nice_master_lvl(region, gamedata, export_path)
def test_airtable_repository_impl():
settings = Settings()
repository = ServiceListRepositoryAirtable(
settings.airtable_api_key, settings.airtable_base,
"servicios_pagos_servientrega_test")
services = repository.get_all()
assert len(services) == 3
def test_protected_ping_success(test_app_tacacs, get_access_token):
access_token = get_access_token(username="******")
print(access_token)
response = test_app_tacacs.get(
"/protected_ping", headers={"Authorization": f"Bearer {access_token}"})
# assert response.status_code == 200
assert response.json() == {
"ping": "pong",
"project": Settings().PROJECT,
"environment": "dev",
"username": "******",
}
async def database():
settings = Settings()
DATABASE_URL = settings.database_url #"mysql://*****:*****@dbmysql/automation"
database = Database(DATABASE_URL)
await database.connect()
query = """CREATE TABLE IF NOT EXISTS itinerario (
id INT NOT NULL AUTO_INCREMENT,
pagado_por_servientrega INT NULL,
referencia VARCHAR(45) NULL,
PRIMARY KEY (id));
"""
await database.execute(query=query)
yield database
await database.execute(query="DROP TABLE `automation`.`itinerario`")
await database.disconnect()
def override_test_settings():
return Settings(db_url=SQLALCHEMY_TEST_DATABASE_URL)
import logging
from sqlalchemy import create_engine
from tenacity import after_log, before_log, retry, stop_after_attempt, wait_fixed
from app.config import Settings
logger = logging.getLogger(__name__)
max_tries = 60 * 5 # 5 minutes
wait_seconds = 1
settings = Settings()
@retry(
stop=stop_after_attempt(max_tries),
wait=wait_fixed(wait_seconds),
before=before_log(logger, logging.INFO),
after=after_log(logger, logging.WARN),
)
def check_db_awake() -> None:
engine = create_engine(settings.DATABASE_URI, pool_pre_ping=True)
try:
with engine.connect() as conn:
conn.execute("SELECT 1")
except Exception as e:
logger.error(e)
raise (e)
finally:
engine.dispose()
def get_settings_override():
return Settings(
testing=True,
database_url=get_settings().database_url_test,
)
def get_settings_override():
return Settings(testing=1)
def get_settings_override():
settings = Settings(testing=1, database_url=config("DATABASE_TEST_URL"))
return settings
def get_settings_override():
"""
Overrides the settings
"""
return Settings(testing=1)
def test_url_trim_slash(self, monkeypatch: MonkeyPatch) -> None:
monkeypatch.setenv("ASSET_URL", "https://example.com/assets/")
settings = Settings()
assert settings.asset_url == "https://example.com/assets"
def get_settings_override():
return Settings(
TESTING=True,
DATABASE_URL=os.environ.get("DATABASE_TEST_URL")
)
def get_settings_override():
"""Override app settings for testing."""
return Settings(testing=1, database_url=os.getenv('DATABASE_TEST_URL'))
def get_settings_override():
return Settings(testing=1,
database_url=os.environ.get("DATABASE_TEST_URL"))
def get_settings() -> Settings:
return Settings()