split tests into packages
This commit is contained in:
collerek
0
tests/test_inheritance/__init__.py
Normal file
0
tests/test_inheritance/__init__.py
Normal file
506
tests/test_inheritance/test_inheritance_concrete.py
Normal file
506
tests/test_inheritance/test_inheritance_concrete.py
Normal file
@ -0,0 +1,506 @@
|
||||
# type: ignore
|
||||
import datetime
|
||||
from typing import List, Optional
|
||||
|
||||
import databases
|
||||
import pytest
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy import create_engine
|
||||
|
||||
import ormar
|
||||
from ormar import ModelDefinitionError, property_field
|
||||
from ormar.exceptions import ModelError
|
||||
from tests.settings import DATABASE_URL
|
||||
|
||||
metadata = sa.MetaData()
|
||||
db = databases.Database(DATABASE_URL)
|
||||
engine = create_engine(DATABASE_URL)
|
||||
|
||||
|
||||
class AuditModel(ormar.Model):
|
||||
class Meta:
|
||||
abstract = True
|
||||
|
||||
created_by: str = ormar.String(max_length=100)
|
||||
updated_by: str = ormar.String(max_length=100, default="Sam")
|
||||
|
||||
@property_field
|
||||
def audit(self): # pragma: no cover
|
||||
return f"{self.created_by} {self.updated_by}"
|
||||
|
||||
|
||||
class DateFieldsModelNoSubclass(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "test_date_models"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
date_id: int = ormar.Integer(primary_key=True)
|
||||
created_date: datetime.datetime = ormar.DateTime(default=datetime.datetime.now)
|
||||
updated_date: datetime.datetime = ormar.DateTime(default=datetime.datetime.now)
|
||||
|
||||
|
||||
class DateFieldsModel(ormar.Model):
|
||||
class Meta:
|
||||
abstract = True
|
||||
metadata = metadata
|
||||
database = db
|
||||
constraints = [ormar.UniqueColumns("creation_date", "modification_date")]
|
||||
|
||||
created_date: datetime.datetime = ormar.DateTime(
|
||||
default=datetime.datetime.now, name="creation_date"
|
||||
)
|
||||
updated_date: datetime.datetime = ormar.DateTime(
|
||||
default=datetime.datetime.now, name="modification_date"
|
||||
)
|
||||
|
||||
|
||||
class Category(DateFieldsModel, AuditModel):
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "categories"
|
||||
constraints = [ormar.UniqueColumns("name", "code")]
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=50, unique=True, index=True)
|
||||
code: int = ormar.Integer()
|
||||
|
||||
@property_field
|
||||
def code_name(self):
|
||||
return f"{self.code}:{self.name}"
|
||||
|
||||
@property_field
|
||||
def audit(self):
|
||||
return f"{self.created_by} {self.updated_by}"
|
||||
|
||||
|
||||
class Subject(DateFieldsModel):
|
||||
class Meta(ormar.ModelMeta):
|
||||
pass
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=50, unique=True, index=True)
|
||||
category: Optional[Category] = ormar.ForeignKey(Category)
|
||||
|
||||
|
||||
class Person(ormar.Model):
|
||||
class Meta:
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=100)
|
||||
|
||||
|
||||
class Car(ormar.Model):
|
||||
class Meta:
|
||||
abstract = True
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=50)
|
||||
owner: Person = ormar.ForeignKey(Person)
|
||||
co_owner: Person = ormar.ForeignKey(Person, related_name="coowned")
|
||||
created_date: datetime.datetime = ormar.DateTime(default=datetime.datetime.now)
|
||||
|
||||
|
||||
class Truck(Car):
|
||||
class Meta:
|
||||
pass
|
||||
|
||||
max_capacity: int = ormar.Integer()
|
||||
|
||||
|
||||
class Bus(Car):
|
||||
class Meta:
|
||||
tablename = "buses"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
owner: Person = ormar.ForeignKey(Person, related_name="buses")
|
||||
max_persons: int = ormar.Integer()
|
||||
|
||||
|
||||
# class PersonsCar(ormar.Model):
|
||||
# class Meta:
|
||||
# tablename = "cars_x_persons"
|
||||
# metadata = metadata
|
||||
# database = db
|
||||
|
||||
|
||||
class Car2(ormar.Model):
|
||||
class Meta:
|
||||
abstract = True
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=50)
|
||||
owner: Person = ormar.ForeignKey(Person, related_name="owned")
|
||||
co_owners: List[Person] = ormar.ManyToMany(
|
||||
Person,
|
||||
# through=PersonsCar,
|
||||
related_name="coowned",
|
||||
)
|
||||
created_date: datetime.datetime = ormar.DateTime(default=datetime.datetime.now)
|
||||
|
||||
|
||||
class Truck2(Car2):
|
||||
class Meta:
|
||||
tablename = "trucks2"
|
||||
|
||||
max_capacity: int = ormar.Integer()
|
||||
|
||||
|
||||
class Bus2(Car2):
|
||||
class Meta:
|
||||
tablename = "buses2"
|
||||
|
||||
max_persons: int = ormar.Integer()
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True, scope="module")
|
||||
def create_test_database():
|
||||
metadata.create_all(engine)
|
||||
yield
|
||||
metadata.drop_all(engine)
|
||||
|
||||
|
||||
def test_init_of_abstract_model():
|
||||
with pytest.raises(ModelError):
|
||||
DateFieldsModel()
|
||||
|
||||
|
||||
def test_duplicated_related_name_on_different_model():
|
||||
with pytest.raises(ModelDefinitionError):
|
||||
|
||||
class Bus3(Car2): # pragma: no cover
|
||||
class Meta:
|
||||
tablename = "buses3"
|
||||
|
||||
owner: Person = ormar.ForeignKey(Person, related_name="buses")
|
||||
max_persons: int = ormar.Integer()
|
||||
|
||||
|
||||
def test_field_redefining_in_concrete_models():
|
||||
class RedefinedField(DateFieldsModel):
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "redefines"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
created_date: str = ormar.String(max_length=200, name="creation_date")
|
||||
|
||||
changed_field = RedefinedField.Meta.model_fields["created_date"]
|
||||
assert changed_field.ormar_default is None
|
||||
assert changed_field.get_alias() == "creation_date"
|
||||
assert any(x.name == "creation_date" for x in RedefinedField.Meta.table.columns)
|
||||
assert isinstance(
|
||||
RedefinedField.Meta.table.columns["creation_date"].type, sa.sql.sqltypes.String,
|
||||
)
|
||||
|
||||
|
||||
def test_model_subclassing_that_redefines_constraints_column_names():
|
||||
with pytest.raises(ModelDefinitionError):
|
||||
|
||||
class WrongField2(DateFieldsModel): # pragma: no cover
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "wrongs"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
created_date: str = ormar.String(max_length=200)
|
||||
|
||||
|
||||
def test_model_subclassing_non_abstract_raises_error():
|
||||
with pytest.raises(ModelDefinitionError):
|
||||
|
||||
class WrongField2(DateFieldsModelNoSubclass): # pragma: no cover
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "wrongs"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
|
||||
|
||||
def test_params_are_inherited():
|
||||
assert Category.Meta.metadata == metadata
|
||||
assert Category.Meta.database == db
|
||||
assert len(Category.Meta.constraints) == 2
|
||||
assert len(Category.Meta.property_fields) == 2
|
||||
|
||||
|
||||
def round_date_to_seconds(
|
||||
date: datetime.datetime,
|
||||
) -> datetime.datetime: # pragma: no cover
|
||||
if date.microsecond >= 500000:
|
||||
date = date + datetime.timedelta(seconds=1)
|
||||
return date.replace(microsecond=0)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_fields_inherited_from_mixin():
|
||||
async with db:
|
||||
async with db.transaction(force_rollback=True):
|
||||
cat = await Category(
|
||||
name="Foo", code=123, created_by="Sam", updated_by="Max"
|
||||
).save()
|
||||
sub = await Subject(name="Bar", category=cat).save()
|
||||
mixin_columns = ["created_date", "updated_date"]
|
||||
mixin_db_columns = ["creation_date", "modification_date"]
|
||||
mixin2_columns = ["created_by", "updated_by"]
|
||||
assert all(field in Category.Meta.model_fields for field in mixin_columns)
|
||||
assert cat.created_date is not None
|
||||
assert cat.updated_date is not None
|
||||
assert all(field in Subject.Meta.model_fields for field in mixin_columns)
|
||||
assert sub.created_date is not None
|
||||
assert sub.updated_date is not None
|
||||
|
||||
assert all(field in Category.Meta.model_fields for field in mixin2_columns)
|
||||
assert all(
|
||||
field not in Subject.Meta.model_fields for field in mixin2_columns
|
||||
)
|
||||
|
||||
inspector = sa.inspect(engine)
|
||||
assert "categories" in inspector.get_table_names()
|
||||
table_columns = [x.get("name") for x in inspector.get_columns("categories")]
|
||||
assert all(
|
||||
col in table_columns for col in mixin_db_columns
|
||||
) # + mixin2_columns)
|
||||
|
||||
assert "subjects" in inspector.get_table_names()
|
||||
table_columns = [x.get("name") for x in inspector.get_columns("subjects")]
|
||||
assert all(col in table_columns for col in mixin_db_columns)
|
||||
|
||||
sub2 = (
|
||||
await Subject.objects.select_related("category")
|
||||
.order_by("-created_date")
|
||||
.exclude_fields("updated_date")
|
||||
.get()
|
||||
)
|
||||
assert round_date_to_seconds(sub2.created_date) == round_date_to_seconds(
|
||||
sub.created_date
|
||||
)
|
||||
assert sub2.category.updated_date is not None
|
||||
assert round_date_to_seconds(
|
||||
sub2.category.created_date
|
||||
) == round_date_to_seconds(cat.created_date)
|
||||
assert sub2.updated_date is None
|
||||
assert sub2.category.created_by == "Sam"
|
||||
assert sub2.category.updated_by == cat.updated_by
|
||||
|
||||
sub3 = (
|
||||
await Subject.objects.prefetch_related("category")
|
||||
.order_by("-created_date")
|
||||
.exclude_fields({"updated_date": ..., "category": {"updated_date"}})
|
||||
.get()
|
||||
)
|
||||
assert round_date_to_seconds(sub3.created_date) == round_date_to_seconds(
|
||||
sub.created_date
|
||||
)
|
||||
assert sub3.category.updated_date is None
|
||||
assert round_date_to_seconds(
|
||||
sub3.category.created_date
|
||||
) == round_date_to_seconds(cat.created_date)
|
||||
assert sub3.updated_date is None
|
||||
assert sub3.category.created_by == "Sam"
|
||||
assert sub3.category.updated_by == cat.updated_by
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_inheritance_with_relation():
|
||||
async with db:
|
||||
async with db.transaction(force_rollback=True):
|
||||
sam = await Person(name="Sam").save()
|
||||
joe = await Person(name="Joe").save()
|
||||
await Truck(
|
||||
name="Shelby wanna be", max_capacity=1400, owner=sam, co_owner=joe
|
||||
).save()
|
||||
await Bus(name="Unicorn", max_persons=50, owner=sam, co_owner=joe).save()
|
||||
|
||||
shelby = await Truck.objects.select_related(["owner", "co_owner"]).get()
|
||||
assert shelby.name == "Shelby wanna be"
|
||||
assert shelby.owner.name == "Sam"
|
||||
assert shelby.co_owner.name == "Joe"
|
||||
assert shelby.max_capacity == 1400
|
||||
|
||||
unicorn = await Bus.objects.select_related(["owner", "co_owner"]).get()
|
||||
assert unicorn.name == "Unicorn"
|
||||
assert unicorn.owner.name == "Sam"
|
||||
assert unicorn.co_owner.name == "Joe"
|
||||
assert unicorn.max_persons == 50
|
||||
|
||||
joe_check = await Person.objects.select_related(
|
||||
["coowned_trucks", "coowned_buses"]
|
||||
).get(name="Joe")
|
||||
assert joe_check.pk == joe.pk
|
||||
assert joe_check.coowned_trucks[0] == shelby
|
||||
assert joe_check.coowned_trucks[0].created_date is not None
|
||||
assert joe_check.coowned_buses[0] == unicorn
|
||||
assert joe_check.coowned_buses[0].created_date is not None
|
||||
|
||||
joe_check = (
|
||||
await Person.objects.exclude_fields(
|
||||
{
|
||||
"coowned_trucks": {"created_date"},
|
||||
"coowned_buses": {"created_date"},
|
||||
}
|
||||
)
|
||||
.prefetch_related(["coowned_trucks", "coowned_buses"])
|
||||
.get(name="Joe")
|
||||
)
|
||||
assert joe_check.pk == joe.pk
|
||||
assert joe_check.coowned_trucks[0] == shelby
|
||||
assert joe_check.coowned_trucks[0].created_date is None
|
||||
assert joe_check.coowned_buses[0] == unicorn
|
||||
assert joe_check.coowned_buses[0].created_date is None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_inheritance_with_multi_relation():
|
||||
async with db:
|
||||
async with db.transaction(force_rollback=True):
|
||||
sam = await Person(name="Sam").save()
|
||||
joe = await Person(name="Joe").save()
|
||||
alex = await Person(name="Alex").save()
|
||||
truck = await Truck2(
|
||||
name="Shelby wanna be 2", max_capacity=1400, owner=sam
|
||||
).save()
|
||||
await truck.co_owners.add(joe)
|
||||
await truck.co_owners.add(alex)
|
||||
|
||||
bus3 = await Bus2(name="Unicorn 3", max_persons=30, owner=joe).save()
|
||||
await bus3.co_owners.add(sam)
|
||||
|
||||
bus = await Bus2(name="Unicorn 2", max_persons=50, owner=sam).save()
|
||||
await bus.co_owners.add(joe)
|
||||
await bus.co_owners.add(alex)
|
||||
|
||||
shelby = await Truck2.objects.select_related(["owner", "co_owners"]).get()
|
||||
assert shelby.name == "Shelby wanna be 2"
|
||||
assert shelby.owner.name == "Sam"
|
||||
assert shelby.co_owners[0].name == "Joe"
|
||||
assert len(shelby.co_owners) == 2
|
||||
assert shelby.max_capacity == 1400
|
||||
|
||||
unicorn = await Bus2.objects.select_related(["owner", "co_owners"]).get(
|
||||
name="Unicorn 2"
|
||||
)
|
||||
assert unicorn.name == "Unicorn 2"
|
||||
assert unicorn.owner.name == "Sam"
|
||||
assert unicorn.co_owners[0].name == "Joe"
|
||||
assert len(unicorn.co_owners) == 2
|
||||
assert unicorn.max_persons == 50
|
||||
|
||||
unicorn = (
|
||||
await Bus2.objects.select_related(["owner", "co_owners"])
|
||||
.order_by("-co_owners__name")
|
||||
.get()
|
||||
)
|
||||
assert unicorn.name == "Unicorn 2"
|
||||
assert unicorn.owner.name == "Sam"
|
||||
assert len(unicorn.co_owners) == 2
|
||||
assert unicorn.co_owners[0].name == "Joe"
|
||||
|
||||
unicorn = (
|
||||
await Bus2.objects.select_related(["owner", "co_owners"])
|
||||
.order_by("co_owners__name")
|
||||
.get()
|
||||
)
|
||||
assert unicorn.name == "Unicorn 2"
|
||||
assert unicorn.owner.name == "Sam"
|
||||
assert len(unicorn.co_owners) == 2
|
||||
assert unicorn.co_owners[0].name == "Alex"
|
||||
|
||||
joe_check = await Person.objects.select_related(
|
||||
["coowned_trucks2", "coowned_buses2"]
|
||||
).get(name="Joe")
|
||||
assert joe_check.pk == joe.pk
|
||||
assert joe_check.coowned_trucks2[0] == shelby
|
||||
assert joe_check.coowned_trucks2[0].created_date is not None
|
||||
assert joe_check.coowned_buses2[0] == unicorn
|
||||
assert joe_check.coowned_buses2[0].created_date is not None
|
||||
|
||||
joe_check = (
|
||||
await Person.objects.exclude_fields(
|
||||
{
|
||||
"coowned_trucks2": {"created_date"},
|
||||
"coowned_buses2": {"created_date"},
|
||||
}
|
||||
)
|
||||
.prefetch_related(["coowned_trucks2", "coowned_buses2"])
|
||||
.get(name="Joe")
|
||||
)
|
||||
assert joe_check.pk == joe.pk
|
||||
assert joe_check.coowned_trucks2[0] == shelby
|
||||
assert joe_check.coowned_trucks2[0].created_date is None
|
||||
assert joe_check.coowned_buses2[0] == unicorn
|
||||
assert joe_check.coowned_buses2[0].created_date is None
|
||||
|
||||
await shelby.co_owners.remove(joe)
|
||||
await shelby.co_owners.remove(alex)
|
||||
await Truck2.objects.delete(name="Shelby wanna be 2")
|
||||
|
||||
unicorn = (
|
||||
await Bus2.objects.select_related(["owner", "co_owners"])
|
||||
.filter(co_owners__name="Joe")
|
||||
.get()
|
||||
)
|
||||
assert unicorn.name == "Unicorn 2"
|
||||
assert unicorn.owner.name == "Sam"
|
||||
assert unicorn.co_owners[0].name == "Joe"
|
||||
assert len(unicorn.co_owners) == 1
|
||||
assert unicorn.max_persons == 50
|
||||
|
||||
unicorn = (
|
||||
await Bus2.objects.select_related(["owner", "co_owners"])
|
||||
.exclude(co_owners__name="Joe")
|
||||
.get()
|
||||
)
|
||||
assert unicorn.name == "Unicorn 2"
|
||||
assert unicorn.owner.name == "Sam"
|
||||
assert unicorn.co_owners[0].name == "Alex"
|
||||
assert len(unicorn.co_owners) == 1
|
||||
assert unicorn.max_persons == 50
|
||||
|
||||
unicorn = await Bus2.objects.get()
|
||||
assert unicorn.name == "Unicorn 2"
|
||||
assert unicorn.owner.name is None
|
||||
assert len(unicorn.co_owners) == 0
|
||||
await unicorn.co_owners.all()
|
||||
|
||||
assert len(unicorn.co_owners) == 2
|
||||
assert unicorn.co_owners[0].name == "Joe"
|
||||
|
||||
await unicorn.owner.load()
|
||||
assert unicorn.owner.name == "Sam"
|
||||
|
||||
unicorns = (
|
||||
await Bus2.objects.select_related(["owner", "co_owners"])
|
||||
.filter(name__contains="Unicorn")
|
||||
.order_by("-name")
|
||||
.all()
|
||||
)
|
||||
assert unicorns[0].name == "Unicorn 3"
|
||||
assert unicorns[0].owner.name == "Joe"
|
||||
assert len(unicorns[0].co_owners) == 1
|
||||
assert unicorns[0].co_owners[0].name == "Sam"
|
||||
|
||||
assert unicorns[1].name == "Unicorn 2"
|
||||
assert unicorns[1].owner.name == "Sam"
|
||||
assert len(unicorns[1].co_owners) == 2
|
||||
assert unicorns[1].co_owners[0].name == "Joe"
|
||||
|
||||
unicorns = (
|
||||
await Bus2.objects.select_related(["owner", "co_owners"])
|
||||
.filter(name__contains="Unicorn")
|
||||
.order_by("-name")
|
||||
.limit(2, limit_raw_sql=True)
|
||||
.all()
|
||||
)
|
||||
assert len(unicorns) == 2
|
||||
assert unicorns[1].name == "Unicorn 2"
|
||||
assert len(unicorns[1].co_owners) == 1
|
||||
157
tests/test_inheritance/test_inheritance_mixins.py
Normal file
157
tests/test_inheritance/test_inheritance_mixins.py
Normal file
@ -0,0 +1,157 @@
|
||||
# type: ignore
|
||||
import datetime
|
||||
from typing import Optional
|
||||
|
||||
import databases
|
||||
import pytest
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy import create_engine
|
||||
|
||||
import ormar
|
||||
from tests.settings import DATABASE_URL
|
||||
|
||||
metadata = sa.MetaData()
|
||||
db = databases.Database(DATABASE_URL)
|
||||
engine = create_engine(DATABASE_URL)
|
||||
|
||||
|
||||
class AuditMixin:
|
||||
created_by: str = ormar.String(max_length=100)
|
||||
updated_by: str = ormar.String(max_length=100, default="Sam")
|
||||
|
||||
|
||||
class DateFieldsMixins:
|
||||
created_date: datetime.datetime = ormar.DateTime(default=datetime.datetime.now)
|
||||
updated_date: datetime.datetime = ormar.DateTime(default=datetime.datetime.now)
|
||||
|
||||
|
||||
class Category(ormar.Model, DateFieldsMixins, AuditMixin):
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "categories"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=50, unique=True, index=True)
|
||||
code: int = ormar.Integer()
|
||||
|
||||
|
||||
class Subject(ormar.Model, DateFieldsMixins):
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "subjects"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=50, unique=True, index=True)
|
||||
category: Optional[Category] = ormar.ForeignKey(Category)
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True, scope="module")
|
||||
def create_test_database():
|
||||
metadata.create_all(engine)
|
||||
yield
|
||||
metadata.drop_all(engine)
|
||||
|
||||
|
||||
def test_field_redefining():
|
||||
class RedefinedField(ormar.Model, DateFieldsMixins):
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "redefined"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
created_date: datetime.datetime = ormar.DateTime(name="creation_date")
|
||||
|
||||
assert RedefinedField.Meta.model_fields["created_date"].ormar_default is None
|
||||
assert (
|
||||
RedefinedField.Meta.model_fields["created_date"].get_alias() == "creation_date"
|
||||
)
|
||||
assert any(x.name == "creation_date" for x in RedefinedField.Meta.table.columns)
|
||||
|
||||
|
||||
def test_field_redefining_in_second_raises_error():
|
||||
class OkField(ormar.Model, DateFieldsMixins): # pragma: no cover
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "oks"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
|
||||
class RedefinedField2(ormar.Model, DateFieldsMixins):
|
||||
class Meta(ormar.ModelMeta):
|
||||
tablename = "redefines2"
|
||||
metadata = metadata
|
||||
database = db
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
created_date: str = ormar.String(max_length=200, name="creation_date")
|
||||
|
||||
assert RedefinedField2.Meta.model_fields["created_date"].ormar_default is None
|
||||
assert (
|
||||
RedefinedField2.Meta.model_fields["created_date"].get_alias() == "creation_date"
|
||||
)
|
||||
assert any(x.name == "creation_date" for x in RedefinedField2.Meta.table.columns)
|
||||
assert isinstance(
|
||||
RedefinedField2.Meta.table.columns["creation_date"].type,
|
||||
sa.sql.sqltypes.String,
|
||||
)
|
||||
|
||||
|
||||
def round_date_to_seconds(
|
||||
date: datetime.datetime,
|
||||
) -> datetime.datetime: # pragma: no cover
|
||||
if date.microsecond >= 500000:
|
||||
date = date + datetime.timedelta(seconds=1)
|
||||
return date.replace(microsecond=0)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_fields_inherited_from_mixin():
|
||||
async with db:
|
||||
async with db.transaction(force_rollback=True):
|
||||
cat = await Category(
|
||||
name="Foo", code=123, created_by="Sam", updated_by="Max"
|
||||
).save()
|
||||
sub = await Subject(name="Bar", category=cat).save()
|
||||
mixin_columns = ["created_date", "updated_date"]
|
||||
mixin2_columns = ["created_by", "updated_by"]
|
||||
assert all(field in Category.Meta.model_fields for field in mixin_columns)
|
||||
assert cat.created_date is not None
|
||||
assert cat.updated_date is not None
|
||||
assert all(field in Subject.Meta.model_fields for field in mixin_columns)
|
||||
assert sub.created_date is not None
|
||||
assert sub.updated_date is not None
|
||||
|
||||
assert all(field in Category.Meta.model_fields for field in mixin2_columns)
|
||||
assert all(
|
||||
field not in Subject.Meta.model_fields for field in mixin2_columns
|
||||
)
|
||||
|
||||
inspector = sa.inspect(engine)
|
||||
assert "categories" in inspector.get_table_names()
|
||||
table_columns = [x.get("name") for x in inspector.get_columns("categories")]
|
||||
assert all(col in table_columns for col in mixin_columns + mixin2_columns)
|
||||
|
||||
assert "subjects" in inspector.get_table_names()
|
||||
table_columns = [x.get("name") for x in inspector.get_columns("subjects")]
|
||||
assert all(col in table_columns for col in mixin_columns)
|
||||
|
||||
sub2 = (
|
||||
await Subject.objects.select_related("category")
|
||||
.order_by("-created_date")
|
||||
.exclude_fields("updated_date")
|
||||
.get()
|
||||
)
|
||||
assert round_date_to_seconds(sub2.created_date) == round_date_to_seconds(
|
||||
sub.created_date
|
||||
)
|
||||
assert sub2.category.updated_date is not None
|
||||
assert round_date_to_seconds(
|
||||
sub2.category.created_date
|
||||
) == round_date_to_seconds(cat.created_date)
|
||||
assert sub2.updated_date is None
|
||||
assert sub2.category.created_by == "Sam"
|
||||
assert sub2.category.updated_by == cat.updated_by
|
||||
Reference in New Issue
Block a user