c3ed2aa479
Using request.netloc can be confusing because you have to check whether or not the port is included. We are going to want this seperation in later patches so expose it on the request. Change-Id: I2e4bad425fdbc2501727b295752900d4ce4086bc
120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import uuid
|
|
|
|
import requests
|
|
import requests_mock
|
|
from requests_mock.tests import base
|
|
|
|
|
|
class RequestTests(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(RequestTests, self).setUp()
|
|
|
|
self.mocker = requests_mock.Mocker()
|
|
self.addCleanup(self.mocker.stop)
|
|
self.mocker.start()
|
|
|
|
def do_request(self, **kwargs):
|
|
method = kwargs.pop('method', 'GET')
|
|
url = kwargs.pop('url', 'http://test.example.com/path')
|
|
status_code = kwargs.pop('status_code', 200)
|
|
data = uuid.uuid4().hex
|
|
|
|
m = self.mocker.register_uri(method,
|
|
url,
|
|
text=data,
|
|
status_code=status_code)
|
|
|
|
resp = requests.request(method, url, **kwargs)
|
|
|
|
self.assertEqual(status_code, resp.status_code)
|
|
self.assertEqual(data, resp.text)
|
|
|
|
self.assertTrue(m.called_once)
|
|
return m.last_request
|
|
|
|
def test_base_params(self):
|
|
req = self.do_request(method='GET', status_code=200)
|
|
|
|
self.assertIs(None, req.allow_redirects)
|
|
self.assertIs(None, req.timeout)
|
|
self.assertIs(True, req.verify)
|
|
self.assertIs(None, req.cert)
|
|
|
|
# actually it's an OrderedDict, but equality works fine
|
|
self.assertEqual({}, req.proxies)
|
|
|
|
def test_allow_redirects(self):
|
|
req = self.do_request(allow_redirects=False, status_code=300)
|
|
self.assertFalse(req.allow_redirects)
|
|
|
|
def test_timeout(self):
|
|
timeout = 300
|
|
req = self.do_request(timeout=timeout)
|
|
self.assertEqual(timeout, req.timeout)
|
|
|
|
def test_verify_false(self):
|
|
verify = False
|
|
req = self.do_request(verify=verify)
|
|
self.assertIs(verify, req.verify)
|
|
|
|
def test_verify_path(self):
|
|
verify = '/path/to/cacerts.pem'
|
|
req = self.do_request(verify=verify)
|
|
self.assertEqual(verify, req.verify)
|
|
|
|
def test_certs(self):
|
|
cert = ('/path/to/cert.pem', 'path/to/key.pem')
|
|
req = self.do_request(cert=cert)
|
|
self.assertEqual(cert, req.cert)
|
|
self.assertTrue(req.verify)
|
|
|
|
def test_proxies(self):
|
|
proxies = {'http': 'foo.bar:3128',
|
|
'http://host.name': 'foo.bar:4012'}
|
|
|
|
req = self.do_request(proxies=proxies)
|
|
|
|
self.assertEqual(proxies, req.proxies)
|
|
self.assertIsNot(proxies, req.proxies)
|
|
|
|
def test_hostname_port_http(self):
|
|
req = self.do_request(url='http://host.example.com:81/path')
|
|
|
|
self.assertEqual('host.example.com:81', req.netloc)
|
|
self.assertEqual('host.example.com', req.hostname)
|
|
self.assertEqual(81, req.port)
|
|
|
|
def test_hostname_port_https(self):
|
|
req = self.do_request(url='https://host.example.com:8080/path')
|
|
|
|
self.assertEqual('host.example.com:8080', req.netloc)
|
|
self.assertEqual('host.example.com', req.hostname)
|
|
self.assertEqual(8080, req.port)
|
|
|
|
def test_hostname_default_port_http(self):
|
|
req = self.do_request(url='http://host.example.com/path')
|
|
|
|
self.assertEqual('host.example.com', req.netloc)
|
|
self.assertEqual('host.example.com', req.hostname)
|
|
self.assertEqual(80, req.port)
|
|
|
|
def test_hostname_default_port_https(self):
|
|
req = self.do_request(url='https://host.example.com/path')
|
|
|
|
self.assertEqual('host.example.com', req.netloc)
|
|
self.assertEqual('host.example.com', req.hostname)
|
|
self.assertEqual(443, req.port)
|