|
| 1 | +import importlib.metadata |
| 2 | + |
1 | 3 | import click
|
2 | 4 | import pytest
|
3 |
| -import werkzeug |
4 | 5 |
|
5 | 6 | import flask
|
6 | 7 | from flask import appcontext_popped
|
@@ -38,34 +39,35 @@ def index():
|
38 | 39 | assert rv.data == b"http://localhost/"
|
39 | 40 |
|
40 | 41 |
|
41 |
| -def test_environ_base_default(app, client, app_ctx): |
| 42 | +def test_environ_base_default(app, client): |
42 | 43 | @app.route("/")
|
43 | 44 | def index():
|
44 |
| - flask.g.user_agent = flask.request.headers["User-Agent"] |
45 |
| - return flask.request.remote_addr |
| 45 | + flask.g.remote_addr = flask.request.remote_addr |
| 46 | + flask.g.user_agent = flask.request.user_agent.string |
| 47 | + return "" |
46 | 48 |
|
47 |
| - rv = client.get("/") |
48 |
| - assert rv.data == b"127.0.0.1" |
49 |
| - assert flask.g.user_agent == f"werkzeug/{werkzeug.__version__}" |
| 49 | + with client: |
| 50 | + client.get("/") |
| 51 | + assert flask.g.remote_addr == "127.0.0.1" |
| 52 | + assert flask.g.user_agent == ( |
| 53 | + f"Werkzeug/{importlib.metadata.version('werkzeug')}" |
| 54 | + ) |
50 | 55 |
|
51 | 56 |
|
52 |
| -def test_environ_base_modified(app, client, app_ctx): |
| 57 | +def test_environ_base_modified(app, client): |
53 | 58 | @app.route("/")
|
54 | 59 | def index():
|
55 |
| - flask.g.user_agent = flask.request.headers["User-Agent"] |
56 |
| - return flask.request.remote_addr |
| 60 | + flask.g.remote_addr = flask.request.remote_addr |
| 61 | + flask.g.user_agent = flask.request.user_agent.string |
| 62 | + return "" |
57 | 63 |
|
58 |
| - client.environ_base["REMOTE_ADDR"] = "0.0.0.0" |
| 64 | + client.environ_base["REMOTE_ADDR"] = "192.168.0.22" |
59 | 65 | client.environ_base["HTTP_USER_AGENT"] = "Foo"
|
60 |
| - rv = client.get("/") |
61 |
| - assert rv.data == b"0.0.0.0" |
62 |
| - assert flask.g.user_agent == "Foo" |
63 | 66 |
|
64 |
| - client.environ_base["REMOTE_ADDR"] = "0.0.0.1" |
65 |
| - client.environ_base["HTTP_USER_AGENT"] = "Bar" |
66 |
| - rv = client.get("/") |
67 |
| - assert rv.data == b"0.0.0.1" |
68 |
| - assert flask.g.user_agent == "Bar" |
| 67 | + with client: |
| 68 | + client.get("/") |
| 69 | + assert flask.g.remote_addr == "192.168.0.22" |
| 70 | + assert flask.g.user_agent == "Foo" |
69 | 71 |
|
70 | 72 |
|
71 | 73 | def test_client_open_environ(app, client, request):
|
|
0 commit comments