Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 18:18:25 +00:00
Code Issues Projects Releases Wiki Activity
6dc6269b8e
mitmproxy/test/examples/test_examples.py

98 lines
4.2 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
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
Revamp how addons work - Addons now nest, which means that addons can manage addons. This has a number of salutary effects - the scripts addon no longer has to poke into the global addons list, we no longer have to replace/remove/boot-outof parent addons when we load scripts, and this paves the way for making our top-level tools into addons themselves. - All addon calls are now wrapped in a safe execution environment where exceptions are caught, and output to stdout/stderr are intercepted and turned into logs. - We no longer support script arguments in sys.argv - creating an option properly is the only way to pass arguments. This means that all scripts are always directly controllable from interctive tooling, and that arguments are type-checked. For now, I've disabled testing of the har dump example - it needs to be moved to the new argument handling, and become a class addon. I'll address that in a separate patch.
2017-04-25 07:06:24 +00:00
from mitmproxy.test import taddons
move Headers/Message/Request/Response into mitmproxy.http
2021-02-04 01:14:27 +00:00
from mitmproxy.http import Headers
Add simple har_dump test
2016-08-03 10:58:41 +00:00
move examples tests (#2199)
2017-03-22 11:02:18 +00:00
from ..mitmproxy import tservers
Add tests for modify_form example
2016-03-07 03:42:10 +00:00
add MultiDict This commit introduces MultiDict, a multi-dictionary similar to ODict, but with improved semantics (as in the Headers class). MultiDict fixes a few issues that were present in the Request/Response API. In particular, `request.cookies["foo"] = "bar"` has previously been a no-op, as the cookies property returned a mutable _copy_ of the cookies.
2016-05-19 01:46:42 +00:00
coverage++
2017-02-10 21:12:24 +00:00
class TestScripts(tservers.MasterTest):
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>")
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(
port=443,
host="example.com",
path="/ReverseGeocode"
)
sim_msg_resp = tutils.tresp()
sim_flow = tflow.tflow(
req=sim_msg_req,
resp=sim_msg_resp
)
view_text, output = v(raw, flow=sim_flow, http_message=sim_flow.request) # simulate request message
assert view_text == "Protobuf (flattened) (addon with custom rules)"
output = list(output) # assure list conversion if generator
assert output == [
[('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 ')]
]
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:
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-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"
Reference in New Issue Copy Permalink