mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
import json, cStringIO
|
|
from libpathod import pathoc, test, version
|
|
import tutils
|
|
|
|
|
|
class TestDaemon:
|
|
@classmethod
|
|
def setUpAll(self):
|
|
self.d = test.Daemon(
|
|
staticdir=tutils.test_data.path("data"),
|
|
anchors=[("/anchor/.*", "202")]
|
|
)
|
|
|
|
@classmethod
|
|
def tearDownAll(self):
|
|
self.d.shutdown()
|
|
|
|
def setUp(self):
|
|
self.d.clear_log()
|
|
|
|
def test_info(self):
|
|
c = pathoc.Pathoc("127.0.0.1", self.d.port)
|
|
c.connect()
|
|
_, _, _, _, content = c.request("get:/api/info")
|
|
assert tuple(json.loads(content)["version"]) == version.IVERSION
|
|
|
|
def test_timeout(self):
|
|
c = pathoc.Pathoc("127.0.0.1", self.d.port)
|
|
c.connect()
|
|
c.settimeout(0.01)
|
|
|
|
s = cStringIO.StringIO()
|
|
c.print_requests(
|
|
["get:'/p/200:p0,10'"], True, True, True, True, s
|
|
)
|
|
assert "Timeout" in s.getvalue()
|
|
|
|
def tval(self, requests, showreq=False, showresp=False, explain=False, hexdump=False):
|
|
c = pathoc.Pathoc("127.0.0.1", self.d.port)
|
|
c.connect()
|
|
s = cStringIO.StringIO()
|
|
c.print_requests(
|
|
requests,
|
|
showreq = showreq,
|
|
showresp = showresp,
|
|
explain = explain,
|
|
hexdump = hexdump,
|
|
fp = s
|
|
)
|
|
return s.getvalue()
|
|
|
|
def test_print_requests(self):
|
|
reqs = [ "get:/api/info:p0,0", "get:/api/info:p0,0" ]
|
|
assert self.tval(reqs).count("200") == 2
|
|
assert self.tval(reqs, showresp=True).count("Date") == 2
|
|
|
|
def test_parse_err(self):
|
|
assert "Error parsing" in self.tval(["foo"])
|
|
|
|
def test_conn_err(self):
|
|
assert "Invalid server response" in self.tval(["get:'/p/200:d2'"])
|
|
|
|
def test_explain(self):
|
|
reqs = [ "get:/api/info:ir,'x'"]
|
|
assert "inject" in self.tval(reqs, explain=True)
|
|
|
|
def test_fileread(self):
|
|
d = tutils.test_data.path("data/request")
|
|
assert "foo" in self.tval(["+%s"%d], explain=True)
|
|
assert "File" in self.tval(["+/nonexistent"])
|
|
|