Skip to content

Commit

Permalink
Fix constraint name token test case
Browse files Browse the repository at this point in the history
  • Loading branch information
leonarduschen committed Apr 26, 2022
1 parent 553811d commit a75ccae
Showing 1 changed file with 14 additions and 19 deletions.
33 changes: 14 additions & 19 deletions tests/test_generators.py
Original file line number Diff line number Diff line change
Expand Up @@ -2399,29 +2399,24 @@ def test_constraint_name_token(self, generator: CodeGenerator) -> None:
validate_code(
generator.generate(),
"""\
from sqlalchemy import CheckConstraint, Column, Integer, MetaData
from sqlalchemy.orm import declarative_base
from sqlalchemy.sql.elements import conv
metadata = MetaData(
naming_convention={
"ck": "ck_%(table_name)s_%(constraint_name)s",
"pk": "pk_%(table_name)s",
}
)
from sqlalchemy import CheckConstraint, Column, Integer, MetaData
from sqlalchemy.orm import declarative_base
from sqlalchemy.sql.elements import conv
Base = declarative_base(metadata=metadata)
Base = declarative_base()
Base.metadata.naming_convention = \
{'ck': 'ck_%(table_name)s_%(constraint_name)s', 'pk': 'pk_%(table_name)s'}
class Simple(Base):
__tablename__ = 'simple'
__table_args__ = (
CheckConstraint('id > 0', name='idcheck'),
CheckConstraint('number > 0', name=conv('non_default_name'))
)
class Simple(Base):
__tablename__ = 'simple'
__table_args__ = (
CheckConstraint('id > 0', name='idcheck'),
CheckConstraint('number > 0', name=conv('non_default_name'))
)
id = Column(Integer, primary_key=True)
number = Column(Integer)
id = Column(Integer, primary_key=True)
number = Column(Integer)
""",
)

Expand Down

0 comments on commit a75ccae

Please sign in to comment.