diff --git a/tests/conftest.py b/tests/conftest.py index 364638db1..47dd4d58b 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,3 +1,5 @@ +import os + import pytest from pytest_cases import fixture @@ -13,18 +15,22 @@ @fixture(scope="session") def api_client(pytestconfig): if pytestconfig.getoption("--drone"): - return OpenCTIApiClient( - "http://opencti:4000", - "bfa014e0-e02e-4aa6-a42b-603b19dcf159", - ssl_verify=False, + api_url = os.getenv("OPENCTI_API_URL", "http://opencti:4000") + api_token = os.getenv( + "OPENCTI_API_TOKEN", "bfa014e0-e02e-4aa6-a42b-603b19dcf159" ) else: - return OpenCTIApiClient( - "http://localhost:4000", - "d434ce02-e58e-4cac-8b4c-42bf16748e84", - ssl_verify=False, + api_url = os.getenv("OPENCTI_API_URL", "http://localhost:4000") + api_token = os.getenv( + "OPENCTI_API_TOKEN", "d434ce02-e58e-4cac-8b4c-42bf16748e84" ) + return OpenCTIApiClient( + api_url, + api_token, + ssl_verify=False, + ) + @fixture(scope="session") def api_connector(api_client):