diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_log.py | 3 | ||||
| -rw-r--r-- | test/tutils.py | 1 |
2 files changed, 3 insertions, 1 deletions
diff --git a/test/test_log.py b/test/test_log.py index 0d5c6c61..2cac1e7c 100644 --- a/test/test_log.py +++ b/test/test_log.py @@ -15,8 +15,9 @@ class DummyIO(StringIO.StringIO): def test_disconnect(): outf = DummyIO() rw = DummyIO() + l = log.ConnectionLogger(outf, False, rw, rw) try: - with log.Log(outf, False, rw, rw) as lg: + with l.ctx() as lg: lg("Test") except netlib.tcp.NetLibDisconnect: pass diff --git a/test/tutils.py b/test/tutils.py index 050fa108..9d9b687d 100644 --- a/test/tutils.py +++ b/test/tutils.py @@ -98,6 +98,7 @@ class DaemonTests(object): ret.append(resp) for frm in c.wait(): ret.append(frm) + c.stop() return ret, logfp.getvalue() |