def test_parse_required(self, app): parser = RequestParser() parser.add_argument('foo', required=True, location='values') expected = { 'foo': 'Missing required parameter in the post body or the query string' } with pytest.raises(BadRequest) as cm: parser.parse_args() assert cm.value.data['message'] == 'Input payload validation failed' assert cm.value.data['errors'] == expected parser = RequestParser() parser.add_argument('bar', required=True, location=['values', 'cookies']) expected = { 'bar': ("Missing required parameter in the post body or the query " "string or the request's cookies") } with pytest.raises(BadRequest) as cm: parser.parse_args() assert cm.value.data['message'] == 'Input payload validation failed' assert cm.value.data['errors'] == expected
def test_viewargs(self, mocker): req = Request.from_values() req.view_args = {'foo': 'bar'} parser = RequestParser() parser.add_argument('foo', location=['view_args']) args = parser.parse_args(req) assert args['foo'] == 'bar' req = mocker.Mock() req.values = () req.json = None req.view_args = {'foo': 'bar'} parser = RequestParser() parser.add_argument('foo', store_missing=True) args = parser.parse_args(req) assert args['foo'] is None
def test_parse_unicode(self, app): req = Request.from_values('/bubble?foo=barß') parser = RequestParser() parser.add_argument('foo') args = parser.parse_args(req) assert args['foo'] == 'barß'
def test_parse_unicode_app(self, app): parser = RequestParser() parser.add_argument('foo') with app.test_request_context('/bubble?foo=barß'): args = parser.parse_args() assert args['foo'] == 'barß'
def test_trim_request_parser(self, app): req = Request.from_values('/bubble?foo= 1 &bar=bees&n=22') parser = RequestParser(trim=False) parser.add_argument('foo') args = parser.parse_args(req) assert args['foo'] == ' 1 ' parser = RequestParser(trim=True) parser.add_argument('foo') args = parser.parse_args(req) assert args['foo'] == '1' parser = RequestParser(trim=True) parser.add_argument('foo', type=int) args = parser.parse_args(req) assert args['foo'] == 1
def test_passing_arguments_object(self, app): req = Request.from_values('/bubble?foo=bar') parser = RequestParser() parser.add_argument(Argument('foo')) args = parser.parse_args(req) assert args['foo'] == 'bar'
def test_type_callable(self, app): req = Request.from_values('/bubble?foo=1') parser = RequestParser() parser.add_argument('foo', type=lambda x: x, required=False), args = parser.parse_args(req) assert args['foo'] == '1'
def test_parse_ignore(self, app): req = Request.from_values('/bubble?foo=bar') parser = RequestParser() parser.add_argument('foo', type=int, ignore=True, store_missing=True), args = parser.parse_args(req) assert args['foo'] is None
def test_parse_choices_insensitive(self, app): req = Request.from_values('/bubble?foo=BAT') parser = RequestParser() parser.add_argument('foo', choices=['bat'], case_sensitive=False), args = parser.parse_args(req) assert 'bat' == args.get('foo') # both choices and args are case_insensitive req = Request.from_values('/bubble?foo=bat') parser = RequestParser() parser.add_argument('foo', choices=['BAT'], case_sensitive=False), args = parser.parse_args(req) assert 'bat' == args.get('foo')
def test_parse_choices_sensitive(self, app): req = Request.from_values('/bubble?foo=BAT') parser = RequestParser() parser.add_argument('foo', choices=['bat'], case_sensitive=True), with pytest.raises(BadRequest): parser.parse_args(req)
def test_parse_choices(self, app): req = Request.from_values('/bubble?foo=bar') parser = RequestParser() parser.add_argument('foo', choices=['bat']), with pytest.raises(BadRequest): parser.parse_args(req)
def test_not_json_location_and_content_type_json(self, app): parser = RequestParser() parser.add_argument('foo', location='args') with app.test_request_context('/bubble', method='get', content_type='application/json'): parser.parse_args() # Should not raise a 400: BadRequest
def test_parse_store_missing(self, app): req = Request.from_values('/bubble') parser = RequestParser() parser.add_argument('foo', store_missing=False) args = parser.parse_args(req) assert 'foo' not in args
def test_parse_lte_gte_mock(self, mocker): mock_type = mocker.Mock() parser = RequestParser() parser.add_argument('foo', type=mock_type, operators=['<=']) parser.parse_args() mock_type.assert_called_with('bar', 'foo', '<=')
def test_parse_gte_lte_eq(self): parser = RequestParser() parser.add_argument('foo', operators=['>=', '<=', '='], action='append'), args = parser.parse_args() assert args['foo'] == ['bar', 'bat', 'foo']
def test_parse_choices_correct(self, app): req = Request.from_values('/bubble?foo=bat') parser = RequestParser() parser.add_argument('foo', choices=['bat']), args = parser.parse_args(req) assert args['foo'] == 'bat'
def test_location_form(self): parser = RequestParser() parser.add_argument('in_form', type=int, location='form') assert parser.__schema__ == [{ 'name': 'in_form', 'type': 'integer', 'in': 'formData', }]
def test_location_files(self): parser = RequestParser() parser.add_argument('in_files', type=FileStorage, location='files') assert parser.__schema__ == [{ 'name': 'in_files', 'type': 'file', 'in': 'formData', }]
def test_unknown_type(self): parser = RequestParser() parser.add_argument('unknown', type=lambda v: v) assert parser.__schema__ == [{ 'name': 'unknown', 'type': 'string', 'in': 'query', }]
def test_location_json(self): parser = RequestParser() parser.add_argument('in_json', type=str, location='json') assert parser.__schema__ == [{ 'name': 'in_json', 'type': 'string', 'in': 'body', }]
def test_default_as_false(self): parser = RequestParser() parser.add_argument('bool', type=inputs.boolean, default=False) assert parser.__schema__ == [{ 'name': 'bool', 'type': 'boolean', 'in': 'query', 'default': False, }]
def test_default(self): parser = RequestParser() parser.add_argument('int', type=int, default=5) assert parser.__schema__ == [{ 'name': 'int', 'type': 'integer', 'in': 'query', 'default': 5, }]
def test_required(self): parser = RequestParser() parser.add_argument('int', type=int, required=True) assert parser.__schema__ == [{ 'name': 'int', 'type': 'integer', 'in': 'query', 'required': True, }]
def test_none_argument(self, app): parser = RequestParser() parser.add_argument('foo', location='json') with app.test_request_context('/bubble', method='post', data=json.dumps({'foo': None}), content_type='application/json'): args = parser.parse_args() assert args['foo'] is None
def test_files_and_body_location(self): parser = RequestParser() parser.add_argument('default', type=int) parser.add_argument('in_files', type=FileStorage, location='files') parser.add_argument('in_json', type=str, location='json') with pytest.raises(SpecsError) as cm: parser.__schema__ assert cm.value.msg == "Can't use formData and body at the same time"
def test_request_parser_remove_argument(self): req = Request.from_values('/bubble?foo=baz') parser = RequestParser() parser.add_argument('foo', type=int) parser_copy = parser.copy() parser_copy.remove_argument('foo') args = parser_copy.parse_args(req) assert args == {}
def test_type_decimal(self, app): parser = RequestParser() parser.add_argument('foo', type=decimal.Decimal, location='json') with app.test_request_context('/bubble', method='post', data=json.dumps({'foo': '1.0025'}), content_type='application/json'): args = parser.parse_args() assert args['foo'] == decimal.Decimal('1.0025')
def test_parse_append_ignore(self, app): parser = RequestParser() parser.add_argument('foo', ignore=True, type=int, action='append', store_missing=True), args = parser.parse_args() assert args['foo'] is None
def test_choices(self): parser = RequestParser() parser.add_argument('string', type=str, choices=['a', 'b']) assert parser.__schema__ == [{ 'name': 'string', 'type': 'string', 'in': 'query', 'enum': ['a', 'b'], 'collectionFormat': 'multi', }]
def test_parse_default_append(self): parser = RequestParser() parser.add_argument('foo', default='bar', action='append', store_missing=True) args = parser.parse_args() assert args['foo'] == 'bar'