test: BI-6496 Add auth_data fixture to testing control_api (#1205)

fake_auth_data

Author: khamitovdr

lib/dl_api_commons/dl_api_commons/flask/middlewares/trust_auth.py

@@ -4,6 +4,7 @@
 import flask
 
 from dl_api_commons.base_models import (
+    AuthData,
     NoAuthData,
     TenantCommon,
     TenantDef,
@@ -21,14 +22,15 @@ class TrustAuthService:
     fake_user_id: Optional[str] = None
     fake_user_name: Optional[str] = None
     fake_tenant: Optional[TenantDef] = None
+    fake_auth_data: Optional[AuthData] = None
 
     def _before_request(self) -> None:
         fake_user_id = self.fake_user_id
         fake_user_name = self.fake_user_name
         fake_tenant = self.fake_tenant
 
         temp_rci = ReqCtxInfoMiddleware.get_temp_rci().clone(
-            auth_data=NoAuthData(),
+            auth_data=NoAuthData() if self.fake_auth_data is None else self.fake_auth_data,
             tenant=TenantCommon() if fake_tenant is None else fake_tenant,
         )
         if fake_user_id is not None:

lib/dl_api_lib/dl_api_lib/app_settings.py

@@ -10,7 +10,10 @@
 import attr
 import pydantic
 
-from dl_api_commons.base_models import TenantDef
+from dl_api_commons.base_models import (
+    AuthData,
+    TenantDef,
+)
 from dl_api_lib.connector_availability.base import ConnectorAvailabilityConfig
 from dl_configs.crypto_keys import CryptoKeysConfig
 from dl_configs.enums import RedisMode
@@ -306,6 +309,7 @@ def jaeger_service_name(self) -> str:
 class ControlApiAppTestingsSettings:
     us_auth_mode_override: Optional[USAuthMode] = attr.ib(default=None)
     fake_tenant: Optional[TenantDef] = attr.ib(default=None)
+    fake_auth_data: Optional[AuthData] = attr.ib(default=None)
 
 
 @attr.s(frozen=True)

lib/dl_api_lib_testing/dl_api_lib_testing/app.py

@@ -140,6 +140,7 @@ def set_up_environment(
             fake_user_id=TEST_USER_ID,
             fake_user_name=TEST_USER_NAME,
             fake_tenant=None if testing_app_settings is None else testing_app_settings.fake_tenant,
+            fake_auth_data=None if testing_app_settings is None else testing_app_settings.fake_auth_data,
         ).set_up(app)
 
         us_auth_mode_override = None if testing_app_settings is None else testing_app_settings.us_auth_mode_override

lib/dl_api_lib_testing/dl_api_lib_testing/base.py

@@ -14,6 +14,7 @@
 from dl_api_client.dsmaker.api.dataset_api import SyncHttpDatasetApiV1
 from dl_api_client.dsmaker.api.http_sync_base import SyncHttpClientBase
 from dl_api_commons.base_models import (
+    AuthData,
     RequestContextInfo,
     TenantCommon,
     TenantDef,
@@ -166,19 +167,27 @@ def control_api_app_factory(self, control_api_app_settings: ControlApiAppSetting
     def fake_tenant(self) -> TenantDef:
         return TenantCommon()
 
+    @pytest.fixture(scope="function")
+    def fake_auth_data(self) -> Optional[AuthData]:
+        return None
+
     @pytest.fixture(scope="function")
     def control_api_app(
         self,
         environment_readiness: None,
         control_api_app_factory: ControlApiAppFactory,
         connectors_settings: dict[ConnectionType, ConnectorSettingsBase],
         fake_tenant: TenantDef,
+        fake_auth_data: Optional[AuthData],
     ) -> Generator[Flask, None, None]:
         """Session-wide test `Flask` application."""
 
         app = control_api_app_factory.create_app(
             connectors_settings=connectors_settings,
-            testing_app_settings=ControlApiAppTestingsSettings(fake_tenant=fake_tenant),
+            testing_app_settings=ControlApiAppTestingsSettings(
+                fake_tenant=fake_tenant,
+                fake_auth_data=fake_auth_data,
+            ),
             close_loop_after_request=False,
         )