requests/test_requests.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from __future__ import with_statement

import unittest
import cookielib

import requests


class RequestsTestSuite(unittest.TestCase):
    """Requests test cases."""

    def setUp(self):
        pass


    def tearDown(self):
        """Teardown."""
        pass


    def test_invalid_url(self):
        self.assertRaises(ValueError, requests.get, 'hiwpefhipowhefopw')


    def test_HTTP_200_OK_GET(self):
        r = requests.get('http://httpbin.org/')
        self.assertEqual(r.status_code, 200)


    def test_HTTPS_200_OK_GET(self):
        r = requests.get('https://github.com/')
        self.assertEqual(r.status_code, 200)


    def test_HTTP_200_OK_GET_WITH_PARAMS(self):
        heads = {'User-agent': 'Mozilla/5.0'}

        r = requests.get('http://httpbin.org/user-agent', headers=heads)

        assert heads['User-agent'] in r.content
        self.assertEqual(r.status_code, 200)


    def test_HTTP_200_OK_GET_WITH_MIXED_PARAMS(self):
        heads = {'User-agent': 'Mozilla/5.0'}

        r = requests.get('http://httpbin.org/get?test=true', params={'q': 'test'}, headers=heads)
        self.assertEqual(r.status_code, 200)


    def test_user_agent_transfers(self):
        """Issue XX"""

        heads = {
            'User-agent':
                'Mozilla/5.0 (github.com/kennethreitz/requests)'
        }

        r = requests.get('http://httpbin.org/user-agent', headers=heads);
        self.assertTrue(heads['User-agent'] in r.content)

        heads = {
            'user-agent':
                'Mozilla/5.0 (github.com/kennethreitz/requests)'
        }

        r = requests.get('http://httpbin.org/user-agent', headers=heads);
        self.assertTrue(heads['user-agent'] in r.content)


    def test_HTTP_200_OK_HEAD(self):
        r = requests.head('http://httpbin.org')
        self.assertEqual(r.status_code, 200)


    def test_HTTPS_200_OK_HEAD(self):
        r = requests.head('https://github.com')
        self.assertEqual(r.status_code, 200)


    def test_AUTH_HTTPS_200_OK_GET(self):
        auth = ('requeststest', 'requeststest')
        url = 'https://convore.com/api/account/verify.json'
        r = requests.get(url, auth=auth)

        self.assertEqual(r.status_code, 200)

        r = requests.get(url)
        self.assertEqual(r.status_code, 200)

        # reset auto authentication
        requests.auth_manager.empty()


    def test_POSTBIN_GET_POST_FILES(self):
        bin = requests.post('http://www.postbin.org/')
        self.assertEqual(bin.status_code, 302)

        post_url = bin.headers['location']
        post = requests.post(post_url, data={'some': 'data'})
        self.assertEqual(post.status_code, 201)

        post2 = requests.post(post_url, files={'some': open('test_requests.py')})
        self.assertEqual(post2.status_code, 201)

        post3 = requests.post(post_url, data='[{"some": "json"}]')
        self.assertEqual(post.status_code, 201)


    def test_POSTBIN_GET_POST_FILES_WITH_PARAMS(self):
        bin = requests.post('http://www.postbin.org/')
        self.assertEqual(bin.status_code, 302)

        post_url = bin.headers['location']

        post2 = requests.post(post_url, files={'some': open('test_requests.py')}, data={'some': 'data'})
        self.assertEqual(post2.status_code, 201)


    def test_POSTBIN_GET_POST_FILES_WITH_HEADERS(self):
        bin = requests.post('http://www.postbin.org/')
        self.assertEqual(bin.status_code, 302)

        post_url = bin.headers['location']

        post2 = requests.post(post_url, files={'some': open('test_requests.py')},
        headers = {'User-Agent': 'requests-tests'})

        self.assertEqual(post2.status_code, 201)


    def test_nonzero_evaluation(self):
        r = requests.get('http://google.com/some-404-url')
        self.assertEqual(bool(r), False)

        r = requests.get('http://google.com/')
        self.assertEqual(bool(r), True)


    def test_request_ok_set(self):
        r = requests.get('http://google.com/some-404-url')
        self.assertEqual(r.ok, False)


    def test_status_raising(self):
        r = requests.get('http://google.com/some-404-url')
        self.assertRaises(requests.HTTPError, r.raise_for_status)

        r = requests.get('http://google.com/')
        self.assertFalse(r.error)
        r.raise_for_status()


    def test_cookie_jar(self):
        jar = cookielib.CookieJar()
        self.assertFalse(jar)
        data = {'cn': 'requests_cookie', 'cv': 'awesome'}
        r = requests.post('http://www.html-kit.com/tools/cookietester/', data=data, cookies=jar, allow_redirects=True)
        self.assertTrue(jar)
        cookie_found = False
        for cookie in jar:
            if cookie.name == 'requests_cookie':
                self.assertEquals(cookie.value, 'awesome')
                cookie_found = True
        self.assertTrue(cookie_found)


    def test_decompress_gzip(self):

        r = requests.get('http://api.stackoverflow.com/1.1/users/495995/top-answer-tags')
        r.content.decode('ascii')


    def test_autoauth(self):

        conv_auth = ('requeststest', 'requeststest')
        requests.auth_manager.add_auth('convore.com', conv_auth)

        r = requests.get('https://convore.com/api/account/verify.json')
        self.assertEquals(r.status_code, 200)


    def test_unicode_get(self):

        requests.get('http://google.com', params={'foo': u'føø'})
        requests.get('http://google.com', params={u'føø': u'føø'})
        requests.get('http://google.com', params={'føø': 'føø'})
        requests.get('http://google.com', params={'foo': u'foo'})
        requests.get('http://google.com/ø', params={'foo': u'foo'})


    def test_httpauth_recursion(self):
        conv_auth = ('requeststest', 'bad_password')

        r = requests.get('https://convore.com/api/account/verify.json', auth=conv_auth)
        self.assertEquals(r.status_code, 401)


    def test_settings(self):
        with requests.settings(timeout=0.0000001):
            self.assertRaises(requests.Timeout, requests.get, 'http://google.com')

        with requests.settings(timeout=10):
            requests.get('http://google.com')


    def test_nonurlencoded_post_data(self):
        requests.post('http://google.com', data='foo')



if __name__ == '__main__':
    unittest.main()