Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

asyncio: fix a few remaining issues in proxy/test_server.py

Browse Source
This commit is contained in:
Aldo Cortesi 2018-04-02 09:46:02 +12:00
parent 3cc5d81a4a
commit 28a8ddc0e8
2 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/mitmproxy/proxy/test_server.py
View File

@ -987,7 +987,7 @@ class TestUpstreamProxySSL(
Client <- HTTPS -> Proxy <- HTTP -> Proxy <- HTTPS -> Server
"""
self.set_addons(RewriteToHttp())
self.set_addons(RewriteToHttps())
self.chain[1].set_addons(RewriteToHttps())
p = self.pathoc()
with p.connect():
resp = p.request("get:'/p/418'")

21
test/mitmproxy/tservers.py
View File

@ -117,6 +117,11 @@ class ProxyThread(threading.Thread):
)
self.tmaster.run()
def set_addons(self, *addons):
self.tmaster.reset(addons)
self.tmaster.addons.trigger("tick")
class ProxyTestBase:
# Test Configuration
@ -180,8 +185,7 @@ class ProxyTestBase:
)
def set_addons(self, *addons):
self.proxy.tmaster.reset(addons)
self.proxy.tmaster.addons.trigger("tick")
self.proxy.set_addons(*addons)
def addons(self):
"""
@ -337,8 +341,7 @@ class SocksModeTest(HTTPProxyTest):
return opts
class ChainProxyTest(ProxyTestBase):
class HTTPUpstreamProxyTest(HTTPProxyTest):
"""
Chain three instances of mitmproxy in a row to test upstream mode.
Proxy order is cls.proxy -> cls.chain[0] -> cls.chain[1]
@ -357,6 +360,12 @@ class ChainProxyTest(ProxyTestBase):
proxy = ProxyThread(cls.masterclass, opts)
proxy.start()
cls.chain.insert(0, proxy)
while 1:
if(
proxy.event_loop and
proxy.event_loop.is_running()
):
break
super().setup_class()
@ -380,7 +389,3 @@ class ChainProxyTest(ProxyTestBase):
mode="upstream:" + s,
)
return opts
class HTTPUpstreamProxyTest(ChainProxyTest, HTTPProxyTest):
pass