From 42119475488653d8665b8a8241e4ac0cd87ddeb5 Mon Sep 17 00:00:00 2001
From: Nikolay Proskurin <n1ckpr0s@yandex.ru>
Date: Tue, 10 Oct 2023 11:59:15 +0200
Subject: [PATCH] BI-4981: use a connector whitelist in RQE apps in tests
---
lib/dl_api_lib_testing/dl_api_lib_testing/app.py | 13 +++++++++----
1 file changed, 9 insertions(+), 4 deletions(-)
diff --git a/lib/dl_api_lib_testing/dl_api_lib_testing/app.py b/lib/dl_api_lib_testing/dl_api_lib_testing/app.py
index 11ace308f..d611e2476 100644
--- a/lib/dl_api_lib_testing/dl_api_lib_testing/app.py
+++ b/lib/dl_api_lib_testing/dl_api_lib_testing/app.py
@@ -57,14 +57,19 @@ class RQEConfigurationMaker:
@contextlib.contextmanager
def sync_rqe_netloc_subprocess_cm(self) -> Generator[RQEBaseURL, None, None]:
+ whitelist = self.bi_test_config.core_test_config.get_core_library_config().core_connector_ep_names
+ env = dict(
+ EXT_QUERY_EXECUTER_SECRET_KEY=self.bi_test_config.ext_query_executer_secret_key,
+ DEV_LOGGING="1",
+ )
+ if whitelist:
+ env["CORE_CONNECTOR_WHITELIST"] = ",".join(whitelist)
+
with WSGIRunner(
module="dl_core.bin.query_executor_sync",
callable="app",
ping_path="/ping",
- env=dict(
- EXT_QUERY_EXECUTER_SECRET_KEY=self.bi_test_config.ext_query_executer_secret_key,
- DEV_LOGGING="1",
- ),
+ env=env,
) as runner:
yield RQEBaseURL( # type: ignore # TODO: fix compatibility of models using `s_attrib` with mypy
host=runner.bind_addr,