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

Add test for body (~b)

Browse Source
This commit is contained in:
Shadab Zafar 2016-07-12 15:50:13 +05:30
parent 88671c3aac
commit 124b87b851
2 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mitmproxy/filt.py
View File

@ -193,12 +193,22 @@ class FBod(_Rex):
help = "Body" help = "Body"
def __call__(self, f): def __call__(self, f):
# HTTPFlow
if hasattr(f, 'request'):
if f.request and f.request.content: if f.request and f.request.content:
if self.re.search(f.request.get_decoded_content()): if self.re.search(f.request.get_decoded_content()):
return True return True
if f.response and f.response.content: if f.response and f.response.content:
if self.re.search(f.response.get_decoded_content()): if self.re.search(f.response.get_decoded_content()):
return True return True
# TCPFlow
elif hasattr(f, 'messages'):
for msg in f.messages:
if self.re.search(msg.content):
return True
return False return False

7
test/mitmproxy/test_filt.py
View File

@ -262,6 +262,13 @@ class TestMatchingTCPFlow:
e = self.err() e = self.err()
assert self.q("~e", e) assert self.q("~e", e)
def test_body(self):
f = self.flow()
assert not self.q("~b nonexistent", f)
assert self.q("~b hello", f)
assert self.q("~b me", f)
@patch('traceback.extract_tb') @patch('traceback.extract_tb')
def test_pyparsing_bug(extract_tb): def test_pyparsing_bug(extract_tb):
"""https://github.com/mitmproxy/mitmproxy/issues/1087""" """https://github.com/mitmproxy/mitmproxy/issues/1087"""