def test_invalid_server_creation(self):
with ErtTestContext("ert/server/rpc/server", self.config) as test_context:
# No such file
with self.assertRaises(IOError):
server = ErtRPCServer("this/is/not/a/file")
# No configuration
with self.assertRaises(ValueError):
server = ErtRPCServer(None)
# Wrong enkf_main type
with self.assertRaises(TypeError):
enkf_main = test_context.getErt()
server = ErtRPCServer(enkf_main=enkf_main.resConfig())
def test_server_creation(self):
with ErtTestContext("ert/server/rpc/server",
self.config) as test_context:
ert = test_context.getErt()
server = ErtRPCServer(ert)
self.assertIsNotNone(server.port)
self.assertEqual(ert, server._config)
self.assertEqual(os.path.basename(self.config),
server._config_file)
thread = Thread(name="ErtRPCServerTest")
thread.run = server.start
thread.start()
server.stop()
def startServer(self, line):
port = self._port
host = self._hostname
if self._server is None:
try:
self._server = ErtRPCServer(self.ert(), host=host, port=port)
except socket.error as e:
print("Unable to start the server on port: %d" % port)
else:
thread = Thread(name="Shell Server Thread")
thread.daemon = True
thread.run = self._server.start
thread.start()
print("Server running on host: '%s' and port: %d" % (self._server.host, self._server.port))
else:
print("A server is already running at host: '%s' and port: %d" % (self._server.host, self._server.port))
class RPCServiceContext(object):
def __init__(self, test_name, model_config, store_area=False):
self._test_context = ErtTest(test_name,
model_config,
store_area=store_area)
self._server = ErtRPCServer(self._test_context.getErt())
def __enter__(self):
""" @rtype: ErtRPCServer"""
thread = Thread(name="ErtRPCServerTest")
thread.run = self._server.start
thread.start()
return self._server
def __exit__(self, exc_type, exc_val, exc_tb):
self._server.stop()
del self._server
del self._test_context
return False
def test_deprecated_server_creation(self):
with ErtTestContext("ert/server/rpc/server",
self.config) as test_context:
ert = test_context.getErt()
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")
server = ErtRPCServer(ert.getUserConfigFile())
self.assertTrue(len(w) > 0)
self.assertTrue(issubclass(w[-1].category, DeprecationWarning))
self.assertIsNotNone(server.port)
expected_config_file = os.path.join(test_context.getCwd(),
os.path.basename(self.config))
self.assertEqual(expected_config_file, server._config_file)
thread = Thread(name="ErtRPCServerTest")
thread.run = server.start
thread.start()
server.stop()
class Server(ErtShellCollection):
def __init__(self, parent):
super(Server, self).__init__("server", parent)
self.shellContext()["server_settings"] = self
self.addShellFunction(name="start",
function=Server.startServer,
help_message="Start the ERT RPC Server")
self.addShellFunction(name="stop",
function=Server.stopServer,
help_message="Stop the ERT RPC Server")
self.addShellFunction(name="inspect",
function=Server.inspect,
help_message="Shows information about the current job queue")
self.addShellProperty(name="hostname",
getter=Server.getHost,
setter=Server.setHost,
help_arguments="[hostname]",
help_message="Show or set the server hostname",
pretty_attribute="Hostname")
self.addShellProperty(name="port",
getter=Server.getPort,
setter=Server.setPort,
help_arguments="[port]",
help_message="Show or set the server port number (0 = automatic)",
pretty_attribute="Port")
self._server = None
""" :type: ErtRPCServer """
self._hostname = "localhost"
self._port = 0
def getHost(self):
return self._hostname
def setHost(self, hostname):
self._hostname = hostname
def getPort(self):
return self._port
def setPort(self, port):
self._port = int(port)
@assertConfigLoaded
def startServer(self, line):
port = self._port
host = self._hostname
if self._server is None:
try:
self._server = ErtRPCServer(self.ert(), host=host, port=port)
except socket.error as e:
print("Unable to start the server on port: %d" % port)
else:
thread = Thread(name="Shell Server Thread")
thread.daemon = True
thread.run = self._server.start
thread.start()
print("Server running on host: '%s' and port: %d" % (self._server.host, self._server.port))
else:
print("A server is already running at host: '%s' and port: %d" % (self._server.host, self._server.port))
def _stopServer(self):
if self._server is not None:
self._server.stop()
self._server = None
print("Server stopped")
def stopServer(self, line):
if self._server is not None:
self._stopServer()
else:
print("No server to stop")
def cleanup(self):
self._stopServer()
ErtShellCollection.cleanup(self)
def inspect(self, line):
if self._server is not None:
if self._server.isRunning():
print("Waiting..: %d" % self._server.getWaitingCount())
print("Running..: %d" % self._server.getRunningCount())
print("Failed...: %d" % self._server.getFailedCount())
print("Succeeded: %d" % self._server.getSuccessCount())
print("Batch#...: %d" % self._server.getBatchNumber())
else:
print("Server is not running any simulations")
else:
print("No server is not available")
def __init__(self, test_name, model_config, store_area=False):
self._test_context = ErtTest(test_name,
model_config,
store_area=store_area)
self._server = ErtRPCServer(self._test_context.getErt())
