Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-27 02:24:18 +00:00
Code Issues Projects Releases Wiki Activity

http2: fix tests

Browse Source
This commit is contained in:
Thomas Kriechbaumer 2015-08-21 09:18:14 +02:00
parent 00ed982ea0
commit 6fc2ff9469
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/http/http2/test_protocol.py
View File

@ -127,7 +127,7 @@ class TestPerformServerConnectionPreface(tservers.ServerTestBase):
protocol.perform_server_connection_preface() protocol.perform_server_connection_preface()
assert protocol.connection_preface_performed assert protocol.connection_preface_performed
tutils.raises(tcp.NetLibIncomplete, protocol.perform_server_connection_preface, force=True) tutils.raises(tcp.NetLibDisconnect, protocol.perform_server_connection_preface, force=True)
class TestPerformClientConnectionPreface(tservers.ServerTestBase): class TestPerformClientConnectionPreface(tservers.ServerTestBase):
@ -194,12 +194,12 @@ class TestServerStreamIds():
class TestApplySettings(tservers.ServerTestBase): class TestApplySettings(tservers.ServerTestBase):
class handler(tcp.BaseHandler): class handler(tcp.BaseHandler):
def handle(self): def handle(self):
# check settings acknowledgement # check settings acknowledgement
assert self.rfile.read(9) == '000000040100000000'.decode('hex') assert self.rfile.read(9) == '000000040100000000'.decode('hex')
self.wfile.write("OK") self.wfile.write("OK")
self.wfile.flush() self.wfile.flush()
self.rfile.safe_read(9) # just to keep the connection alive a bit longer
ssl = True ssl = True
@ -295,6 +295,7 @@ class TestReadRequest(tservers.ServerTestBase):
self.wfile.write( self.wfile.write(
b'000006000100000001666f6f626172'.decode('hex')) b'000006000100000001666f6f626172'.decode('hex'))
self.wfile.flush() self.wfile.flush()
self.rfile.safe_read(9) # just to keep the connection alive a bit longer
ssl = True ssl = True
@ -385,13 +386,13 @@ class TestReadRequestConnect(tservers.ServerTestBase):
class TestReadResponse(tservers.ServerTestBase): class TestReadResponse(tservers.ServerTestBase):
class handler(tcp.BaseHandler): class handler(tcp.BaseHandler):
def handle(self): def handle(self):
self.wfile.write( self.wfile.write(
b'00000801040000000188628594e78c767f'.decode('hex')) b'00000801040000000188628594e78c767f'.decode('hex'))
self.wfile.write( self.wfile.write(
b'000006000100000001666f6f626172'.decode('hex')) b'000006000100000001666f6f626172'.decode('hex'))
self.wfile.flush() self.wfile.flush()
self.rfile.safe_read(9) # just to keep the connection alive a bit longer
ssl = True ssl = True