77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
proxy.py
|
||
|
|
~~~~~~~~
|
||
|
|
⚡⚡⚡ Fast, Lightweight, Programmable Proxy Server in a single Python file.
|
||
|
|
|
||
|
|
:copyright: (c) 2013-present by Abhinav Singh and contributors.
|
||
|
|
:license: BSD, see LICENSE for more details.
|
||
|
|
"""
|
||
|
|
import json
|
||
|
|
import http.client
|
||
|
|
import urllib.request
|
||
|
|
|
||
|
|
from proxy import TestCase
|
||
|
|
from proxy.common.constants import DEFAULT_CLIENT_RECVBUF_SIZE, PROXY_AGENT_HEADER_VALUE
|
||
|
|
from proxy.common.utils import socket_connection, build_http_request, build_http_response
|
||
|
|
from proxy.http.codes import httpStatusCodes
|
||
|
|
from proxy.http.methods import httpMethods
|
||
|
|
|
||
|
|
|
||
|
|
class TestProxyPyEmbedded(TestCase):
|
||
|
|
"""This test case is a demonstration of proxy.TestCase and also serves as
|
||
|
|
integration test suite for proxy.py."""
|
||
|
|
|
||
|
|
PROXY_PY_STARTUP_FLAGS = TestCase.DEFAULT_PROXY_PY_STARTUP_FLAGS + [
|
||
|
|
'--enable-web-server',
|
||
|
|
]
|
||
|
|
|
||
|
|
def test_with_proxy(self) -> None:
|
||
|
|
"""Makes a HTTP request to in-build web server via proxy server."""
|
||
|
|
with socket_connection(('localhost', self.PROXY_PORT)) as conn:
|
||
|
|
conn.send(
|
||
|
|
build_http_request(
|
||
|
|
httpMethods.GET, b'http://localhost:%d/' % self.PROXY_PORT,
|
||
|
|
headers={
|
||
|
|
b'Host': b'localhost:%d' % self.PROXY_PORT,
|
||
|
|
})
|
||
|
|
)
|
||
|
|
response = conn.recv(DEFAULT_CLIENT_RECVBUF_SIZE)
|
||
|
|
self.assertEqual(
|
||
|
|
response,
|
||
|
|
build_http_response(
|
||
|
|
httpStatusCodes.NOT_FOUND, reason=b'NOT FOUND',
|
||
|
|
headers={
|
||
|
|
b'Server': PROXY_AGENT_HEADER_VALUE,
|
||
|
|
b'Connection': b'close'
|
||
|
|
}
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_proxy_vcr(self) -> None:
|
||
|
|
"""With VCR enabled, proxy.py will cache responses for all HTTP(s)
|
||
|
|
requests made during the test. When test is re-run, until explicitly
|
||
|
|
disabled, proxy.py will replay responses from cache avoiding calls to
|
||
|
|
upstream servers.
|
||
|
|
|
||
|
|
This feature only works iff proxy.py is used as a proxy server
|
||
|
|
for all HTTP(s) requests made during the test.
|
||
|
|
|
||
|
|
Below we make a HTTP GET request using Python's urllib library."""
|
||
|
|
with self.vcr():
|
||
|
|
self.make_http_request_using_proxy()
|
||
|
|
|
||
|
|
def test_proxy_no_vcr(self) -> None:
|
||
|
|
self.make_http_request_using_proxy()
|
||
|
|
|
||
|
|
def make_http_request_using_proxy(self) -> None:
|
||
|
|
proxy_handler = urllib.request.ProxyHandler({
|
||
|
|
'http': 'http://localhost:%d' % self.PROXY_PORT,
|
||
|
|
})
|
||
|
|
opener = urllib.request.build_opener(proxy_handler)
|
||
|
|
r: http.client.HTTPResponse = opener.open('http://httpbin.org/get')
|
||
|
|
self.assertEqual(r.status, 200)
|
||
|
|
data = json.loads(r.read(DEFAULT_CLIENT_RECVBUF_SIZE))
|
||
|
|
self.assertEqual(data['args'], {})
|
||
|
|
self.assertEqual(data['headers']['Host'], 'httpbin.org')
|