fix: Fixed test and client

Author: Robin VAN DE MERGHEL

diracx-client/src/diracx/client/_generated/_client.py

@@ -15,7 +15,14 @@
 from . import models as _models
 from ._configuration import DiracConfiguration
 from ._utils.serialization import Deserializer, Serializer
-from .operations import AuthOperations, ConfigOperations, JobsOperations, PilotsOperations, WellKnownOperations
+from .operations import (
+    AuthOperations,
+    ConfigOperations,
+    JobsOperations,
+    PilotsInternalOperations,
+    PilotsOperations,
+    WellKnownOperations,
+)
 
 
 class Dirac:  # pylint: disable=client-accepts-api-version-keyword
@@ -31,6 +38,8 @@ class Dirac:  # pylint: disable=client-accepts-api-version-keyword
     :vartype jobs: _generated.operations.JobsOperations
     :ivar pilots: PilotsOperations operations
     :vartype pilots: _generated.operations.PilotsOperations
+    :ivar pilots_internal: PilotsInternalOperations operations
+    :vartype pilots_internal: _generated.operations.PilotsInternalOperations
     :keyword endpoint: Service URL. Required. Default value is "".
     :paramtype endpoint: str
     """
@@ -68,6 +77,7 @@ def __init__(  # pylint: disable=missing-client-constructor-parameter-credential
         self.config = ConfigOperations(self._client, self._config, self._serialize, self._deserialize)
         self.jobs = JobsOperations(self._client, self._config, self._serialize, self._deserialize)
         self.pilots = PilotsOperations(self._client, self._config, self._serialize, self._deserialize)
+        self.pilots_internal = PilotsInternalOperations(self._client, self._config, self._serialize, self._deserialize)
 
     def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse:
         """Runs the network request through the client's chained policies.

diracx-client/src/diracx/client/_generated/aio/_client.py

@@ -15,7 +15,14 @@
 from .. import models as _models
 from .._utils.serialization import Deserializer, Serializer
 from ._configuration import DiracConfiguration
-from .operations import AuthOperations, ConfigOperations, JobsOperations, PilotsOperations, WellKnownOperations
+from .operations import (
+    AuthOperations,
+    ConfigOperations,
+    JobsOperations,
+    PilotsInternalOperations,
+    PilotsOperations,
+    WellKnownOperations,
+)
 
 
 class Dirac:  # pylint: disable=client-accepts-api-version-keyword
@@ -31,6 +38,8 @@ class Dirac:  # pylint: disable=client-accepts-api-version-keyword
     :vartype jobs: _generated.aio.operations.JobsOperations
     :ivar pilots: PilotsOperations operations
     :vartype pilots: _generated.aio.operations.PilotsOperations
+    :ivar pilots_internal: PilotsInternalOperations operations
+    :vartype pilots_internal: _generated.aio.operations.PilotsInternalOperations
     :keyword endpoint: Service URL. Required. Default value is "".
     :paramtype endpoint: str
     """
@@ -68,6 +77,7 @@ def __init__(  # pylint: disable=missing-client-constructor-parameter-credential
         self.config = ConfigOperations(self._client, self._config, self._serialize, self._deserialize)
         self.jobs = JobsOperations(self._client, self._config, self._serialize, self._deserialize)
         self.pilots = PilotsOperations(self._client, self._config, self._serialize, self._deserialize)
+        self.pilots_internal = PilotsInternalOperations(self._client, self._config, self._serialize, self._deserialize)
 
     def send_request(
         self, request: HttpRequest, *, stream: bool = False, **kwargs: Any

diracx-client/src/diracx/client/_generated/aio/operations/__init__.py

@@ -15,6 +15,7 @@
 from ._operations import ConfigOperations  # type: ignore
 from ._operations import JobsOperations  # type: ignore
 from ._operations import PilotsOperations  # type: ignore
+from ._operations import PilotsInternalOperations  # type: ignore
 
 from ._patch import __all__ as _patch_all
 from ._patch import *
@@ -26,6 +27,7 @@
     "ConfigOperations",
     "JobsOperations",
     "PilotsOperations",
+    "PilotsInternalOperations",
 ]
 __all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore
 _patch_sdk()