X-Git-Url: http://average.org/gitweb/?a=blobdiff_plain;f=test%2Fcommon.py;h=df430233adbff50eddca0ac7c7bb217292df3e0f;hb=dbdf9d63af31770ad57302e16b17a2fdc526773f;hp=1672f46029c663eb79436d0add7bbc5ef9259227;hpb=7bf96d53196eb6caf035335260a2dc25cec72e57;p=loctrkd.git diff --git a/test/common.py b/test/common.py index 1672f46..df43023 100644 --- a/test/common.py +++ b/test/common.py @@ -4,6 +4,7 @@ from configparser import ConfigParser, SectionProxy from contextlib import closing, ExitStack from http.server import HTTPServer, SimpleHTTPRequestHandler from importlib import import_module +from logging import DEBUG, StreamHandler from multiprocessing import Process from os import kill, unlink from signal import SIGINT @@ -16,7 +17,7 @@ from socket import ( socket, SocketType, ) -from sys import exit +from sys import exit, stderr from tempfile import mkstemp from time import sleep from typing import Optional @@ -26,7 +27,9 @@ NUMPORTS = 3 class TestWithServers(TestCase): - def setUp(self, *args: str, httpd: bool = False) -> None: + def setUp( + self, *args: str, httpd: bool = False, verbose: bool = False + ) -> None: freeports = [] with ExitStack() as stack: for _ in range(NUMPORTS): @@ -40,6 +43,7 @@ class TestWithServers(TestCase): "port": str(freeports[0]), "publishurl": "ipc://" + self.tmpfilebase + ".pub", "listenurl": "ipc://" + self.tmpfilebase + ".pul", + "protocols": "zx303proto", } self.conf["storage"] = { "dbfn": self.tmpfilebase + ".storage.sqlite", @@ -60,7 +64,10 @@ class TestWithServers(TestCase): kwargs = {"handle_hibernate": False} else: kwargs = {} - cls = import_module("gps303." + srvname, package=".") + cls = import_module("loctrkd." + srvname, package=".") + if verbose: + cls.log.addHandler(StreamHandler(stderr)) + cls.log.setLevel(DEBUG) p = Process(target=cls.runserver, args=(self.conf,), kwargs=kwargs) p.start() self.children.append((srvname, p))