Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 01:03:24 +00:00
Code Releases Activity

Updated test for filters. Mock replaced with real Request object

Browse Source
This commit is contained in:
Max Shytikov
2014-04-30 03:48:29 +02:00
parent 998dde61ec
commit 65c2797f94
1 changed files with 20 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
tests/unit/test_filters.py
View File

@@ -1,28 +1,37 @@
import mock
from vcr.filters import _remove_headers, _remove_query_parameters
from vcr.request import Request
def test_remove_headers():
request = mock.Mock(headers=[('hello','goodbye'),('secret','header')])
assert _remove_headers(request, ['secret']).headers == frozenset([('hello','goodbye')])
headers = {'hello': ['goodbye'], 'secret': ['header']}
request = Request('GET', 'http://google.com', '', headers)
_remove_headers(request, ['secret'])
assert request.headers == {'hello': ['goodbye']}
def test_remove_headers_empty():
request = mock.Mock(headers=[('hello','goodbye'),('secret','header')])
assert _remove_headers(request, []).headers == frozenset([('hello','goodbye'),('secret','header')])
headers = {'hello': ['goodbye'], 'secret': ['header']}
request = Request('GET', 'http://google.com', '', headers)
_remove_headers(request, [])
assert request.headers == headers
def test_remove_query_parameters():
request = mock.Mock(url='http://g.com/?q=cowboys&w=1')
assert _remove_query_parameters(request, ['w']).path == '/?q=cowboys'
uri = 'http://g.com/?q=cowboys&w=1'
request = Request('GET', uri, '', {})
_remove_query_parameters(request, ['w'])
assert request.uri == 'http://g.com/?q=cowboys'
def test_remove_all_query_parameters():
request = mock.Mock(url='http://g.com/?q=cowboys&w=1')
assert _remove_query_parameters(request, ['w','q']).path == '/'
uri = 'http://g.com/?q=cowboys&w=1'
request = Request('GET', uri, '', {})
_remove_query_parameters(request, ['w', 'q'])
assert request.uri == 'http://g.com/'
def test_remove_nonexistent_query_parameters():
request = mock.Mock(url='http://g.com/')
assert _remove_query_parameters(request, ['w','q']).path == '/'
uri = 'http://g.com/'
request = Request('GET', uri, '', {})
_remove_query_parameters(request, ['w', 'q'])
assert request.uri == 'http://g.com/'