import datetime import mock import pytest from openapi_core.schema.schemas.exceptions import InvalidSchemaValue from openapi_core.schema.schemas.models import Schema class TestSchemaIteritems(object): @pytest.fixture def schema(self): properties = { 'application/json': mock.sentinel.application_json, 'text/csv': mock.sentinel.text_csv, } return Schema('object', properties=properties) @property def test_valid(self, schema): for name in schema.properties.keys(): assert schema[name] == schema.properties[name] class TestSchemaUnmarshal(object): def test_deprecated(self): schema = Schema('string', deprecated=True) value = 'test' with pytest.warns(DeprecationWarning): result = schema.unmarshal(value) assert result == value def test_string_valid(self): schema = Schema('string') value = 'test' result = schema.unmarshal(value) assert result == value def test_string_none(self): schema = Schema('string') value = None with pytest.raises(InvalidSchemaValue): schema.unmarshal(value) def test_string_default(self): default_value = 'default' schema = Schema('string', default=default_value) value = None with pytest.raises(InvalidSchemaValue): schema.unmarshal(value) def test_string_default_nullable(self): default_value = 'default' schema = Schema('string', default=default_value, nullable=True) value = None result = schema.unmarshal(value) assert result == default_value def test_string_format_date(self): schema = Schema('string', schema_format='date') value = '2018-01-02' result = schema.unmarshal(value) assert result == datetime.date(2018, 1, 2) def test_string_format_custom(self): custom_format = 'custom' schema = Schema('string', schema_format=custom_format) value = 'x' with mock.patch.dict( Schema.FORMAT_CALLABLE_GETTER, {custom_format: lambda x: x + '-custom'}, ): result = schema.unmarshal(value) assert result == 'x-custom' def test_string_format_unknown(self): unknown_format = 'unknown' schema = Schema('string', schema_format=unknown_format) value = 'x' result = schema.unmarshal(value) assert result == 'x' def test_string_format_invalid_value(self): custom_format = 'custom' schema = Schema('string', schema_format=custom_format) value = 'x' with mock.patch.dict( Schema.FORMAT_CALLABLE_GETTER, {custom_format: mock.Mock(side_effect=ValueError())}, ), pytest.raises( InvalidSchemaValue, message='Failed to format value' ): schema.unmarshal(value) def test_integer_valid(self): schema = Schema('integer') value = '123' result = schema.unmarshal(value) assert result == int(value) def test_integer_enum_invalid(self): schema = Schema('integer', enum=[1, 2, 3]) value = '123' with pytest.raises(InvalidSchemaValue): schema.unmarshal(value) def test_integer_enum(self): schema = Schema('integer', enum=[1, 2, 3]) value = '2' result = schema.unmarshal(value) assert result == int(value) def test_integer_default(self): default_value = '123' schema = Schema('integer', default=default_value) value = None with pytest.raises(InvalidSchemaValue): schema.unmarshal(value) def test_integer_default_nullable(self): default_value = '123' schema = Schema('integer', default=default_value, nullable=True) value = None result = schema.unmarshal(value) assert result == default_value def test_integer_invalid(self): schema = Schema('integer') value = 'abc' with pytest.raises(InvalidSchemaValue): schema.unmarshal(value) class TestSchemaValidate(object): @pytest.mark.parametrize('schema_type', [ 'boolean', 'array', 'integer', 'number', 'string', ]) def test_null(self, schema_type): schema = Schema(schema_type) value = None with pytest.raises(InvalidSchemaValue): schema.validate(value) @pytest.mark.parametrize('schema_type', [ 'boolean', 'array', 'integer', 'number', 'string', ]) def test_nullable(self, schema_type): schema = Schema(schema_type, nullable=True) value = None result = schema.validate(value) assert result is None @pytest.mark.parametrize('value', [False, True]) def test_boolean(self, value): schema = Schema('boolean') result = schema.validate(value) assert result == value @pytest.mark.parametrize('value', [1, 3.14, 'true', [True, False]]) def test_boolean_invalid(self, value): schema = Schema('boolean') with pytest.raises(InvalidSchemaValue): schema.validate(value) @pytest.mark.parametrize('value', [[1, 2], (3, 4)]) def test_array(self, value): schema = Schema('array') result = schema.validate(value) assert result == value @pytest.mark.parametrize('value', [False, 1, 3.14, 'true']) def test_array_invalid(self, value): schema = Schema('array') with pytest.raises(InvalidSchemaValue): schema.validate(value) @pytest.mark.parametrize('value', [1, 3]) def test_integer(self, value): schema = Schema('integer') result = schema.validate(value) assert result == value @pytest.mark.parametrize('value', [False, 3.14, 'true', [1, 2]]) def test_integer_invalid(self, value): schema = Schema('integer') with pytest.raises(InvalidSchemaValue): schema.validate(value) @pytest.mark.parametrize('value', [1, 3.14]) def test_number(self, value): schema = Schema('number') result = schema.validate(value) assert result == value @pytest.mark.parametrize('value', [False, 'true', [1, 3]]) def test_number_invalid(self, value): schema = Schema('number') with pytest.raises(InvalidSchemaValue): schema.validate(value) @pytest.mark.parametrize('value', ['true', b'true']) def test_string(self, value): schema = Schema('string') result = schema.validate(value) assert result == value @pytest.mark.parametrize('value', [False, 1, 3.14, [1, 3]]) def test_string_invalid(self, value): schema = Schema('string') with pytest.raises(InvalidSchemaValue): schema.validate(value)