def test_(url, config, expected):
http = HTTP(None)
mw_cls = loader.get_class("urlsession")
middleware = mw_cls(http, **config)
# No session yet
r_url, r_body, r_headers = middleware.process_request(url, {}, {})
assert not r_body # Should not be set
assert not r_headers # Should not be set
assert r_url == url # Should not be changed due to empty session
# Set session as number
session_id = 123
http.set_session_id(session_id)
r_url, r_body, r_headers = middleware.process_request(url, {}, {})
assert not r_body # Should not be set
assert not r_headers # Should not be set
r_expected = expected.format(session_id)
assert r_url == r_expected # Should not be changed due to empty session
# Set session as string
session_id = "12345"
http.set_session_id(session_id)
r_url, r_body, r_headers = middleware.process_request(url, {}, {})
assert not r_body # Should not be set
assert not r_headers # Should not be set
r_expected = expected.format(session_id)
assert r_url == r_expected # Should not be changed due to empty session
def test_jsonsession(body, config, session_id, expected):
http = HTTP(None)
http.set_session_id(session_id)
mw_cls = loader.get_class("jsonsession")
middleware = mw_cls(http, **config)
url = "http://127.0.0.1"
# POST
r_url, r_body, r_headers = middleware.process_post(url, body.copy(), {})
assert r_body == expected
assert not r_headers # Should not be set
assert r_url == url # Should not be changed
# PUT
r_url, r_body, r_headers = middleware.process_put(url, body.copy(), {})
assert r_body == expected
assert not r_headers # Should not be set
assert r_url == url # Should not be changed