diff options
| author | Gabriel Falcão <gabriel@nacaolivre.org> | 2018-11-04 23:34:43 +0100 |
|---|---|---|
| committer | Gabriel Falcão <gabriel@nacaolivre.org> | 2018-11-04 23:34:43 +0100 |
| commit | 067c4bda72a442e1771374f9d69a5a1c77c88de5 (patch) | |
| tree | 907b5303ea0c77fbc64daa67e79920577c8edc2f /tests/functional/testserver.py | |
| parent | 2314893ec5ab46513e91ab867d7b55a72968909e (diff) | |
| download | httpretty-black.tar.gz | |
prettify code with blackblack
Diffstat (limited to 'tests/functional/testserver.py')
| -rw-r--r-- | tests/functional/testserver.py | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/tests/functional/testserver.py b/tests/functional/testserver.py index 38d2ce4..0a874ea 100644 --- a/tests/functional/testserver.py +++ b/tests/functional/testserver.py @@ -36,6 +36,7 @@ from tornado.httpserver import HTTPServer from tornado.ioloop import IOLoop from httpretty import HTTPretty from httpretty.core import old_socket as true_socket + # from httpretty.compat import PY3 from httpretty.compat import binary_type from httpretty.compat import text_type @@ -44,10 +45,11 @@ from multiprocessing import Process def utf8(s): if isinstance(s, text_type): - s = s.encode('utf-8') + s = s.encode("utf-8") return binary_type(s) + # if not PY3: # bytes = lambda s, *args: str(s) @@ -71,14 +73,14 @@ class TornadoServer(object): @classmethod def get_handlers(cls): - return Application([ - (r"/go-for-bubbles/?", BubblesHandler), - (r"/come-again/?", ComeHandler), - ]) + return Application( + [(r"/go-for-bubbles/?", BubblesHandler), (r"/come-again/?", ComeHandler)] + ) def start(self): def go(app, port, data={}): from httpretty import HTTPretty + HTTPretty.disable() http = HTTPServer(app) @@ -114,10 +116,12 @@ class TCPServer(object): def go(port): from httpretty import HTTPretty + HTTPretty.disable() import socket + s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) - s.bind(('localhost', port)) + s.bind(("localhost", port)) s.listen(True) conn, addr = s.accept() @@ -145,11 +149,11 @@ class TCPClient(object): def __init__(self, port): self.port = int(port) self.sock = true_socket(socket.AF_INET, socket.SOCK_STREAM) - self.sock.connect(('localhost', self.port)) + self.sock.connect(("localhost", self.port)) def send(self, data): if isinstance(data, text_type): - data = data.encode('utf-8') + data = data.encode("utf-8") self.sock.sendall(data) return self.sock.recv(len(data) + 11) |
