def test_get_request_body_args(self):
"""
Tests getting the request body args
from a flask request object.
"""
app = Flask('myapp')
body = dict(x=1)
with app.test_request_context('/', data=json.dumps(body), content_type='application/json'):
q, b = get_request_query_body_args(request)
self.assertDictEqual(b, body)
with app.test_request_context('/', data=body): # Form encoded
q, b = get_request_query_body_args(request)
self.assertDictEqual(b, dict(x=['1']))
def test_get_request_body_args(self):
"""
Tests getting the request body args
from a flask request object.
"""
app = Flask('myapp')
body = dict(x=1)
with app.test_request_context('/',
data=json.dumps(body),
content_type='application/json'):
q, b, headers = get_request_query_body_args(request)
self.assertDictEqual(b, body)
with app.test_request_context('/', data=body): # Form encoded
q, b, headers = get_request_query_body_args(request)
self.assertDictEqual(b, dict(x=['1']))
def test_headers_copyable(self):
"""
Tests that the headers returned from get_request_query_body_args
appropriately returns the headers as a dictionary that can be copied
"""
app = Flask('myapp')
with app.test_request_context('/'):
q, b, headers = get_request_query_body_args(request)
headers2 = headers.copy()
self.assertDictEqual(headers, headers2)
def test_get_request_body_args_nested(self):
"""
Tests getting nested body args which seems to
be handled slightly differnetly.
"""
app = Flask('myapp')
body = dict(x=1, y=dict(x=1))
with app.test_request_context('/', data=json.dumps(body), content_type='application/json'):
q, b = get_request_query_body_args(request)
self.assertDictEqual(b, body)
def test_headers_copyable(self):
"""
Tests that the headers returned from get_request_query_body_args
appropriately returns the headers as a dictionary that can be copied
"""
app = Flask('myapp')
with app.test_request_context('/'):
q, b, headers = get_request_query_body_args(request)
headers2 = headers.copy()
self.assertDictEqual(headers, headers2)
def test_get_request_body_args_nested(self):
"""
Tests getting nested body args which seems to
be handled slightly differnetly.
"""
app = Flask('myapp')
body = dict(x=1, y=dict(x=1))
with app.test_request_context('/',
data=json.dumps(body),
content_type='application/json'):
q, b, headers = get_request_query_body_args(request)
self.assertDictEqual(b, body)
def test_get_request_query_body_args(self):
"""
Tests the private get_request_query_body_args
method.
"""
query_args = dict(x=1)
form = dict(x=2)
mck = mock.Mock(args=query_args, form=form, get_json=mock.Mock(return_value=None))
q, b = get_request_query_body_args(mck)
self.assertDictEqual(query_args, q)
self.assertDictEqual(form, b)
mck = mock.MagicMock(args=query_args, form=None, get_json=mock.Mock(return_value=form))
q, b = get_request_query_body_args(mck)
self.assertDictEqual(query_args, q)
self.assertDictEqual(form, b)
mck = mock.MagicMock(args=query_args, form=None, get_json=mock.Mock(return_value=None), data=json.dumps(form))
q, b = get_request_query_body_args(mck)
self.assertDictEqual(query_args, q)
self.assertDictEqual({}, b)
def test_get_request_query_body_args(self):
"""
Tests the private get_request_query_body_args
method.
"""
query_args = dict(x=1)
form = dict(x=2)
mck = mock.Mock(args=query_args, form=form,
get_json=mock.Mock(return_value=None), headers={})
q, b, h = get_request_query_body_args(mck)
self.assertDictEqual(query_args, q)
self.assertDictEqual(form, b)
mck = mock.MagicMock(args=query_args, form=None,
get_json=mock.Mock(return_value=form), headers={})
q, b, h = get_request_query_body_args(mck)
self.assertDictEqual(query_args, q)
self.assertDictEqual(form, b)
mck = mock.MagicMock(args=query_args, form=None,
get_json=mock.Mock(return_value=None), data=json.dumps(form), headers={})
q, b, h = get_request_query_body_args(mck)
self.assertDictEqual(query_args, q)
self.assertDictEqual({}, b)
