228 lines
8.3 KiB
Python
228 lines
8.3 KiB
Python
# #!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# <HTTPretty - HTTP client mock for Python>
|
|
# Copyright (C) <2011-2012> Gabriel Falcão <gabriel@nacaolivre.org>
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person
|
|
# obtaining a copy of this software and associated documentation
|
|
# files (the "Software"), to deal in the Software without
|
|
# restriction, including without limitation the rights to use,
|
|
# copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
# copies of the Software, and to permit persons to whom the
|
|
# Software is furnished to do so, subject to the following
|
|
# conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be
|
|
# included in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
|
# OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
|
# OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
import httplib2
|
|
from sure import that, within, microseconds
|
|
from httpretty import HTTPretty, httprettified
|
|
|
|
|
|
@httprettified
|
|
@within(two=microseconds)
|
|
def test_httpretty_should_mock_a_simple_get_with_httplib2_read(now):
|
|
u"HTTPretty should mock a simple GET with httplib2.context.http"
|
|
|
|
HTTPretty.register_uri(HTTPretty.GET, "http://yipit.com/",
|
|
body="Find the best daily deals")
|
|
|
|
_, got = httplib2.Http().request('http://yipit.com', 'GET')
|
|
assert that(got).equals('Find the best daily deals')
|
|
assert that(HTTPretty.last_request.method).equals('GET')
|
|
assert that(HTTPretty.last_request.path).equals('/')
|
|
|
|
|
|
@httprettified
|
|
@within(two=microseconds)
|
|
def test_httpretty_should_mock_headers_httplib2(now):
|
|
u"HTTPretty should mock basic headers with httplib2"
|
|
|
|
HTTPretty.register_uri(HTTPretty.GET, "http://github.com/",
|
|
body="this is supposed to be the response",
|
|
status=201)
|
|
|
|
headers, _ = httplib2.Http().request('http://github.com', 'GET')
|
|
assert that(headers['status']).equals('201')
|
|
assert that(dict(headers)).equals({
|
|
'content-type': 'text/plain; charset=utf-8',
|
|
'connection': 'close',
|
|
'content-length': '35',
|
|
'status': '201',
|
|
'server': 'Python/HTTPretty',
|
|
'date': now.strftime('%a, %d %b %Y %H:%M:%S GMT'),
|
|
})
|
|
|
|
|
|
@httprettified
|
|
@within(two=microseconds)
|
|
def test_httpretty_should_allow_adding_and_overwritting_httplib2(now):
|
|
u"HTTPretty should allow adding and overwritting headers with httplib2"
|
|
|
|
HTTPretty.register_uri(HTTPretty.GET, "http://github.com/foo",
|
|
body="this is supposed to be the response",
|
|
adding_headers={
|
|
'Server': 'Apache',
|
|
'Content-Length': '27',
|
|
'Content-Type': 'application/json',
|
|
})
|
|
|
|
headers, _ = httplib2.Http().request('http://github.com/foo', 'GET')
|
|
|
|
assert that(dict(headers)).equals({
|
|
'content-type': 'application/json',
|
|
'content-location': 'http://github.com/foo',
|
|
'connection': 'close',
|
|
'content-length': '27',
|
|
'status': '200',
|
|
'server': 'Apache',
|
|
'date': now.strftime('%a, %d %b %Y %H:%M:%S GMT'),
|
|
})
|
|
|
|
|
|
@httprettified
|
|
@within(two=microseconds)
|
|
def test_httpretty_should_allow_forcing_headers_httplib2(now):
|
|
u"HTTPretty should allow forcing headers with httplib2"
|
|
|
|
HTTPretty.register_uri(HTTPretty.GET, "http://github.com/foo",
|
|
body="this is supposed to be the response",
|
|
forcing_headers={
|
|
'Content-Type': 'application/xml',
|
|
})
|
|
|
|
headers, _ = httplib2.Http().request('http://github.com/foo', 'GET')
|
|
|
|
assert that(dict(headers)).equals({
|
|
'content-location': 'http://github.com/foo', # httplib2 FORCES
|
|
# content-location
|
|
# even if the
|
|
# server does not
|
|
# provide it
|
|
'content-type': 'application/xml',
|
|
'status': '200', # httplib2 also ALWAYS put status on headers
|
|
})
|
|
|
|
|
|
@httprettified
|
|
@within(two=microseconds)
|
|
def test_httpretty_should_allow_adding_and_overwritting_by_kwargs_u2(now):
|
|
u"HTTPretty should allow adding and overwritting headers by keyword args " \
|
|
"with httplib2"
|
|
|
|
HTTPretty.register_uri(HTTPretty.GET, "http://github.com/foo",
|
|
body="this is supposed to be the response",
|
|
server='Apache',
|
|
content_length='27',
|
|
content_type='application/json')
|
|
|
|
headers, _ = httplib2.Http().request('http://github.com/foo', 'GET')
|
|
|
|
assert that(dict(headers)).equals({
|
|
'content-type': 'application/json',
|
|
'content-location': 'http://github.com/foo', # httplib2 FORCES
|
|
# content-location
|
|
# even if the
|
|
# server does not
|
|
# provide it
|
|
'connection': 'close',
|
|
'content-length': '27',
|
|
'status': '200',
|
|
'server': 'Apache',
|
|
'date': now.strftime('%a, %d %b %Y %H:%M:%S GMT'),
|
|
})
|
|
|
|
|
|
@httprettified
|
|
@within(two=microseconds)
|
|
def test_rotating_responses_with_httplib2(now):
|
|
u"HTTPretty should support rotating responses with httplib2"
|
|
|
|
HTTPretty.register_uri(
|
|
HTTPretty.GET, "https://api.yahoo.com/test",
|
|
responses=[
|
|
HTTPretty.Response(body="first response", status=201),
|
|
HTTPretty.Response(body='second and last response', status=202),
|
|
])
|
|
|
|
headers1, body1 = httplib2.Http().request(
|
|
'https://api.yahoo.com/test', 'GET')
|
|
|
|
assert that(headers1['status']).equals('201')
|
|
assert that(body1).equals('first response')
|
|
|
|
headers2, body2 = httplib2.Http().request(
|
|
'https://api.yahoo.com/test', 'GET')
|
|
|
|
assert that(headers2['status']).equals('202')
|
|
assert that(body2).equals('second and last response')
|
|
|
|
headers3, body3 = httplib2.Http().request(
|
|
'https://api.yahoo.com/test', 'GET')
|
|
|
|
assert that(headers3['status']).equals('202')
|
|
assert that(body3).equals('second and last response')
|
|
|
|
|
|
@httprettified
|
|
@within(two=microseconds)
|
|
def test_can_inspect_last_request(now):
|
|
u"HTTPretty.last_request is a mimetools.Message request from last match"
|
|
|
|
HTTPretty.register_uri(HTTPretty.POST, "http://api.github.com/",
|
|
body='{"repositories": ["HTTPretty", "lettuce"]}')
|
|
|
|
headers, body = httplib2.Http().request(
|
|
'http://api.github.com', 'POST',
|
|
body='{"username": "gabrielfalcao"}',
|
|
headers={
|
|
'content-type': 'text/json',
|
|
},
|
|
)
|
|
|
|
assert that(HTTPretty.last_request.method).equals('POST')
|
|
assert that(HTTPretty.last_request.body).equals(
|
|
'{"username": "gabrielfalcao"}',
|
|
)
|
|
assert that(HTTPretty.last_request.headers['content-type']).equals(
|
|
'text/json',
|
|
)
|
|
assert that(body).equals('{"repositories": ["HTTPretty", "lettuce"]}')
|
|
|
|
|
|
@httprettified
|
|
@within(two=microseconds)
|
|
def test_can_inspect_last_request_with_ssl(now):
|
|
u"HTTPretty.last_request is recorded even when mocking 'https' (SSL)"
|
|
|
|
HTTPretty.register_uri(HTTPretty.POST, "https://secure.github.com/",
|
|
body='{"repositories": ["HTTPretty", "lettuce"]}')
|
|
|
|
headers, body = httplib2.Http().request(
|
|
'https://secure.github.com', 'POST',
|
|
body='{"username": "gabrielfalcao"}',
|
|
headers={
|
|
'content-type': 'text/json',
|
|
},
|
|
)
|
|
|
|
assert that(HTTPretty.last_request.method).equals('POST')
|
|
assert that(HTTPretty.last_request.body).equals(
|
|
'{"username": "gabrielfalcao"}',
|
|
)
|
|
assert that(HTTPretty.last_request.headers['content-type']).equals(
|
|
'text/json',
|
|
)
|
|
assert that(body).equals('{"repositories": ["HTTPretty", "lettuce"]}')
|