Rooba
/
mitmproxy
mirror of https://github.com/mitmproxy/mitmproxy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity
mitmproxy/test/examples/test_examples.py

125 lines
4.7 KiB
Python
Raw Normal View History

Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
from mitmproxy import contentviews
[autofix.ci] apply automated fixes
2022-11-29 13:28:41 +00:00
from mitmproxy.http import Headers
from mitmproxy.test import taddons
replace tutils.raises with pytest.raises + shim
2017-02-01 15:48:46 +00:00
from mitmproxy.test import tflow
tests: clean up test/mitmproxy/tutils.py Remove aliases, use mitmproxy/test/tutils instead.
2016-11-01 20:44:18 +00:00
from mitmproxy.test import tutils
Add simple har_dump test
2016-08-03 10:58:41 +00:00
Add tests for modify_form example
2016-03-07 03:42:10 +00:00
tests++
2022-03-15 17:33:46 +00:00
class TestScripts:
test: shift test_data out of our public API
2018-04-22 23:05:58 +00:00
def test_add_header(self, tdata):
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
with taddons.context() as tctx:
docs++
2021-02-09 00:05:02 +00:00
a = tctx.script(tdata.path("../examples/addons/anatomy2.py"))
restructure examples - restructure examples (fix #4031) - remove example dependencies from setup.py, we do not need special dependencies for our supported addons. - unify how we generate docs from code - improve example docs
2020-06-22 23:53:39 +00:00
f = tflow.tflow()
a.request(f)
assert f.request.headers["myheader"] == "value"
Move har_extractor tests to the examples file
2016-03-09 18:21:29 +00:00
test: shift test_data out of our public API
2018-04-22 23:05:58 +00:00
def test_custom_contentviews(self, tdata):
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
with taddons.context() as tctx:
restructure examples - restructure examples (fix #4031) - remove example dependencies from setup.py, we do not need special dependencies for our supported addons. - unify how we generate docs from code - improve example docs
2020-06-22 23:53:39 +00:00
tctx.script(tdata.path("../examples/addons/contentview.py"))
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
swapcase = contentviews.get("swapcase")
_, fmt = swapcase(b"<html>Test!</html>")
make it black!
2022-04-26 11:53:35 +00:00
assert any(b"tEST!" in val[0][1] for val in fmt)
disable harparser on py3
2016-07-07 08:21:15 +00:00
Grpc contentview (#4851) * Partial gRPC contentview prototype, not linted, no tests, not as add-on * Linted (flake8) * Save dev state * Rewrote of protobuf parser, use decoding strategy, reduced rendered data. Parser uses generators * minor cleanup * fix: preferred encoding was provided as function instead of value * flake8: line length * Backlinked message tree objects, temporary debug out * Partial implementation of gRPC definitions. Save state to fix a cras (data invalidate in edit mode) * hack: deal with missing exception handling for generator based content views * gRPC/Protoparser descriptions (with test code) * replaced manual gzip decoding with mitmproxy.net.encoding.decode * Refactored typing imports * Reafctoring * distinguish request vs response definitions, separate view config from parser config * Code cleaning, moved customized protobuf definitions to example addon * final cleanup * changelog * Stubs for tests * Fixed render_riority of addon example * Started adding tests * Work on tests * mypy * Added pseudo encoder to tests, to cover special decodings * Example addon test added * finalized tests, no 100 percent coverage possible, see comments un uncovered code * minor adjustments * fixup tests * Typos Co-authored-by: Maximilian Hils <git@maximilianhils.com>
2021-10-12 11:32:56 +00:00
def test_custom_grpc_contentview(self, tdata):
with taddons.context() as tctx:
tctx.script(tdata.path("../examples/addons/contentview-custom-grpc.py"))
v = contentviews.get("customized gRPC/protobuf")
p = tdata.path("mitmproxy/contentviews/test_grpc_data/msg1.bin")
with open(p, "rb") as f:
raw = f.read()
sim_msg_req = tutils.treq(
make it black!
2022-04-26 11:53:35 +00:00
port=443, host="example.com", path="/ReverseGeocode"
Grpc contentview (#4851) * Partial gRPC contentview prototype, not linted, no tests, not as add-on * Linted (flake8) * Save dev state * Rewrote of protobuf parser, use decoding strategy, reduced rendered data. Parser uses generators * minor cleanup * fix: preferred encoding was provided as function instead of value * flake8: line length * Backlinked message tree objects, temporary debug out * Partial implementation of gRPC definitions. Save state to fix a cras (data invalidate in edit mode) * hack: deal with missing exception handling for generator based content views * gRPC/Protoparser descriptions (with test code) * replaced manual gzip decoding with mitmproxy.net.encoding.decode * Refactored typing imports * Reafctoring * distinguish request vs response definitions, separate view config from parser config * Code cleaning, moved customized protobuf definitions to example addon * final cleanup * changelog * Stubs for tests * Fixed render_riority of addon example * Started adding tests * Work on tests * mypy * Added pseudo encoder to tests, to cover special decodings * Example addon test added * finalized tests, no 100 percent coverage possible, see comments un uncovered code * minor adjustments * fixup tests * Typos Co-authored-by: Maximilian Hils <git@maximilianhils.com>
2021-10-12 11:32:56 +00:00
)
sim_msg_resp = tutils.tresp()
make it black!
2022-04-26 11:53:35 +00:00
sim_flow = tflow.tflow(req=sim_msg_req, resp=sim_msg_resp)
Grpc contentview (#4851) * Partial gRPC contentview prototype, not linted, no tests, not as add-on * Linted (flake8) * Save dev state * Rewrote of protobuf parser, use decoding strategy, reduced rendered data. Parser uses generators * minor cleanup * fix: preferred encoding was provided as function instead of value * flake8: line length * Backlinked message tree objects, temporary debug out * Partial implementation of gRPC definitions. Save state to fix a cras (data invalidate in edit mode) * hack: deal with missing exception handling for generator based content views * gRPC/Protoparser descriptions (with test code) * replaced manual gzip decoding with mitmproxy.net.encoding.decode * Refactored typing imports * Reafctoring * distinguish request vs response definitions, separate view config from parser config * Code cleaning, moved customized protobuf definitions to example addon * final cleanup * changelog * Stubs for tests * Fixed render_riority of addon example * Started adding tests * Work on tests * mypy * Added pseudo encoder to tests, to cover special decodings * Example addon test added * finalized tests, no 100 percent coverage possible, see comments un uncovered code * minor adjustments * fixup tests * Typos Co-authored-by: Maximilian Hils <git@maximilianhils.com>
2021-10-12 11:32:56 +00:00
make it black!
2022-04-26 11:53:35 +00:00
view_text, output = v(
raw, flow=sim_flow, http_message=sim_flow.request
) # simulate request message
Grpc contentview (#4851) * Partial gRPC contentview prototype, not linted, no tests, not as add-on * Linted (flake8) * Save dev state * Rewrote of protobuf parser, use decoding strategy, reduced rendered data. Parser uses generators * minor cleanup * fix: preferred encoding was provided as function instead of value * flake8: line length * Backlinked message tree objects, temporary debug out * Partial implementation of gRPC definitions. Save state to fix a cras (data invalidate in edit mode) * hack: deal with missing exception handling for generator based content views * gRPC/Protoparser descriptions (with test code) * replaced manual gzip decoding with mitmproxy.net.encoding.decode * Refactored typing imports * Reafctoring * distinguish request vs response definitions, separate view config from parser config * Code cleaning, moved customized protobuf definitions to example addon * final cleanup * changelog * Stubs for tests * Fixed render_riority of addon example * Started adding tests * Work on tests * mypy * Added pseudo encoder to tests, to cover special decodings * Example addon test added * finalized tests, no 100 percent coverage possible, see comments un uncovered code * minor adjustments * fixup tests * Typos Co-authored-by: Maximilian Hils <git@maximilianhils.com>
2021-10-12 11:32:56 +00:00
assert view_text == "Protobuf (flattened) (addon with custom rules)"
output = list(output) # assure list conversion if generator
assert output == [
make it black!
2022-04-26 11:53:35 +00:00
[
("text", "[message] "),
("text", "position "),
("text", "1 "),
("text", " "),
],
[
("text", "[double] "),
("text", "latitude "),
("text", "1.1 "),
("text", "38.89816675798073 "),
],
[
("text", "[double] "),
("text", "longitude "),
("text", "1.2 "),
("text", "-77.03829828366696 "),
],
[
("text", "[string] "),
("text", "country "),
("text", "3 "),
("text", "de_DE "),
],
[
("text", "[uint32] "),
("text", " "),
("text", "6 "),
("text", "1 "),
],
[
("text", "[string] "),
("text", "app "),
("text", "7 "),
("text", "de.mcdonalds.mcdonaldsinfoapp "),
],
Grpc contentview (#4851) * Partial gRPC contentview prototype, not linted, no tests, not as add-on * Linted (flake8) * Save dev state * Rewrote of protobuf parser, use decoding strategy, reduced rendered data. Parser uses generators * minor cleanup * fix: preferred encoding was provided as function instead of value * flake8: line length * Backlinked message tree objects, temporary debug out * Partial implementation of gRPC definitions. Save state to fix a cras (data invalidate in edit mode) * hack: deal with missing exception handling for generator based content views * gRPC/Protoparser descriptions (with test code) * replaced manual gzip decoding with mitmproxy.net.encoding.decode * Refactored typing imports * Reafctoring * distinguish request vs response definitions, separate view config from parser config * Code cleaning, moved customized protobuf definitions to example addon * final cleanup * changelog * Stubs for tests * Fixed render_riority of addon example * Started adding tests * Work on tests * mypy * Added pseudo encoder to tests, to cover special decodings * Example addon test added * finalized tests, no 100 percent coverage possible, see comments un uncovered code * minor adjustments * fixup tests * Typos Co-authored-by: Maximilian Hils <git@maximilianhils.com>
2021-10-12 11:32:56 +00:00
]
test: shift test_data out of our public API
2018-04-22 23:05:58 +00:00
def test_modify_form(self, tdata):
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
with taddons.context() as tctx:
restructure examples - restructure examples (fix #4031) - remove example dependencies from setup.py, we do not need special dependencies for our supported addons. - unify how we generate docs from code - improve example docs
2020-06-22 23:53:39 +00:00
sc = tctx.script(tdata.path("../examples/addons/http-modify-form.py"))
Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
form_header = Headers(content_type="application/x-www-form-urlencoded")
f = tflow.tflow(req=tutils.treq(headers=form_header))
sc.request(f)
Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
assert f.request.urlencoded_form["mitmproxy"] == "rocks"
Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
f.request.headers["content-type"] = ""
sc.request(f)
assert list(f.request.urlencoded_form.items()) == [("foo", "bar")]
Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
test: shift test_data out of our public API
2018-04-22 23:05:58 +00:00
def test_modify_querystring(self, tdata):
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
with taddons.context() as tctx:
make it black!
2022-04-26 11:53:35 +00:00
sc = tctx.script(
tdata.path("../examples/addons/http-modify-query-string.py")
)
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
f = tflow.tflow(req=tutils.treq(path="/search?q=term"))
Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
sc.request(f)
assert f.request.query["mitmproxy"] == "rocks"
Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
f.request.path = "/"
sc.request(f)
assert f.request.query["mitmproxy"] == "rocks"
Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
test: shift test_data out of our public API
2018-04-22 23:05:58 +00:00
def test_redirect_requests(self, tdata):
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
with taddons.context() as tctx:
restructure examples - restructure examples (fix #4031) - remove example dependencies from setup.py, we do not need special dependencies for our supported addons. - unify how we generate docs from code - improve example docs
2020-06-22 23:53:39 +00:00
sc = tctx.script(tdata.path("../examples/addons/http-redirect-requests.py"))
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
f = tflow.tflow(req=tutils.treq(host="example.org"))
sc.request(f)
assert f.request.host == "mitmproxy.org"
Convert examples and example tests for new-style scripts Remove the test that just loads all the example scripts for now - it's a very low-value test, and we need to think of something better.
2016-07-14 04:20:27 +00:00
test: shift test_data out of our public API
2018-04-22 23:05:58 +00:00
def test_send_reply_from_proxy(self, tdata):
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
with taddons.context() as tctx:
restructure examples - restructure examples (fix #4031) - remove example dependencies from setup.py, we do not need special dependencies for our supported addons. - unify how we generate docs from code - improve example docs
2020-06-22 23:53:39 +00:00
sc = tctx.script(tdata.path("../examples/addons/http-reply-from-proxy.py"))
Remove watchdog, solidify script testing - Remove the watchdog dependency. We now just stat the script file every 2 seconds to check for an updated mtime. - Further solidify our script testing, and in particular make the example tests nicer. These should exemplify how we want users to test their own addon scripts. More work on addon testing to follow.
2017-04-25 19:13:36 +00:00
f = tflow.tflow(req=tutils.treq(host="example.com", port=80))
sc.request(f)
assert f.response.content == b"Hello World"