[telepathy-gabble/master] move install_colourer() to servicetest.py
Dafydd Harries
dafydd.harries at collabora.co.uk
Tue Sep 8 03:58:02 PDT 2009
---
tests/twisted/gabbletest.py | 26 +-------------------------
tests/twisted/servicetest.py | 25 +++++++++++++++++++++++++
2 files changed, 26 insertions(+), 25 deletions(-)
diff --git a/tests/twisted/gabbletest.py b/tests/twisted/gabbletest.py
index dce3bfe..24ed95f 100644
--- a/tests/twisted/gabbletest.py
+++ b/tests/twisted/gabbletest.py
@@ -342,30 +342,6 @@ def make_stream(event_func, authenticator=None, protocol=None, port=4242, resour
port = reactor.listenTCP(port, factory)
return (stream, port)
-def install_colourer():
- def red(s):
- return '\x1b[31m%s\x1b[0m' % s
-
- def green(s):
- return '\x1b[32m%s\x1b[0m' % s
-
- patterns = {
- 'handled': green,
- 'not handled': red,
- }
-
- class Colourer:
- def __init__(self, fh, patterns):
- self.fh = fh
- self.patterns = patterns
-
- def write(self, s):
- f = self.patterns.get(s, lambda x: x)
- self.fh.write(f(s))
-
- sys.stdout = Colourer(sys.stdout, patterns)
- return sys.stdout
-
def disconnect_conn(q, conn, stream, expected=[]):
call_async(q, conn, 'Disconnect')
@@ -386,7 +362,7 @@ def exec_test_deferred(fun, params, protocol=None, timeout=None,
colourer = None
if sys.stdout.isatty():
- colourer = install_colourer()
+ colourer = servicetest.install_colourer()
queue = servicetest.IteratingEventQueue(timeout)
queue.verbose = (
diff --git a/tests/twisted/servicetest.py b/tests/twisted/servicetest.py
index 0b8fe18..129c0d6 100644
--- a/tests/twisted/servicetest.py
+++ b/tests/twisted/servicetest.py
@@ -6,6 +6,7 @@ Infrastructure code for testing connection managers.
from twisted.internet import glib2reactor
from twisted.internet.protocol import Protocol, Factory, ClientFactory
glib2reactor.install()
+import sys
import pprint
import unittest
@@ -429,6 +430,30 @@ def assertLength(length, value):
"expected: length %d, got length %d:\n%s" % (
length, len(value), pretty(value))
+def install_colourer():
+ def red(s):
+ return '\x1b[31m%s\x1b[0m' % s
+
+ def green(s):
+ return '\x1b[32m%s\x1b[0m' % s
+
+ patterns = {
+ 'handled': green,
+ 'not handled': red,
+ }
+
+ class Colourer:
+ def __init__(self, fh, patterns):
+ self.fh = fh
+ self.patterns = patterns
+
+ def write(self, s):
+ f = self.patterns.get(s, lambda x: x)
+ self.fh.write(f(s))
+
+ sys.stdout = Colourer(sys.stdout, patterns)
+ return sys.stdout
+
if __name__ == '__main__':
unittest.main()
--
1.5.6.5
More information about the telepathy-commits
mailing list