From ccb11979515d3de638f9078e5a6b9d89271648dd Mon Sep 17 00:00:00 2001 From: inso <insomniak.fr@gmaiL.com> Date: Mon, 21 Nov 2016 07:30:28 +0100 Subject: [PATCH] Fix tests --- src/sakia/core/net/node.py | 6 +++--- src/sakia/tests/mocks/server.py | 14 +++++++------- src/sakia/tests/quamash_utils.py | 1 + 3 files changed, 11 insertions(+), 10 deletions(-) diff --git a/src/sakia/core/net/node.py b/src/sakia/core/net/node.py index 44b51959..9ab058ca 100644 --- a/src/sakia/core/net/node.py +++ b/src/sakia/core/net/node.py @@ -11,7 +11,7 @@ from ...tools.decorators import asyncify from duniterpy.api import bma, errors from duniterpy.api.bma import ConnectionHandler -from aiohttp.errors import WSClientDisconnectedError, WSServerHandshakeError, ClientResponseError +from aiohttp.errors import WSServerHandshakeError, ClientResponseError from aiohttp.errors import ClientError, DisconnectedError from asyncio import TimeoutError import logging @@ -367,7 +367,7 @@ class Node(QObject): break elif msg.tp == aiohttp.MsgType.error: break - except (WSServerHandshakeError, WSClientDisconnectedError, ClientResponseError, ValueError) as e: + except (WSServerHandshakeError, ClientResponseError, ValueError) as e: logging.debug("Websocket block {0} : {1} - {2}".format(type(e).__name__, str(e), self.pubkey[:5])) await self.request_current_block() except (ClientError, gaierror, TimeoutError, DisconnectedError) as e: @@ -566,7 +566,7 @@ class Node(QObject): break elif msg.tp == aiohttp.MsgType.error: break - except (WSServerHandshakeError, WSClientDisconnectedError, ClientResponseError, ValueError) as e: + except (WSServerHandshakeError, ClientResponseError, ValueError) as e: logging.debug("Websocket peer {0} : {1} - {2}".format(type(e).__name__, str(e), self.pubkey[:5])) await self.request_peers() except (ClientError, gaierror, TimeoutError, DisconnectedError) as e: diff --git a/src/sakia/tests/mocks/server.py b/src/sakia/tests/mocks/server.py index 14bb1317..c4945626 100644 --- a/src/sakia/tests/mocks/server.py +++ b/src/sakia/tests/mocks/server.py @@ -44,12 +44,7 @@ class MockServer(): self.requests = [] self.app = web.Application(loop=self.lp, middlewares=[self.middleware_factory]) - - self.handler = self.app.make_handler( - keep_alive_on=False, - access_log=log.access_logger, - ) - + self.handler = None self.port = self.find_unused_port() def get_request(self, i): @@ -90,12 +85,17 @@ class MockServer(): return peer_document_generator(self.port) async def create_server(self, ssl_ctx=None): - srv = await self.lp.create_server(self.handler, '127.0.0.1', self.port) protocol = "https" if ssl_ctx else "http" url = "{}://127.0.0.1:{}".format(protocol, self.port) self.add_route('GET', '/network/peering', bma_peering_generator(self.port)) + self.handler = self.app.make_handler( + keep_alive_on=False, + access_log=log.access_logger, + ) + + srv = await self.lp.create_server(self.handler, '127.0.0.1', self.port) return srv, self.port, url async def close(self): diff --git a/src/sakia/tests/quamash_utils.py b/src/sakia/tests/quamash_utils.py index 7c11293c..591b566d 100644 --- a/src/sakia/tests/quamash_utils.py +++ b/src/sakia/tests/quamash_utils.py @@ -10,6 +10,7 @@ class QuamashTest: def setUpQuamash(self): self.qapplication = get_application() self.lp = quamash.QSelectorEventLoop(self.qapplication) + self.qapplication.setQuitOnLastWindowClosed(False) asyncio.set_event_loop(self.lp) self.lp.set_exception_handler(lambda l, c: unitttest_exception_handler(self, l, c)) self.exceptions = [] -- GitLab