enshell security token

src/ansys/pyensight/core/common.py

@@ -166,7 +166,7 @@ def populate_service_host_port(  # pragma: no cover
 
 
 def launch_enshell_interface(
-    enshell_grpc_channel: Any, grpc_port: int, timeout: float
+    enshell_grpc_channel: Any, grpc_port: int, timeout: float, secret_key: str
 ) -> enshell_grpc.EnShellGRPC:
     """Launch the EnShell gRPC Interface.
 
@@ -186,9 +186,11 @@ def launch_enshell_interface(
     """
     if enshell_grpc_channel:  # pragma: no cover
         enshell = enshell_grpc.EnShellGRPC()  # pragma: no cover
+        enshell.set_security_token(secret_key)
         enshell.connect_existing_channel(enshell_grpc_channel)  # pragma: no cover
     else:
         enshell = enshell_grpc.EnShellGRPC(port=grpc_port)
+        enshell.set_security_token(secret_key)
         time_start = time.time()
         while time.time() - time_start < timeout:  # pragma: no cover
             if enshell.is_connected():

src/ansys/pyensight/core/dockerlauncher.py

@@ -427,7 +427,10 @@ def connect(self):
         #
         # Start up the EnShell gRPC interface
         self._enshell = launch_enshell_interface(
-            self._enshell_grpc_channel, self._service_host_port["grpc"][1], self._timeout
+            self._enshell_grpc_channel,
+            self._service_host_port["grpc"][1],
+            self._timeout,
+            self._secret_key,
         )
         log_dir = "/data"
         if self._enshell_grpc_channel:  # pragma: no cover

src/ansys/pyensight/core/enshell_grpc.py

@@ -51,7 +51,7 @@ def __init__(
         self._stub = None
         #
         # self._security_token = str(random.randint(0, 1000000))
-        self._security_token: Optional[int] = None
+        self._security_token: Optional[str] = None
         #
         # values found from EnShell in the Container
         self._cei_home = None
@@ -80,7 +80,7 @@ def port(self):
         """
         return self._port
 
-    def set_security_token(self, n: Optional[int] = None):
+    def set_security_token(self, n: Optional[str] = None):
         """set the security token for the gRPC connection.
 
         EnShell supports a security token in either numeric (-security {int}) or

tests/conftest.py

@@ -64,6 +64,7 @@ def docker_launcher_session() -> "Session":
 @pytest.fixture
 def enshell_mock():
     mocked_grpc = mock.MagicMock("GRPC")
+    mocked_grpc.set_security_token = lambda secret: True
     mocked_grpc.command = mock.MagicMock("command")
     mocked_grpc.is_connected = lambda: True
     mocked_grpc.connect = mock.MagicMock("execute_connection")