77 lines
2.0 KiB
Python
77 lines
2.0 KiB
Python
|
"""Test module for requests_toolbelt.utils.formdata."""
|
||
|
|
try:
|
||
|
|
from urllib.parse import parse_qs
|
||
|
|
except ImportError:
|
||
|
|
from urlparse import parse_qs
|
||
|
|
|
||
|
|
from requests_toolbelt.utils.formdata import urlencode
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
dict_query = {
|
||
|
|
'first_nested': {
|
||
|
|
'second_nested': {
|
||
|
|
'third_nested': {
|
||
|
|
'fourth0': 'fourth_value0',
|
||
|
|
'fourth1': 'fourth_value1',
|
||
|
|
},
|
||
|
|
'third0': 'third_value0',
|
||
|
|
},
|
||
|
|
'second0': 'second_value0',
|
||
|
|
},
|
||
|
|
'outter': 'outter_value',
|
||
|
|
}
|
||
|
|
|
||
|
|
list_query = [
|
||
|
|
('first_nested', [
|
||
|
|
('second_nested', [
|
||
|
|
('third_nested', [
|
||
|
|
('fourth0', 'fourth_value0'),
|
||
|
|
('fourth1', 'fourth_value1'),
|
||
|
|
]),
|
||
|
|
('third0', 'third_value0'),
|
||
|
|
]),
|
||
|
|
('second0', 'second_value0'),
|
||
|
|
]),
|
||
|
|
('outter', 'outter_value'),
|
||
|
|
]
|
||
|
|
|
||
|
|
mixed_dict_query = {
|
||
|
|
'first_nested': {
|
||
|
|
'second_nested': [
|
||
|
|
('third_nested', {
|
||
|
|
'fourth0': 'fourth_value0',
|
||
|
|
'fourth1': 'fourth_value1',
|
||
|
|
}),
|
||
|
|
('third0', 'third_value0'),
|
||
|
|
],
|
||
|
|
'second0': 'second_value0',
|
||
|
|
},
|
||
|
|
'outter': 'outter_value',
|
||
|
|
}
|
||
|
|
|
||
|
|
expected_parsed_query = {
|
||
|
|
'first_nested[second0]': ['second_value0'],
|
||
|
|
'first_nested[second_nested][third0]': ['third_value0'],
|
||
|
|
'first_nested[second_nested][third_nested][fourth0]': ['fourth_value0'],
|
||
|
|
'first_nested[second_nested][third_nested][fourth1]': ['fourth_value1'],
|
||
|
|
'outter': ['outter_value'],
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("query", [dict_query, list_query, mixed_dict_query])
|
||
|
|
def test_urlencode_flattens_nested_structures(query):
|
||
|
|
"""Show that when parsed, the structure is conveniently flat."""
|
||
|
|
parsed = parse_qs(urlencode(query))
|
||
|
|
|
||
|
|
assert parsed == expected_parsed_query
|
||
|
|
|
||
|
|
|
||
|
|
def test_urlencode_catches_invalid_input():
|
||
|
|
"""Show that queries are loosely validated."""
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
urlencode(['fo'])
|
||
|
|
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
urlencode([('foo', 'bar', 'bogus')])
|