diff --git a/doccano_client/beta/tests/test_full_integration_tests.py b/doccano_client/beta/tests/test_full_integration_tests.py index b4dc878..82ac7a5 100644 --- a/doccano_client/beta/tests/test_full_integration_tests.py +++ b/doccano_client/beta/tests/test_full_integration_tests.py @@ -1,13 +1,17 @@ from unittest import TestCase import pytest +from dotenv import load_dotenv from ..client import DoccanoClient from ..models import Example, Project, SpanType +import os -DOCCANO_ENDPOINT = "http://localhost" # TODO: Make this a pytest parameter or ENV variable -DOCCANO_USER = "admin" # TODO: Make this a pytest parameter or ENV variable -DOCCANO_PASS = "password" # TODO: Make this a pytest parameter or ENV variable +load_dotenv() + +DOCCANO_ENDPOINT = os.getenv("DOCCANO_ENDPOINT", "http://localhost") +DOCCANO_USER = os.getenv("DOCCANO_USER", "admin") +DOCCANO_PASS = os.getenv("DOCCANO_PASS", "password") @pytest.mark.localintegrationtest diff --git a/pyproject.toml b/pyproject.toml index 0b4c7ba..79ca09c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -46,6 +46,7 @@ mkdocs-material = "^8.5.3" mkdocs-same-dir = "^0.1.1" vcrpy = "^4.2.1" mkdocstrings = {extras = ["python", "crystal"], version = "^0.19.0"} +python-dotenv = "^1.0.0" [tool.poetry.extras] spacy = ["spacy", "spacy-partial-tagger", "tqdm"]