Prodream/ormar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add other valid field types, better parse model fields to pydantic model with optional values

Browse Source
This commit is contained in:
collerek
2020-08-03 13:06:52 +02:00
parent f27e69f87f
commit 0bd964bdc4
6 changed files with 120 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
tests/test_columns.py
View File

@ -1,3 +1,6 @@
import datetime
import pydantic
import pytest
import sqlalchemy
@ -12,20 +15,34 @@ class ExampleModel(Model):
__tablename__ = "example"
__metadata__ = metadata
test = fields.Integer(primary_key=True)
test2 = fields.String(length=250)
test_string = fields.String(length=250)
test_text = fields.Text()
test_bool = fields.Boolean(nullable=False)
test_float = fields.Float()
test_datetime = fields.DateTime(default=datetime.datetime.now)
test_date = fields.Date(default=datetime.date.today)
test_time = fields.Time(default=datetime.time)
test_json = fields.JSON(default={})
test_bigint = fields.BigInteger(default=0)
test_decimal = fields.Decimal(length=10, precision=2)
class ExampleModel2(Model):
__tablename__ = "example2"
__metadata__ = metadata
test = fields.Integer(name='test12', primary_key=True)
test2 = fields.String('test22', length=250)
test_string = fields.String('test_string2', length=250)
def test_not_nullable_field_is_required():
with pytest.raises(pydantic.error_wrappers.ValidationError):
ExampleModel(test=1, test_string='test')
def test_model_attribute_access():
example = ExampleModel(test=1, test2='test')
example = ExampleModel(test=1, test_string='test', test_bool=True)
assert example.test == 1
assert example.test2 == 'test'
assert example.test_string == 'test'
example.test = 12
assert example.test == 12
@ -35,7 +52,7 @@ def test_model_attribute_access():
def test_primary_key_access_and_setting():
example = ExampleModel(pk=1, test2='test')
example = ExampleModel(pk=1, test_string='test', test_bool=True)
assert example.pk == 1
example.pk = 2
@ -49,4 +66,4 @@ def test_wrong_model_definition():
__tablename__ = "example3"
__metadata__ = metadata
test = fields.Integer(name='test12', primary_key=True)
test2 = fields.String('test22', name='test22', length=250)
test_string = fields.String('test_string2', name='test_string2', length=250)