chore: refactor and add components integration tests (#3607)

* improve inegration tests

* add fixes

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Nicolò Boschi 2024-09-02 15:21:47 +02:00 • committed by GitHub
commit 96872f3aa5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
32 changed files with 523 additions and 131 deletions

View file

@ -0,0 +1,34 @@
import os.path
# we need to import tmpdir
def get_required_env_var(var: str) -> str:
"""
Get the value of the specified environment variable.
Args:
var (str): The environment variable to get.
Returns:
str: The value of the environment variable.
Raises:
ValueError: If the environment variable is not set.
"""
value = os.getenv(var)
if not value:
raise ValueError(f"Environment variable {var} is not set")
return value
def get_openai_api_key() -> str:
return get_required_env_var("OPENAI_API_KEY")
def get_astradb_application_token() -> str:
return get_required_env_var("ASTRA_DB_APPLICATION_TOKEN")
def get_astradb_api_endpoint() -> str:
return get_required_env_var("ASTRA_DB_API_ENDPOINT")