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()