mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 11:19:23 +00:00
60 lines
1.7 KiB
Python
60 lines
1.7 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, s
|
|
)
|
|
assert "Timeout" in s.getvalue()
|
|
|
|
def tval(self, requests, verbose=False):
|
|
c = pathoc.Pathoc("127.0.0.1", self.d.port)
|
|
c.connect()
|
|
s = cStringIO.StringIO()
|
|
c.print_requests(requests, verbose, True, s)
|
|
return s.getvalue()
|
|
|
|
def test_print_requests(self):
|
|
reqs = [ "get:/api/info:p0,0", "get:/api/info:p0,0" ]
|
|
assert self.tval(reqs, False).count("200") == 2
|
|
assert self.tval(reqs, 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_fileread(self):
|
|
d = tutils.test_data.path("data/request")
|
|
assert "foo" in self.tval(["+%s"%d])
|
|
assert "File" in self.tval(["+/nonexistent"])
|