Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 16:17:49 +00:00
Code Issues Projects Releases Wiki Activity
c8f1b3d7f9
mitmproxy/test/examples/test_examples.py

64 lines
2.5 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
The final piece: netlib -> mitproxy.net
2016-10-19 22:56:38 +00:00
from mitmproxy.net.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:
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
a = tctx.script(tdata.path("../examples/addons/scripting-minimal-example.py"))
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
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