async def test_shutdown_service_pass(self, test_server, loop):
# GIVEN a service is running at a given URL
app = web.Application()
with patch.object(FoglampMicroservice, "__init__", return_value=None):
m = mServiceThing()
# fill route table
routes.setup(app, m)
server = await test_server(app)
server.start_server(loop=loop)
# WHEN shutdown call is made at the valid URL
with patch.object(utils._logger, "info") as log:
service = ServiceRecord("d", "test", "Southbound", "http",
server.host, 1, server.port)
url_shutdown = "{}://{}:{}/foglamp/service/shutdown".format(
service._protocol, service._address,
service._management_port)
log_params1 = "Shutting down the %s service %s ...", service._type, service._name
log_params2 = 'Service %s, id %s at url %s successfully shutdown', service._name, service._id, url_shutdown
resp = await utils.shutdown_service(service, loop=loop)
# THEN shutdown returns success
assert resp is True
log.assert_called_with(*log_params2)
assert 2 == log.call_count
def _make_microservice_management_app(self):
# create web server application
self._microservice_management_app = web.Application(middlewares=[middleware.error_middleware])
# register supported urls
routes.setup(self._microservice_management_app)
# create http protocol factory for handling requests
self._microservice_management_handler = self._microservice_management_app.make_handler()
async def test_ping_service_fail_bad_url(self, test_server, loop):
# GIVEN a service is running at a given URL
app = web.Application()
with patch.object(FoglampMicroservice, "__init__", return_value=None):
m = mServiceThing()
m._start_time = time.time()
# fill route table
routes.setup(app, m)
server = await test_server(app)
server.start_server(loop=loop)
# WHEN the service is pinged with a BAD URL
with patch.object(utils._logger, "error") as log:
service = ServiceRecord("d", "test", "Southbound", "http",
server.host + "1", 1, server.port)
url_ping = "{}://{}:{}/foglamp/service/ping".format(
service._protocol, service._address,
service._management_port)
log_params = 'Ping not received for Service %s id %s at url %s attempt_count %s', service._name, service._id, \
url_ping, utils._MAX_ATTEMPTS+1
resp = await utils.ping_service(service, loop=loop)
# THEN ping response is NOT received
assert resp is False
log.assert_called_once_with(*log_params)
def _make_core_app():
"""Creates the Service management REST server Core a.k.a. service registry
:rtype: web.Application
"""
app = web.Application(middlewares=[middleware.error_middleware])
management_routes.setup(app, is_core=True)
return app
async def test_ping_service_pass(self, aiohttp_server, loop):
# GIVEN a service is running at a given URL
app = web.Application()
with patch.object(FoglampMicroservice, "__init__", return_value=None):
m = mServiceThing()
m._start_time = time.time()
# fill route table
routes.setup(app, m)
server = await aiohttp_server(app)
await server.start_server(loop=loop)
# WHEN the service is pinged with a valid URL
with patch.object(utils._logger, "info") as log:
service = ServiceRecord("d", "test", "Southbound", "http", server.host, 1, server.port)
url_ping = "{}://{}:{}/foglamp/service/ping".format(service._protocol, service._address, service._management_port)
log_params = 'Ping received for Service %s id %s at url %s', service._name, service._id, url_ping
resp = await utils.ping_service(service, loop=loop)
# THEN ping response is received
assert resp is True
log.assert_called_once_with(*log_params)
async def test_shutdown_service_fail_bad_url(self, aiohttp_server, loop):
# GIVEN a service is running at a given URL
app = web.Application()
with patch.object(FoglampMicroservice, "__init__", return_value=None):
m = mServiceThing()
# fill route table
routes.setup(app, m)
server = await aiohttp_server(app)
await server.start_server(loop=loop)
# WHEN shutdown call is made at the invalid URL
with patch.object(utils._logger, "info") as log1:
with patch.object(utils._logger, "exception") as log2:
service = ServiceRecord("d", "test", "Southbound", "http", server.host, 1, server.port+1)
log_params1 = "Shutting down the %s service %s ...", service._type, service._name
resp = await utils.shutdown_service(service, loop=loop)
# THEN shutdown fails
assert resp is False
log1.assert_called_with(*log_params1)
assert log2.called is True
def client(self, loop, test_client):
app = web.Application(middlewares=[middleware.error_middleware])
management_routes.setup(app, Server, True)
return loop.run_until_complete(test_client(app))
