mitmproxy/test/test_pathoc.py

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_showresp(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_showreq(self):
        reqs = [ "get:/api/info:p0,0", "get:/api/info:p0,0" ]
        assert self.tval(reqs, showreq=True).count("unprintables escaped") == 2
        assert self.tval(reqs, showreq=True, hexdump=True).count("hex dump") == 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"])