Skip to content

Commit 46c9892

Browse files
authored
Check connection GrpcServer before returning (#558)
* check connection GrpcServer * remove dummy test
1 parent 2fe32c0 commit 46c9892

File tree

2 files changed

+12
-13
lines changed

2 files changed

+12
-13
lines changed

ansys/dpf/core/server_types.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -538,6 +538,7 @@ def __init__(self,
538538
launch_server=True,
539539
docker_name=None,
540540
use_pypim=True,
541+
num_connection_tryouts=3,
541542
):
542543
# Load DPFClientAPI
543544
from ansys.dpf.core.misc import is_pypim_configured
@@ -569,8 +570,19 @@ def __init__(self,
569570
self._input_port = port
570571
self.live = True
571572
self._create_shutdown_funcs()
573+
self._check_first_call(num_connection_tryouts)
572574
self.set_as_global(as_global=as_global)
573575

576+
def _check_first_call(self, num_connection_tryouts):
577+
for i in range(num_connection_tryouts):
578+
try:
579+
self.version
580+
break
581+
except errors.DPFServerException as e:
582+
if ("GOAWAY" not in str(e.args) and "unavailable" not in str(e.args)) \
583+
or i == (num_connection_tryouts - 1):
584+
raise e
585+
574586
@property
575587
def version(self):
576588
from ansys.dpf.gate import data_processing_capi, integral_types

tests/test_server.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -75,19 +75,6 @@ def test_start_local_server_with_config(self, server_config):
7575
assert has_local_server()
7676
shutdown_all_session_servers()
7777

78-
def test_connect_to_server(self, server_config):
79-
set_server_configuration(server_config)
80-
print(dpf.core.SERVER_CONFIGURATION)
81-
shutdown_all_session_servers()
82-
start_local_server(timeout=10.)
83-
print("has_local_server", has_local_server())
84-
if hasattr(dpf.core.SERVER, "ip"):
85-
connect_to_server(ip=dpf.core.SERVER.ip, port=dpf.core.SERVER.port, timeout=10.,
86-
as_global=False)
87-
else:
88-
connect_to_server(timeout=10., as_global=False)
89-
assert has_local_server()
90-
9178
def test_shutdown_all_session_servers(self, server_config):
9279
set_server_configuration(server_config)
9380
print(dpf.core.SERVER_CONFIGURATION)

0 commit comments

Comments
 (0)